Coverage for src/tests/modules/club/repositories/test_country_db_repository.py: 100%

1"""Module for testing the country db repository.""" 

2 

3import pytest 

4 

5from kwai_bc_club.domain.country import CountryEntity 

6from kwai_bc_club.repositories.country_db_repository import CountryDbRepository 

7from kwai_bc_club.repositories.country_repository import ( 

8 CountryNotFoundException, 

9 CountryRepository, 

10) 

11from kwai_core.db.database import Database 

12 

13 

14pytestmark = pytest.mark.db 

15 

16 

17@pytest.fixture 

18async def country_repo(database: Database) -> CountryRepository: 

19 """A fixture for a country repository.""" 

20 return CountryDbRepository(database) 

21 

22 

23async def test_create_country(make_country_in_db): 

24 """Test creating a country.""" 

25 country = await make_country_in_db( 

26 CountryEntity(iso_2="XX", iso_3="XXX", name="Test") 

27 ) 

28 assert not country.id.is_empty(), "Country should be created" 

29 

30 

31async def test_get_country_by_iso_2( 

32 country_repo: CountryRepository, make_country_in_db 

33): 

34 """Testing getting a country with an iso_2 code.""" 

35 await make_country_in_db(CountryEntity(iso_2="XX", iso_3="XXX", name="Test")) 

36 country = await country_repo.get_by_iso_2("XX") 

37 assert country.iso_2 == "XX", "The returned country should have iso_2 XX." 

38 

39 

40async def test_delete_country(country_repo: CountryRepository, make_country_in_db): 

41 """Test deleting a country with an iso_2 code.""" 

42 country = await make_country_in_db( 

43 CountryEntity(iso_2="XX", iso_3="XXX", name="Test") 

44 ) 

45 await country_repo.delete(country) 

46 with pytest.raises(CountryNotFoundException): 

47 await country_repo.get_by_iso_2("XX")