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)