def test_update_courses(self): u = User(name="unknown", email="*****@*****.**", status="inactive") _id = u.save() u = User.get(_id) self.assertFalse(u.courses, "Courses count") courses = [Course.get(1), Course.get(2), Course.get(3)] u.courses = courses u.update() u = User.get(_id) self.assertTrue(u.courses, "Courses count") self.assertEqual(u.courses, courses, "Update courses") u.delete()
def test_added_category(self): u = User(name="unknown", email="*****@*****.**", status="inactive") courses = [Course.get(1), Course.get(2), Course.get(3)] u.courses = courses _id = u.save() u = User.get(id=_id) self.assertEqual(len(u.courses), len(courses), "Count courses") self.assertEqual(u.courses, courses, "Saving courses") u.delete()