Coverage for tests/test_tar_meta.py: 100%

1from datetime import datetime 

2 

3import pytest 

4 

5from pepdbagent.models import TarNamespaceModel 

6 

7from .utils import PEPDBAgentContextManager 

8 

9 

10@pytest.mark.skipif( 

11 not PEPDBAgentContextManager().db_setup(), 

12 reason="DB is not setup", 

13) 

14class TestGeoTar: 

15 """ 

16 Test project methods 

17 """ 

18 

19 test_namespace = "namespace1" 

20 

21 tar_info = TarNamespaceModel( 

22 namespace=test_namespace, 

23 submission_date=datetime.now(), 

24 start_period=datetime.now(), 

25 end_period=datetime.now(), 

26 number_of_projects=1, 

27 file_path="blabla/test.tar", 

28 ) 

29 

30 def test_create_meta_tar(self): 

31 with PEPDBAgentContextManager(add_data=True) as agent: 

32 agent.namespace.upload_tar_info(tar_info=self.tar_info) 

33 

34 result = agent.namespace.get_tar_info(namespace=self.test_namespace) 

35 

36 assert result.count == 1 

37 

38 def test_delete_meta_tar(self): 

39 with PEPDBAgentContextManager(add_data=True) as agent: 

40 agent.namespace.upload_tar_info(tar_info=self.tar_info) 

41 

42 result = agent.namespace.get_tar_info(namespace=self.test_namespace) 

43 assert result.count == 1 

44 

45 agent.namespace.delete_tar_info() 

46 

47 result = agent.namespace.get_tar_info(namespace=self.test_namespace) 

48 assert result.count == 0