class TestDbCommonSerializer(unittest.TestCase): def setUp(self): self.s = Serializer() def tearDown(self): pass def test_document_without_id(self): doc = Document('shiroyuki') encoded_doc = self.s.encode(doc) self.assertFalse('_id' in encoded_doc) self.assertFalse('id' in encoded_doc) self.assertEqual(encoded_doc['name'], 'shiroyuki') def test_document_with_pseudo_id(self): doc = Document('shiroyuki') doc.id = PseudoObjectId() encoded_doc = self.s.encode(doc) self.assertFalse('_id' in encoded_doc) self.assertFalse('id' in encoded_doc) self.assertEqual(encoded_doc['name'], 'shiroyuki') def test_document_with_actual_id(self): doc = Document('shiroyuki') doc.id = ObjectId() encoded_doc = self.s.encode(doc) self.assertTrue('_id' in encoded_doc) self.assertFalse('id' in encoded_doc) self.assertEqual(encoded_doc['name'], 'shiroyuki')
def setUp(self): self.s = Serializer()