40 lines
936 B
Python
40 lines
936 B
Python
""" Test base JSON-RPC classes."""
|
|
import sys
|
|
|
|
from ..base import JSONRPCBaseRequest, JSONRPCBaseResponse
|
|
|
|
if sys.version_info < (2, 7):
|
|
import unittest2 as unittest
|
|
else:
|
|
import unittest
|
|
|
|
|
|
class TestJSONRPCBaseRequest(unittest.TestCase):
|
|
|
|
""" Test JSONRPCBaseRequest functionality."""
|
|
|
|
def test_data(self):
|
|
request = JSONRPCBaseRequest()
|
|
self.assertEqual(request.data, {})
|
|
|
|
with self.assertRaises(ValueError):
|
|
request.data = []
|
|
|
|
with self.assertRaises(ValueError):
|
|
request.data = None
|
|
|
|
|
|
class TestJSONRPCBaseResponse(unittest.TestCase):
|
|
|
|
""" Test JSONRPCBaseResponse functionality."""
|
|
|
|
def test_data(self):
|
|
response = JSONRPCBaseResponse(result="")
|
|
self.assertEqual(response.data, {})
|
|
|
|
with self.assertRaises(ValueError):
|
|
response.data = []
|
|
|
|
with self.assertRaises(ValueError):
|
|
response.data = None
|