class DocumentTransformerTest(TestCase):
def setUp(self):
self.transformer = DocumentTransformer('tests.lisogo.model')
def test_DocumentTransformerIncomingForSimpleDocument(self):
doc = ConcreteDocument('foo', 'bar')
son = doc.toSON()
transformed = self.transformer.transform_incoming(son.copy(), None)
self.assertEqual(transformed, son)
def test_DocumentTransformerIncomingForNestedDocument(self):
doc = DocumentWithNested('foo', NestedDocument('bar'))
son = doc.toSON()
nested_son = doc.nested.toSON()
transformed = self.transformer.transform_incoming(son.copy(), None)
self.assertEqual(transformed['data'], son['data'])
self.assertEqual(transformed['nested'], nested_son)
def test_DocumentTransformerOutgoingForSimpleDocument(self):
doc = ConcreteDocument('foo', 'bar')
son = doc.toSON()
transformed = self.transformer.transform_incoming(son.copy(), None)
back = self.transformer.transform_outgoing(transformed.copy(), None)
self.assertEqual(back, son)
def test_DocumentTransformerOutgoingForNestedDocument(self):
doc = DocumentWithNested('foo', NestedDocument('bar'))
son = doc.toSON()
transformed = self.transformer.transform_incoming(son.copy(), None)
back = self.transformer.transform_outgoing(transformed.copy(), None)
self.assertEqual(back['nested'], son['nested'])
def setUp(self):
self.transformer = DocumentTransformer('tests.lisogo.model')
