1from __future__ import print_function 

2from builtins import str 

3import os 

4import unittest 

5import shutil 

6import yaml 

7from HMpTy.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 

41# RELOAD TEST DATA 

42from fundamentals.mysql import execute_mysql_script 

43exception = execute_mysql_script( 

44 pathToScript=pathToInputDir + "/tcs_cat_ned_d_v13_1_0.sql", 

45 log=log, 

46 dbConn=dbConn 

47) 

48 

49from fundamentals.mysql import writequery 

50sqlQuery = """ALTER TABLE tcs_cat_ned_d_v13_1_0 DROP COLUMN htm16ID, DROP COLUMN htm10ID, DROP COLUMN htm13ID""" 

51try: 

52 writequery( 

53 log=log, 

54 sqlQuery=sqlQuery, 

55 dbConn=dbConn 

56 ) 

57except: 

58 pass 

59 

60 

61class test_add_htm_ids_to_mysql_database_table(unittest.TestCase): 

62 

63 def test_add_htm_ids_to_mysql_database_table_function(self): 

64 

65 from HMpTy.mysql import add_htm_ids_to_mysql_database_table 

66 add_htm_ids_to_mysql_database_table( 

67 raColName="raDeg", 

68 declColName="decDeg", 

69 tableName="tcs_cat_ned_d_v13_1_0", 

70 dbConn=dbConn, 

71 log=log, 

72 primaryIdColumnName="primaryId", 

73 dbSettings=settings["database settings"] 

74 )