1from __future__ import print_function 

2from builtins import str 

3import os 

4import unittest 

5import shutil 

6import yaml 

7from marshallEngine.utKit import utKit 

8from fundamentals import tools 

9from os.path import expanduser 

10home = expanduser("~") 

11 

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

13# settingsFile = packageDirectory + "/test_settings.yaml" 

14settingsFile = home + "/git_repos/_misc_/settings/marshall/test_settings.yaml" 

15 

16su = tools( 

17 arguments={"settingsFile": settingsFile}, 

18 docString=__doc__, 

19 logLevel="DEBUG", 

20 options_first=False, 

21 projectName=None, 

22 defaultSettingsFile=False 

23) 

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

25 

26# SETUP PATHS TO COMMON DIRECTORIES FOR TEST DATA 

27moduleDirectory = os.path.dirname(__file__) 

28pathToInputDir = moduleDirectory + "/input/" 

29pathToOutputDir = moduleDirectory + "/output/" 

30 

31try: 

32 shutil.rmtree(pathToOutputDir) 

33except: 

34 pass 

35# COPY INPUT TO OUTPUT DIR 

36shutil.copytree(pathToInputDir, pathToOutputDir) 

37 

38# Recursively create missing directories 

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

40 os.makedirs(pathToOutputDir) 

41 

42 

43class test_update_transient_summaries(unittest.TestCase): 

44 

45 def test_update_transient_summaries_function(self): 

46 

47 from marshallEngine.housekeeping import update_transient_summaries 

48 updater = update_transient_summaries( 

49 log=log, 

50 settings=settings, 

51 dbConn=dbConn 

52 ) 

53 updater._add_galactic_coords() 

54 

55 def test_update_transient_summaries_function2(self): 

56 

57 from marshallEngine.housekeeping import update_transient_summaries 

58 updater = update_transient_summaries( 

59 log=log, 

60 settings=settings, 

61 dbConn=dbConn 

62 ) 

63 updater._add_distances() 

64 

65 def test_update_transient_summaries_function4(self): 

66 

67 from marshallEngine.housekeeping import update_transient_summaries 

68 updater = update_transient_summaries( 

69 log=log, 

70 settings=settings, 

71 dbConn=dbConn 

72 ) 

73 updater._update_htm_columns() 

74 

75 def test_update_transient_summaries_function3(self): 

76 

77 from marshallEngine.housekeeping import update_transient_summaries 

78 updater = update_transient_summaries( 

79 log=log, 

80 settings=settings, 

81 dbConn=dbConn 

82 ).update() 

83 

84 def test_update_transient_summaries_function_exception(self): 

85 

86 from marshallEngine.housekeeping import update_transient_summaries 

87 try: 

88 this = update_transient_summaries( 

89 log=log, 

90 settings=settings, 

91 fakeKey="break the code" 

92 ) 

93 this.get() 

94 assert False 

95 except Exception as e: 

96 assert True 

97 print(str(e)) 

98 

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

100 

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