def test_add_clinics_to_user(self): self.setup_test_data() user = User() ona_user = OnaUser( user=user, username='******', refresh_token="c563e9") values = {'group': groups.CLINIC_MANAGER, 'clinics': ["1", "2", "3"]} ona_user.update(values) self.assertEqual(len(ona_user.user.clinics), 3)
def test_update_new_user_group(self): user = User() ona_user = OnaUser( user=user, username='******', refresh_token="c563e9") self._create_municipality('Brazlandia') municipality = Municipality.get(Municipality.name == 'Brazlandia') values = {'group': groups.MUNICIPALITY_MANAGER, 'municipality': municipality.id} ona_user.update(values) self.assertEqual(ona_user.user.group.name, groups.MUNICIPALITY_MANAGER) self.assertEqual(ona_user.user.location, municipality)