1from __future__ import print_function 

2from builtins import str 

3import os 

4import unittest 

5import shutil 

6import yaml 

7from atelParser.utKit import utKit 

8from fundamentals import tools 

9from os.path import expanduser 

10home = expanduser("~") 

11 

12packageDirectory = utKit("").get_project_root() 

13settingsFile = packageDirectory + "/test_settings.yaml" 

14 

15su = tools( 

16 arguments={"settingsFile": settingsFile}, 

17 docString=__doc__, 

18 logLevel="DEBUG", 

19 options_first=False, 

20 projectName=None, 

21 defaultSettingsFile=False 

22) 

23arguments, settings, log, dbConn = su.setup() 

24 

25# SETUP PATHS TO COMMON DIRECTORIES FOR TEST DATA 

26moduleDirectory = os.path.dirname(__file__) 

27pathToInputDir = moduleDirectory + "/input/" 

28pathToOutputDir = moduleDirectory + "/output/" 

29 

30try: 

31 shutil.rmtree(pathToOutputDir) 

32except: 

33 pass 

34# COPY INPUT TO OUTPUT DIR 

35shutil.copytree(pathToInputDir, pathToOutputDir) 

36 

37# Recursively create missing directories 

38if not os.path.exists(pathToOutputDir): 

39 os.makedirs(pathToOutputDir) 

40 

41settings["atel-directory"] = pathToOutputDir + "atel-directory" 

42 

43 

44class test_download(unittest.TestCase): 

45 

46 def test_get_latest_atel_number_function(self): 

47 

48 from atelParser import download 

49 atels = download( 

50 log=log, 

51 settings=settings 

52 ) 

53 latestNumber = atels.get_latest_atel_number() 

54 

55 def testget_list_of_atels_still_to_download_function(self): 

56 

57 from atelParser import download 

58 atels = download( 

59 log=log, 

60 settings=settings 

61 ) 

62 atelsToDownload = atels.get_list_of_atels_still_to_download() 

63 atels.maxsleep = 7 

64 atels.download_list_of_atels(atelsToDownload[:1]) 

65 

66 def test_download_function_exception(self): 

67 

68 from atelParser import download 

69 try: 

70 this = download( 

71 log=log, 

72 settings=settings, 

73 fakeKey="break the code" 

74 ) 

75 this.get() 

76 assert False 

77 except Exception as e: 

78 assert True 

79 print(str(e)) 

80 

81 # x-print-testpage-for-pessto-marshall-web-object 

82 

83 # x-class-to-test-named-worker-function