def test_delete_filter(self):
db_filter = Filter(id=self.__class__.filter_id)
delete_id = db_filter.delete_filter()
self.assertEqual(delete_id, self.__class__.filter_id)
connection = Connection(id=self.__class__.conn_id)
connection.delete_conn()
def test_delete_source(self):
source = Source(id=self.__class__.source_id)
delete_id = source.delete_source()
_filter = Filter(id=self.__class__.filter_id)
_filter.delete_filter()
connection = Connection(id=self.__class__.conn_id)
connection.delete_conn()
self.assertEqual(delete_id, self.__class__.source_id)
def test_get_conn(self):
connection = Connection(id=self.__class__.conn_id)
with connection.get_conn() as conn:
cur = conn.execute(
select([func.count()]).select_from(text("public.x_pms_crs_ids"))
)
res = cur.fetchone()
self.assertEqual(res[0], 7)
def test_create_filter(self):
connection = Connection(**CONNECT_ARGS)
connection.create_conn()
self.__class__.conn_id = connection.conn_base.id
FILTER_ARGS['conn_id'] = self.__class__.conn_id
db_filter = Filter(**FILTER_ARGS)
db_filter.create_filter()
self.__class__.filter_id = db_filter.filter_base.id
self.assertTrue(type(db_filter.filter_base.id) == int)
def test_create_source(self):
connection = Connection(**CONNECT_ARGS)
connection.create_conn()
self.__class__.conn_id = connection.conn_base.id
SOURCE_ARGS['conn_id'] = self.__class__.conn_id
source = Source(**SOURCE_ARGS)
source.create_source()
self.__class__.source_id = source.source_base.id
self.assertTrue(type(source.source_base.id) == int)
def get_filter_value(self):
if self.filter_base.filter_type == "feedback":
connection = Connection(id=self.filter_base.conn_id)
query = select([
getattr(func, self.filter_base.operation)(text(
self.filter_base.field))
]).select_from(text(self.filter_base.table))
with connection.get_conn() as conn:
cur = conn.execute(query)
return cur.fetchone()[0]
else:
return self.filter_base.value
def test_delete_conn(self):
connection = Connection(id=self.__class__.conn_id)
deleted_id = connection.delete_conn()
self.assertEqual(deleted_id, self.__class__.conn_id)
def test_filter_conn(self):
conn_array = Connection.fetch_conn(type="redshift")
for conn in conn_array:
self.assertEqual(conn.name, "xyz")
def test_list_conn(self):
conn_array = Connection.list_conn()
for conn in conn_array:
self.assertEqual(conn.name, "xyz")
def test_update_conn(self):
connection = Connection(id=self.__class__.conn_id)
connection.conn_base.password = "******"
connection.conn_base.name = "xyz"
connection.update_conn(password_update=True)
self.assertEqual(connection.conn_base.name, "xyz")
def test_fetch_conn(self):
connection = Connection(id=self.__class__.conn_id)
self.assertEqual(connection.conn_base.name, "warehouse")
def test_create_conn(self):
connection = Connection(**CONNECT_ARGS)
connection.create_conn()
self.__class__.conn_id = connection.conn_base.id
self.assertTrue(type(connection.conn_base.id) == int)