def test_get(self):
count = 5
program_families = ProgramFamilyFactory.create_batch(count)
with self.login(email=self.basic_user().email):
response = self.client.get(self.url)
assert response.data["count"] == count
assert all([
ProgramFamilyListView.serialize(program_family)
in response.data["results"]
for program_family in program_families
])
def test_get_entrepreneur_with_program_family_interests(self):
program_families = ProgramFamilyFactory.create_batch(2)
context = UserContext(user_type=ENTREPRENEUR_USER_TYPE,
program_families=program_families)
user = context.user
with self.login(email=self.basic_user().email):
url = reverse(UserDetailView.view_name, args=[user.id])
response = self.client.get(url)
program_family_ids = [
datum["id"] for datum in response.data["program_families"]
]
assert all([
program_family.id in program_family_ids
for program_family in program_families
])