97 lines
3.0 KiB
Python
97 lines
3.0 KiB
Python
# This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
|
|
import unittest
|
|
import json
|
|
import datetime
|
|
from decimal import Decimal
|
|
|
|
from trytond.protocols.jsonrpc import JSONEncoder, JSONDecoder, JSONRequest
|
|
from trytond.protocols.xmlrpc import client, XMLRequest
|
|
|
|
|
|
class JSONTestCase(unittest.TestCase):
|
|
'Test JSON'
|
|
|
|
def test_json_request(self):
|
|
req = JSONRequest.from_values(
|
|
data=b'{"method": "method", "params": ["foo", "bar"]}',
|
|
content_type='text/json',
|
|
)
|
|
self.assertEqual(req.parsed_data,
|
|
{'method': 'method', 'params': ['foo', 'bar']})
|
|
self.assertEqual(req.rpc_method, 'method')
|
|
self.assertEqual(req.rpc_params, ['foo', 'bar'])
|
|
|
|
def dumps_loads(self, value):
|
|
self.assertEqual(json.loads(
|
|
json.dumps(value, cls=JSONEncoder),
|
|
object_hook=JSONDecoder()), value)
|
|
|
|
def test_datetime(self):
|
|
'Test datetime'
|
|
self.dumps_loads(datetime.datetime.now())
|
|
|
|
def test_date(self):
|
|
'Test date'
|
|
self.dumps_loads(datetime.date.today())
|
|
|
|
def test_time(self):
|
|
'Test time'
|
|
self.dumps_loads(datetime.datetime.now().time())
|
|
|
|
def test_bytes(self):
|
|
'Test bytes'
|
|
self.dumps_loads(bytes(b'foo'))
|
|
self.dumps_loads(bytearray(b'foo'))
|
|
|
|
def test_decimal(self):
|
|
'Test Decimal'
|
|
self.dumps_loads(Decimal('3.141592653589793'))
|
|
|
|
|
|
class XMLTestCase(unittest.TestCase):
|
|
'Test XML'
|
|
|
|
def test_xml_request(self):
|
|
req = XMLRequest.from_values(
|
|
data=b"<?xml version='1.0'?>\n<methodCall>\n<methodName>method</methodName>\n<params>\n<param>\n<value><string>foo</string></value>\n</param>\n<param>\n<value><string>bar</string></value>\n</param>\n</params>\n</methodCall>\n",
|
|
content_type='text/xml')
|
|
self.assertEqual(req.parsed_data, (('foo', 'bar'), 'method'))
|
|
self.assertEqual(req.rpc_method, 'method')
|
|
self.assertEqual(req.rpc_params, ('foo', 'bar'))
|
|
|
|
def dumps_loads(self, value):
|
|
s = client.dumps((value,), allow_none=True)
|
|
result, _ = client.loads(s)
|
|
result, = result
|
|
self.assertEqual(value, result)
|
|
|
|
def test_decimal(self):
|
|
'Test Decimal'
|
|
self.dumps_loads(Decimal('3.141592653589793'))
|
|
|
|
def test_bytes(self):
|
|
'Test bytes'
|
|
self.dumps_loads(bytes(b'foo'))
|
|
self.dumps_loads(bytearray(b'foo'))
|
|
|
|
def test_date(self):
|
|
'Test date'
|
|
self.dumps_loads(datetime.date.today())
|
|
|
|
def test_time(self):
|
|
'Test time'
|
|
self.dumps_loads(datetime.datetime.now().time())
|
|
|
|
def test_none(self):
|
|
'Test None'
|
|
self.dumps_loads(None)
|
|
|
|
|
|
def suite():
|
|
suite_ = unittest.TestSuite()
|
|
suite_.addTests(unittest.TestLoader().loadTestsFromTestCase(JSONTestCase))
|
|
suite_.addTests(unittest.TestLoader().loadTestsFromTestCase(XMLTestCase))
|
|
return suite_
|