def test_all_1(self):
table_name = 'testing'
c_1 = Column(str, 'column_1')
c_2 = Column(str, 'column_2')
q = Query([c_1, c_2], table_name)
response = q.all(fakeSessionValidData)
self.assertEqual([{'a': 10}, {'b': 20}], response)
def test_all_2(self):
table_name = 'testing'
c_1 = Column(str, 'column_1')
c_2 = Column(str, 'column_2')
q = Query([c_1, c_2], table_name)
response = q.all(fakeSessionEmptyData)
self.assertEqual({}, response)
def test_parse_in_list_2(self):
table_name = 'testing'
c_1 = Column(str, 'column_1')
c_2 = Column(str, 'column_2')
q = Query([c_1, c_2], table_name)
response = q._parse_in_list([10, 20])
self.assertEqual('(10, 20)', response)
def test_first_2(self):
table_name = 'testing'
c_1 = Column(str, 'column_1')
c_2 = Column(str, 'column_2')
q = Query([c_1, c_2], table_name)
response = q.first(fakeSessionValidData)
self.assertEqual({'a': 10}, response)
def test_parse_in_list(self):
table_name = 'testing'
c_1 = Column(str, 'column_1')
c_2 = Column(str, 'column_2')
q = Query([c_1, c_2], table_name)
response = q._parse_in_list(['v1', 'v2'])
self.assertEqual('(\'v1\', \'v2\')', response)
def test_query_simple_1(self):
table_name = 'testing'
c_1 = Column(str, 'column_1')
c_2 = Column(str, 'column_2')
q = Query([c_1, c_2], table_name)
expected_response = 'SELECT column_1, column_2 FROM ' + \
self.table_date_range
response = q.assemble()
self.assertEquals(expected_response, response)
def test_query_order_3(self):
table_name = 'testing'
c_1 = Column(str, 'column_1')
c_2 = Column(str, 'column_2')
q = Query([c_1, c_2], table_name)
q = q.order_by(c_1, desc=True)
expected_response = 'SELECT column_1, column_2 FROM ' + \
self.table_date_range + ' ORDER BY column_1 DESC'
response = q.assemble()
self.assertEquals(expected_response, response)
def basic_test_2(self):
c_1 = Column(str, 'c1')
c_2 = Column(int, 'c2')
q = Query([c_1, c_2], 'test')
q.filter(or_(c_1 == 10, c_2 == 10))
expected_query = 'SELECT c1, c2 FROM ' +\
'' + DATABASE_NAME + '.test WHERE _PARTITIONTIME BETWEEN ' +\
'TIMESTAMP(\'2010-01-01\') AND ' +\
'TIMESTAMP(\'2030-01-01\')'
self.assertEquals(q.assemble(), expected_query)
def test_query_filter_2(self):
table_name = 'testing'
c_1 = Column(str, 'column_1')
c_2 = Column(str, 'column_2')
q = Query([c_1, c_2], table_name)
# Wrong comparision
q = q.filter(c_1 == 10)
expected_response = 'SELECT column_1, column_2 FROM ' + \
self.table_date_range
response = q.assemble()
self.assertEquals(expected_response, response)
def test_query_filter_1(self):
table_name = 'testing'
c_1 = Column(str, 'column_1')
c_2 = Column(str, 'column_2')
q = Query([c_1, c_2], table_name)
q = q.filter(c_1 == 'test')
expected_response = 'SELECT column_1, column_2 FROM ' + \
self.table_date_range + ' AND ' + \
'column_1 = \'test\''
response = q.assemble()
self.assertEquals(expected_response, response)
def test_query_limit_2(self):
table_name = 'testing'
c_1 = Column(str, 'column_1')
c_2 = Column(str, 'column_2')
q = Query([c_1, c_2], table_name)
q = q.limit(100)
q = q.limit(101)
expected_response = 'SELECT column_1, column_2 FROM ' + \
self.table_date_range + ' LIMIT 101'
response = q.assemble()
self.assertEquals(expected_response, response)
def test_query_filter_by_date(self):
begin_date = '2015-12-01'
end_date = '2016-01-01'
table_date_range = '' + DATABASE_NAME + '.testing WHERE _PARTITIONTIME BETWEEN ' +\
'TIMESTAMP(\'' + begin_date + '\') AND TIMESTAMP' +\
'(\'' + end_date + '\')'
table_name = 'testing'
c_1 = Column(str, 'column_1')
c_2 = Column(str, 'column_2')
q = Query([c_1, c_2], table_name).filter_by_date(begin_date, end_date)
expected_response = 'SELECT column_1, column_2 FROM ' + \
table_date_range
response = q.assemble()
self.assertEquals(expected_response, response)
def query(self, *args):
if not self.__tablename__:
logging.error("Table name not defined...")
return None
columns = []
if not args:
columns = self._get_all_columns()
else:
columns = [i for i in args]
if not columns:
return None
return Query(columns, self.__tablename__)
def test_get_filter_keys(self):
c_1 = Column(str, 'column_1')
c_2 = Column(str, 'column_2')
q = Query([c_1, c_2], "")
table_name = 'adwords_account_report'
q.table_name = table_name
key = q._get_filter_key()
self.assertEquals(key, 'account_id')
table_name = 'adwords_campaign_report'
q.table_name = table_name
q.table_name = table_name
key = q._get_filter_key()
self.assertEquals(key, 'campaign_id')
table_name = 'adwords_adgroup_report'
q.table_name = table_name
key = q._get_filter_key()
self.assertEquals(key, 'adgroup_id')
table_name = 'adwords_ad_report'
q.table_name = table_name
key = q._get_filter_key()
self.assertEquals(key, 'ad_id')
table_name = 'adwords_keyword_report'
q.table_name = table_name
key = q._get_filter_key()
self.assertEquals(key, 'keyword_id')