def setUp(self): super(TestProfileUpdates, self).setUp() self.afghanistan = CountryFactory(country='Afghanistan', code='AF') self.myanmar = CountryFactory(country='Myanmar', code='MM') self.usa = CountryFactory(country='United States', code='US') self.user = UserFactory(first_name="FN", last_name="LN", username="******") self.user.set_password('password') self.user.save() self.tola_user = TolaUserFactory(user=self.user, country=self.usa) self.tola_user.countries.add(self.afghanistan, self.usa) self.tola_user.save() self.countries_assigned = {self.afghanistan.id, self.usa.id}
def setUp(self): super(TestProjectCountryUserAccess, self).setUp() # create and login user self.user = UserFactory(first_name="FN", last_name="LN", username="******") self.user.set_password('password') self.user.save() self.tola_user = TolaUserFactory(user=self.user) self.client.login(username="******", password='******') self.afghanistan = CountryFactory(country='Afghanistan', code='AF')