Coverage for src/tests/modules/club/repositories/test_file_upload_preview_repository.py: 89%

1"""Module for testing the file upload preview repository.""" 

2 

3import pytest 

4 

5from kwai_bc_club.domain.file_upload import FileUploadEntity 

6from kwai_bc_club.repositories.file_upload_preview_repository import ( 

7 FileUploadPreviewRepository, 

8) 

9from kwai_bc_club.repositories.file_upload_repository import ( 

10 DuplicateMemberUploadedException, 

11) 

12from kwai_core.domain.value_objects.owner import Owner 

13 

14 

15async def test_upload(owner: Owner, make_member): 

16 """Test upload.""" 

17 repo = FileUploadPreviewRepository() 

18 file_upload = FileUploadEntity(filename="test.csv", owner=owner) 

19 try: 

20 await repo.save_member(file_upload, make_member()) 

21 except Exception as error: 

22 pytest.fail(error) 

23 

24 

25async def test_duplicate_upload(owner: Owner, make_member): 

26 """Test if a duplicate member exception is raised when member is uploaded twice.""" 

27 repo = FileUploadPreviewRepository() 

28 file_upload = FileUploadEntity(filename="test.csv", owner=owner) 

29 member = make_member() 

30 with pytest.raises(DuplicateMemberUploadedException): 

31 await repo.save_member(file_upload, member) 

32 await repo.save_member(file_upload, member)