Coverage for sherlock/imports/tests/test_veron.py: 86%
49 statements
« prev ^ index » next coverage.py v7.2.2, created at 2023-10-10 13:58 +0000
« prev ^ index » next coverage.py v7.2.2, created at 2023-10-10 13:58 +0000
1from __future__ import print_function
2from builtins import str
3import os
4import unittest
5import shutil
6import yaml
7from sherlock.utKit import utKit
8from fundamentals import tools
9from os.path import expanduser
10home = expanduser("~")
12packageDirectory = utKit("").get_project_root()
13settingsFile = packageDirectory + "/test_settings.yaml"
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()
25# SETUP PATHS TO COMMON DIRECTORIES FOR TEST DATA
26moduleDirectory = os.path.dirname(__file__)
27pathToInputDir = moduleDirectory + "/input/"
28pathToOutputDir = moduleDirectory + "/output/"
30try:
31 shutil.rmtree(pathToOutputDir)
32except:
33 pass
34# COPY INPUT TO OUTPUT DIR
35shutil.copytree(pathToInputDir, pathToOutputDir)
37# Recursively create missing directories
38if not os.path.exists(pathToOutputDir):
39 os.makedirs(pathToOutputDir)
41# SETUP ALL DATABASE CONNECTIONS
42from sherlock import database
43db = database(
44 log=log,
45 settings=settings
46)
47dbConns, dbVersions = db.connect()
48transientsDbConn = dbConns["transients"]
49cataloguesDbConn = dbConns["catalogues"]
51try:
52 from fundamentals.mysql import writequery
53 sqlQuery = """drop table IF EXISTS tcs_cat_veron_v1_0;""" % locals()
54 writequery(
55 log=log,
56 sqlQuery=sqlQuery,
57 dbConn=cataloguesDbConn
58 )
59except:
60 pass
62class test_veron(unittest.TestCase):
64 def test_veron_function(self):
66 from sherlock.imports import veron
67 catalogue = veron(
68 log=log,
69 settings=settings,
70 pathToDataFile=pathToInputDir + "/veron-test-data.txt",
71 version="1.0",
72 catalogueName="veron"
73 )
74 catalogue.ingest()
76 def test_veron_function_exception(self):
78 from sherlock.imports import veron
79 try:
80 this = veron(
81 log=log,
82 settings=settings,
83 fakeKey="break the code"
84 )
85 this.get()
86 assert False
87 except Exception as e:
88 assert True
89 print(str(e))
91 # x-class-to-test-named-worker-function