class ItemTestCase(unittest.TestCase): def setUp(self): self.user = User('mock-name', 'mock-pass') self.user.id = 123 self.item = Item('mock-item', self.user) def test_buy(self): self.item.buy(self.user, True) self.assertEqual(self.item.bought_by, self.user.id) self.assertTrue(self.item.bought_date, not None) def test_serialize(self): data = {'id': self.item.id, 'name': self.item.name, 'create_date': self.item.create_date, 'created_by': self.item.created_by, 'bought_date': self.item.bought_date, 'bought_by': self.item.bought_by} self.assertEqual(self.item.serialize(), data)