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)
