def test_add_item_to_user(self):
u = User(first_name="tanawat", last_name="khunlertkit", username="******")
item = Item(name="car")
self.assertTrue(u.add_to_datastore())
item.put()
u.add_item(item)
self.assertFalse(u.add_to_datastore())
self.assertEqual(1, len(Item.query().fetch()))
u_list = User.query(User.username == "tanawat").fetch()
self.assertEqual(1, len(u_list))
self.assertEqual(1, len(u_list[0].items))
def test_add_duplicate(self): u = User(first_name='A', last_name='B', username='******') u.put() self.assertEqual(len(User.query().fetch()), 1) item = Item(name='car') res = u.add_item(item) expected = 'added item car to user C' self.assertEqual(res, expected) res = u.add_item(item) expected = 'C already has car' self.assertEqual(res, expected)
