class TestLambda(unittest.TestCase):
def setUp(self):
with open('data/lambda/event1.json') as inf:
self.event = json.load(inf)
uri = 'bolt://localhost:7687'
user = '******'
password = os.environ['NEO_PASSWORD']
self.driver = GraphDatabase.driver(uri, auth=(user, password))
props = Props('../config/props-icdc.yml')
self.schema = ICDC_Schema(['data/icdc-model.yml', 'data/icdc-model-props.yml'], props)
config = BentoConfig('../config/config.ini')
self.processor = FileLoader('', self.driver, self.schema, config, 'ming-icdc-file-loader', 'Final/Data_loader/Manifests')
self.loader = DataLoader(self.driver, self.schema)
self.file_list = [
"data/Dataset/COP-program.txt",
"data/Dataset/NCATS-COP01-case.txt",
"data/Dataset/NCATS-COP01-diagnosis.txt",
"data/Dataset/NCATS-COP01_cohort_file.txt",
"data/Dataset/NCATS-COP01_study_file.txt"
]
def test_join_path(self):
self.assertEqual(self.processor.join_path(), '')
self.assertEqual(self.processor.join_path('abc'), 'abc')
self.assertEqual(self.processor.join_path('/abc'), '/abc')
self.assertEqual(self.processor.join_path('/abc/'), '/abc')
self.assertEqual(self.processor.join_path('abd/def', 'ghi.zip'), 'abd/def/ghi.zip')
self.assertEqual(self.processor.join_path('abd/def/', 'ghi.zip'), 'abd/def/ghi.zip')
self.assertEqual(self.processor.join_path('abd/def//', '//ghi.zip'), 'abd/def/ghi.zip')
self.assertEqual(self.processor.join_path('http://abd/def//', '//ghi.zip//'), 'http://abd/def/ghi.zip')
# Test multiple paths joining
self.assertEqual(self.processor.join_path('abd/def', 'xy/z', 'ghi.zip'), 'abd/def/xy/z/ghi.zip')
self.assertEqual(self.processor.join_path('abd/def/', '/xy/z/' , 'ghi.zip'), 'abd/def/xy/z/ghi.zip')
self.assertEqual(self.processor.join_path('abd/def/', '///xy/z///', '///ghi.zip'), 'abd/def/xy/z/ghi.zip')
def test_lambda(self):
load_result = self.loader.load(self.file_list, True, False, 'upsert', False, 1)
self.assertIsInstance(load_result, dict, msg='Load data failed!')
self.assertTrue(self.processor.handler(self.event))
