1from __future__ import print_function 

2from builtins import str 

3import os 

4import unittest 

5import shutil 

6import unittest 

7import yaml 

8import sys 

9from fundamentals.utKit import utKit 

10from fundamentals import tools 

11from os.path import expanduser 

12home = expanduser("~") 

13 

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

15settingsFile = packageDirectory + "/test_settings.yaml" 

16# settingsFile = home + "/.config/soxspipe.recipes/soxspipe.recipes.yaml" 

17su = tools( 

18 arguments={"settingsFile": settingsFile}, 

19 docString=__doc__, 

20 logLevel="DEBUG", 

21 options_first=False, 

22 projectName=None, 

23 defaultSettingsFile=False 

24) 

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

26 

27# SETUP AND TEARDOWN FIXTURE FUNCTIONS FOR THE ENTIRE MODULE 

28moduleDirectory = os.path.dirname(__file__) 

29utKit = utKit(moduleDirectory) 

30log, dbConn, pathToInputDir, pathToOutputDir = utKit.setupModule() 

31utKit.tearDownModule() 

32 

33try: 

34 shutil.rmtree(pathToOutputDir) 

35except: 

36 pass 

37# COPY INPUT TO OUTPUT DIR 

38shutil.copytree(pathToInputDir, pathToOutputDir) 

39 

40# Recursively create missing directories 

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

42 os.makedirs(pathToOutputDir) 

43 

44# shutil.copytree(pathToInputDir + "/mysql_scripts", 

45# pathToOutputDir + "/mysql_scripts") 

46 

47 

48class test_directory_script_runner(unittest.TestCase): 

49 

50 def test_directory_script_runner_function(self): 

51 

52 from fundamentals.mysql import directory_script_runner 

53 directory_script_runner( 

54 log=log, 

55 pathToScriptDirectory=pathToOutputDir + "/mysql_scripts", 

56 databaseName="unit_tests", 

57 loginPath="unittesting" 

58 ) 

59 

60 def test_directory_script_runner_function_02(self): 

61 

62 from fundamentals.mysql import directory_script_runner 

63 directory_script_runner( 

64 log=log, 

65 pathToScriptDirectory=pathToOutputDir + "/mysql_scripts", 

66 databaseName="unit_tests", 

67 loginPath="unittesting", 

68 successRule="passed", 

69 failureRule="failed" 

70 ) 

71 

72 def test_directory_script_runner_function_03(self): 

73 

74 from fundamentals.mysql import directory_script_runner 

75 directory_script_runner( 

76 log=log, 

77 pathToScriptDirectory=pathToOutputDir + "/mysql_scripts", 

78 databaseName="unit_tests", 

79 loginPath="unittesting", 

80 successRule="delete", 

81 failureRule="failed" 

82 ) 

83 

84 def test_directory_script_runner_function_04(self): 

85 

86 from fundamentals.mysql import directory_script_runner 

87 directory_script_runner( 

88 log=log, 

89 pathToScriptDirectory=pathToOutputDir + "/mysql_scripts", 

90 databaseName="unit_tests", 

91 loginPath="unittesting", 

92 successRule="delete", 

93 failureRule="delete" 

94 ) 

95 

96 def test_directory_script_runner_function_05(self): 

97 

98 from fundamentals.mysql import directory_script_runner 

99 directory_script_runner( 

100 log=log, 

101 pathToScriptDirectory=pathToOutputDir + "/mysql_scripts", 

102 force=True, 

103 databaseName="unit_tests", 

104 loginPath="unittesting", 

105 successRule="delete", 

106 failureRule="failed" 

107 ) 

108 

109 def test_directory_script_runner_function_06(self): 

110 

111 from fundamentals.mysql import directory_script_runner 

112 directory_script_runner( 

113 log=log, 

114 pathToScriptDirectory=pathToOutputDir + "/mysql_scripts", 

115 dbConn=dbConn, 

116 successRule="delete", 

117 failureRule="failed" 

118 ) 

119 

120 def test_01_execute_mysql_script_function_01(self): 

121 

122 pathToScript = pathToOutputDir + \ 

123 "/mysql_scripts/marshall_schema.sql" 

124 

125 from fundamentals.mysql import execute_mysql_script 

126 exception = execute_mysql_script( 

127 pathToScript=pathToScript, 

128 log=log, 

129 dbConn=dbConn 

130 ) 

131 if exception: 

132 print(exception) 

133 

134 pathToScript = pathToOutputDir + \ 

135 "/mysql_scripts/42340d_since_20160921_tns_conesearch_phot copy.sql" 

136 

137 from fundamentals.mysql import execute_mysql_script 

138 exception = execute_mysql_script( 

139 pathToScript=pathToScript, 

140 log=log, 

141 dbConn=dbConn 

142 ) 

143 if exception: 

144 print(exception) 

145 

146 pathToScript = pathToOutputDir + \ 

147 "/mysql_scripts/42340d_since_20160921_tns_conesearch_phot.sql" 

148 

149 from fundamentals.mysql import execute_mysql_script 

150 exception = execute_mysql_script( 

151 pathToScript=pathToScript, 

152 log=log, 

153 dbConn=dbConn 

154 ) 

155 if exception: 

156 print(exception) 

157 

158 def test_directory_script_runner_function_exception(self): 

159 

160 from fundamentals.mysql import directory_script_runner 

161 try: 

162 directory_script_runner( 

163 log=log, 

164 settings=settings, 

165 fakeKey="break the code" 

166 ) 

167 assert False 

168 except Exception as e: 

169 assert True 

170 # print(str(e)) 

171 

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

173 

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