Files
tryton/trytond/tests/test_protocols.py
T
Cédric Krier 29129855ec Add support for Python 3
issue3211
review7311002
2016-02-07 16:46:16 +01:00

79 lines
2.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
from trytond.protocols.xmlrpc import client
class JSONTestCase(unittest.TestCase):
'Test JSON'
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 dumps_loads(self, value):
s = client.dumps((value,))
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_