def test_sql_parser_hour_and_minute_bad(self):
self.__ensure_schema()
with psycopg2.connect(os.environ['GREASE_TEST_DSN']) as conn:
with conn.cursor() as cursor:
source = sql_source()
cursor.execute("""
CREATE TABLE IF NOT EXISTS
test_data
(
id SERIAL PRIMARY KEY NOT NULL,
name_fs VARCHAR,
name_ls VARCHAR
);
""")
conn.commit()
cursor.execute("""
INSERT INTO
test_data
(name_fs, name_ls)
VALUES
('sally', 'sue');
""")
conn.commit()
self.assertTrue(
source.parse_source({
'name':
'example_source',
'job':
'example_job',
'exe_env':
'general',
'source':
'sql_source',
'type':
'postgresql',
'dsn':
'GREASE_TEST_DSN',
'query':
'select * from test_data;',
'hour': (datetime.datetime.utcnow() +
datetime.timedelta(hours=6)).hour,
'minute': (datetime.datetime.utcnow() +
datetime.timedelta(minutes=10)).minute,
'logic': {}
}))
data = source.get_data()
self.assertTrue(isinstance(data, list))
self.assertEqual(len(data), 0)
cursor.execute("""
DROP TABLE public.test_data
""")
self.__cleanup_schema()
def test_sql_parser_mock(self):
source = sql_source()
conf = Configuration()
mock = {'id': 1, 'name_fs': 'sally', 'name_ls': 'sue'}
fil = open(conf.greaseDir + 'etc' + conf.fs_sep + 'test.mock.sql.json',
'w')
fil.write(json.dumps(mock))
fil.close()
mockData = source.mock_data({})
self.assertEqual(len(mockData), 1)
self.assertEqual(mock.get('id'), 1)
self.assertEqual(mock.get('name_fs'), mockData[0].get('name_fs'))
self.assertEqual(mock.get('name_ls'), mockData[0].get('name_ls'))
os.remove(conf.greaseDir + 'etc' + conf.fs_sep + 'test.mock.sql.json')
def test_sql_parser_bad_type(self):
source = sql_source()
self.assertFalse(source.parse_source({
'name': 'example_source',
'job': 'example_job',
'exe_env': 'general',
'source': 'sql_source',
'type': 'mssql',
'dsn': 'GREASE_TEST_DSN',
'query': 'select * from test_data;',
'logic': {}
}))
data = source.get_data()
self.assertTrue(isinstance(data, list))
self.assertEqual(len(data), 0)
def test_sql_parser(self):
self.__ensure_schema()
with psycopg2.connect(os.environ['GREASE_TEST_DSN']) as conn:
with conn.cursor() as cursor:
source = sql_source()
cursor.execute("""
CREATE TABLE IF NOT EXISTS
test_data
(
id SERIAL PRIMARY KEY NOT NULL,
name_fs VARCHAR,
name_ls VARCHAR
);
""")
conn.commit()
cursor.execute("""
INSERT INTO
test_data
(name_fs, name_ls)
VALUES
('sally', 'sue');
""")
conn.commit()
self.assertTrue(
source.parse_source({
'name': 'example_source',
'job': 'example_job',
'exe_env': 'general',
'source': 'sql_source',
'type': 'postgresql',
'dsn': 'GREASE_TEST_DSN',
'query': 'select * from test_data;',
'logic': {}
}))
data = source.get_data()
self.assertTrue(isinstance(data, list))
self.assertEqual(len(data), 1)
self.assertIsInstance(data[0], dict)
self.assertEqual(data[0].get('id'), 1)
self.assertEqual(data[0].get('name_fs'), 'sally')
self.assertEqual(data[0].get('name_ls'), 'sue')
cursor.execute("""
DROP TABLE public.test_data
""")
self.__cleanup_schema()
def test_type(self):
inst = sql_source()
self.assertTrue(isinstance(inst, sql_source))
