def test_all_mock(self, client, mocker): # arrange test_countries = [CountryMother.one().build(), CountryMother.two().build()] mocker.patch('modules.DBRepo.CountryDBRepo.CountryDBRepo.get_all', return_value=test_countries) expected_resp = [countryToDict(test_countries[i]) for i in range(len(test_countries))] view = CountriesListView() # act resp = view.get(HttpRequest()) # assert assert expected_resp == resp.data
def test_get_all_my_mock(self): # arrange expected_countries = [ CountryMother.one().build(), CountryMother.two().build() ] usecase = CountryUsecases(CountryDBRepoMock()) # act result_countries = usecase.get_all_countries() # assert for i in range(len(expected_countries)): assert result_countries[i].name == expected_countries[i].name
def test_get_success(self, mocker): # arrange expected_countries = [ CountryMother.one().build(), CountryMother.two().build() ] mocker.patch('modules.DBRepo.CountryDBRepo.CountryDBRepo.get_all', return_value=expected_countries) usecase = CountryUsecases(CountryDBRepo()) # act result_countries = usecase.get_all_countries() # assert for i in range(len(expected_countries)): assert result_countries[i].name == expected_countries[i].name
def test_get_one_my_mock(self): # arrange test_id = 1 expected_country = CountryMother.one().build() usecase = CountryUsecases(CountryDBRepoMock()) # act result_countries = usecase.get_country(test_id) # assert assert result_countries.name == expected_country.name
def test_get_mock(self, client, mocker): # arrange test_country = CountryMother.one().build() test_id = 1 mocker.patch('modules.DBRepo.CountryDBRepo.CountryDBRepo.get', return_value=test_country) view = CountryView() # act resp = view.get(HttpRequest(), test_id) # assert assert countryToDict(test_country) == resp.data
def test_get_one_success_london(self, mocker): # arrange test_id = 1 expected_country = CountryMother.one().build() mocker.patch('modules.DBRepo.CountryDBRepo.CountryDBRepo.get', return_value=expected_country) usecase = CountryUsecases(CountryDBRepo()) # act result_countries = usecase.get_country(test_id) # assert assert result_countries.name == expected_country.name
def get(country_id) -> Country: return CountryMother.one().build()
def get_all() -> List[Country]: return [CountryMother.one().build(), CountryMother.two().build()]