def test_init(self):
cat = ojc.ObjectCatalog(12345, 'fits_test', 'test.fits', None, self.table, 'TESTER', False,
'services.ini', 'db-test')
self.assertTrue(cat.constDict)
with patch('databaseapps.objectcatalog.ingestutils.resolveDbObject', side_effect=[('PROD', 'SE_OBJECT'), ('test','test_temp')]):
with capture_output() as (out, _):
cat = ojc.ObjectCatalog(12345, 'fits_test', 'test.fits', 'test_temp', self.table, 0, True,
'services.ini', 'db-test')
self.assertFalse(cat.constDict)
output = out.getvalue().strip()
self.assertTrue('test_temp' in output)
self.assertTrue('PROD' in output)
def test_info(self):
cat = ojc.ObjectCatalog(12345, 'fits_test', 'test.fits', None, self.table, 'TESTER', False,
'services.ini', 'db-test')
with capture_output() as (out, _):
cat.info('MY TEST MESSAGE')
output = out.getvalue().strip()
self.assertTrue(output.endswith('TEST MESSAGE'))
def test_getObjectColumns(self):
self.assertRaises(SystemExit, ojc.ObjectCatalog, 12345, 'bad_test', 'test.fits', 'test_temp', self.table, 0, True,
'services.ini', 'db-test')
cat = ojc.ObjectCatalog(12345, 'fits_test2', 'test.fits', None, self.table, 'TESTER', False,
'services.ini', 'db-test')
self.assertTrue(cat.dbDict)
def test_loadingTarget(self):
cat = ojc.ObjectCatalog(12345, 'fits_test', 'test.fits', None, self.table, 'TESTER', False,
'services.ini', 'db-test')
self.assertFalse(cat.loadingTarget())
cat.temptable = 'SE_OBJECT'
self.assertTrue(cat.loadingTarget())
cat.tempschema = 'test'
self.assertFalse(cat.loadingTarget())
def test_insert_many(self):
cat = ojc.ObjectCatalog(12345, 'fits_test', 'test.fits', None, self.table, 'TESTER', False,
'services.ini', 'db-test')
cat.insert_many('', [], {})
rows = [('fname', 12.34, 2, 0)]
columns = ['filename', 'ra', 'count', 'rownum']
cat.insert_many('DATAFILE_INGEST_TEST', columns, rows)
def test_numAlreadyIngested(self):
cat = ojc.ObjectCatalog(12345, 'fits_test', 'test.fits', None, self.table, 'TESTER', False,
'services.ini', 'db-test')
cat.targetschema = "'" + cat.targetschema
cat.targettable += "'"
self.assertEqual(cat.numAlreadyIngested()[0], 4)
cat.shortfilename = 'test2.fits'
self.assertEqual(cat.numAlreadyIngested()[0], 0)
cat.targetschema += 'x'
self.assertRaises(Exception, cat.numAlreadyIngested)
def test_parseFitsTypeLength(self):
formats = {'count': 'J',
'ra': 'E',
'comment': '2A',
'naxis': '3J'}
cat = ojc.ObjectCatalog(12345, 'fits_test', 'test.fits', None, self.table, 'TESTER', False,
'services.ini', 'db-test')
sizes, types = cat.parseFitsTypeLength(formats)
self.assertEqual(sizes['naxis'], 3)
self.assertEqual(sizes['ra'], 1)
self.assertEqual(types['naxis'], 'J')
self.assertEqual(types['ra'], 'E')
def test_checkForArrays(self):
cat = ojc.ObjectCatalog(12345, 'fits_test', 'test.fits', None, self.table, 'TESTER', False,
'services.ini', 'db-test')
cat.objhdu = 'blah'
initial = copy.deepcopy(cat.dbDict)
cat.checkForArrays(cat.dbDict)
self.assertDictEqual(initial, cat.dbDict)
test_dict = OrderedDict({'blah': {'a_3': [['temp'], 'h', 'float_external', ['0']],
'RA': [['ra'], 'h', 'float_external', ['0']],
'RA_1': [['ra'], 'h', 'float external', ['0']]},
})
initial = copy.deepcopy(test_dict)
cat.checkForArrays(test_dict)
self.assertNotEqual(test_dict, initial)
def test_getConstValuesFromHeader(self):
cat = ojc.ObjectCatalog(12345, 'fits_test', 'test.fits', None, self.table, 'TESTER', False,
'services.ini', 'db-test')
length = len(cat.constlist)
cat.getConstValuesFromHeader('TESTER')
self.assertNotEqual(len(cat.constlist), length)