test_jsog.py

import jsog
import unittest

class TestJSOG(unittest.TestCase):

	def test_encode_reference(self):
		inner = { "foo": "bar" }
		outer = { "inner1": inner, "inner2": inner }
		encoded = jsog.encode(outer)

		inner1 = encoded['inner1']
		inner2 = encoded['inner2']

		# one has @id, one has @ref
		self.assertNotEqual('@id' in inner1, '@id' in inner2)
		self.assertNotEqual('@ref' in inner1, '@ref' in inner2)

		if '@id' in inner1:
			self.assertEqual(inner1['@id'], inner2['@ref'])
		else:
			self.assertEqual(inner1['@ref'], inner2['@id'])

	def test_decode_reference(self):
		JSOGIFIED = '{"@id":"1","foo":"foo","inner1":{"@id":"2","bar":"bar"},"inner2":{"@ref":"2"}}'
		parsed = jsog.loads(JSOGIFIED)

		inner1 = parsed['inner1']
		inner2 = parsed['inner2']
		self.assertTrue(inner1 is inner2)

	def test_encode_circular(self):
		thing = {}
		thing['me'] = thing

		encoded = jsog.encode(thing)

		self.assertTrue(encoded['@id'])
		self.assertTrue(encoded['me']['@ref'] == encoded['@id'])

	def test_decode_circular(self):
		thing = {}
		thing['me'] = thing

		encoded = jsog.encode(thing)
		back = jsog.decode(encoded)

		self.assertFalse('@id' in back)
		self.assertTrue(back['me'] is back)

	def test_encode_null(self):
		encoded = jsog.encode(None)
		self.assertEqual(encoded, None)

	def test_decode_null(self):
		decoded = jsog.decode(None)
		self.assertEqual(decoded, None)

	def test_decode_plain_json(self):
		json = { "foo": "bar" }
		decoded = jsog.decode(json)
		self.assertEqual(json, decoded)

if __name__ == '__main__':
	unittest.main()