def test_get_split_by_DB2_int(self, mock1):
# Mock table found DB2
with open(BASE_DIR + '/fixtures/db2_table_eval_mock.txt',
'r') as file_h:
sqoop_eval_output1 = file_h.read()
# Mock no primary key
with open(BASE_DIR + '/fixtures/eval_empty.txt', 'r') as file_h:
sqoop_eval_output2 = file_h.read()
# Mock integer columns
with open(BASE_DIR + '/fixtures/db2_primary_eval_mock.txt',
'r') as file_h:
sqoop_eval_output3 = file_h.read()
# Mock integer column in index found
with open(BASE_DIR + '/fixtures/db2_uniqidx_eval_mock.txt',
'r') as file_h:
sqoop_eval_output4 = file_h.read()
mock1.side_effect = [(0, sqoop_eval_output1, ''),
(0, sqoop_eval_output2, ''),
(0, sqoop_eval_output3, ''),
(0, sqoop_eval_output4, '')]
table_obj = ItTable(self.mock_claim_tbl_dict_DB2[0], self.cfg_mgr)
split_by_obj = Get_Auto_Split(self.cfg_mgr)
result = split_by_obj.get_split_by_column(table_obj)
print "Result: ", result
self.assertEquals(result, "KEY")
def test_get_split_by_teradata_pk(self, mock1, mock2):
mock2.return_value = [['KEY']]
with open(BASE_DIR + '/fixtures/td_primary_eval_mock.txt',
'r') as file_h:
sqoop_eval_output = file_h.read()
mock1.return_value = (0, sqoop_eval_output, '')
table_obj = ItTable(self.mock_claim_tbl_dict[0], self.cfg_mgr)
split_by_obj = Get_Auto_Split(self.cfg_mgr)
result = split_by_obj.get_split_by_column(table_obj)
self.assertEquals(result, "KEY")
def test_get_split_by_mysql_int(self, mock1, mock2):
mock2.return_value = [['HIT_ID']]
with open(BASE_DIR + '/fixtures/eval_empty.txt', 'r') as file_h:
sqoop_eval_output1 = file_h.read()
with open(BASE_DIR + '/fixtures/mysqls_intprimary_eval_mock.txt',
'r') as file_h:
sqoop_eval_output2 = file_h.read()
mock1.side_effect = [(0, sqoop_eval_output1, ''),
(0, sqoop_eval_output2, '')]
table_obj = ItTable(self.mock_claim_tbl_dict_mysql[0], self.cfg_mgr)
split_by_obj = Get_Auto_Split(self.cfg_mgr)
result = split_by_obj.get_split_by_column(table_obj)
self.assertEquals(result, 'HIT_ID')
def test_get_split_by_teradata_int(self, mock1, mock2):
mock2.return_value = [['LOAD_CTL_KEY']]
with open(BASE_DIR + '/fixtures/td_valprimary_eval_mock.txt',
'r') as file_h:
sqoop_eval_output1 = file_h.read()
with open(BASE_DIR + '/fixtures/td_noprimary_eval_mock.txt',
'r') as file_h:
sqoop_eval_output2 = file_h.read()
mock1.side_effect = [(0, sqoop_eval_output1, ''),
(0, sqoop_eval_output2, '')]
table_obj = ItTable(self.mock_claim_tbl_dict[0], self.cfg_mgr)
split_by_obj = Get_Auto_Split(self.cfg_mgr)
result = split_by_obj.get_split_by_column(table_obj)
self.assertEquals(result, "LOAD_CTL_KEY")
def test_get_split_by_DB2_errEvalTable_NoIntCol(self, mock1):
# Mock table found
with open(BASE_DIR + '/fixtures/db2_err_eval_mock.txt', 'r') as file_h:
sqoop_eval_output1 = file_h.read()
# Mock table found uDB2
with open(BASE_DIR + '/fixtures/db2_err_eval_mock.txt', 'r') as file_h:
sqoop_eval_output2 = file_h.read()
# Mock integer columns
with open(BASE_DIR + '/fixtures/db2_err_eval_mock.txt', 'r') as file_h:
sqoop_eval_output3 = file_h.read()
mock1.side_effect = [(0, sqoop_eval_output1, ''),
(0, sqoop_eval_output2, ''),
(0, sqoop_eval_output3, '')]
table_obj = ItTable(self.mock_claim_tbl_dict_DB2[0], self.cfg_mgr)
split_by_obj = Get_Auto_Split(self.cfg_mgr)
result = split_by_obj.get_split_by_column(table_obj)
print "Result: ", result
self.assertEquals(result, "no-split")
def test_get_split_by_teradata_warehouse(self, mock1):
mock1.return_value = [['LOAD_CTL_KEY']]
table_obj = ItTable(self.mock_warehouse_tbl_dict[0], self.cfg_mgr)
split_by_obj = Get_Auto_Split(self.cfg_mgr)
result = split_by_obj.get_split_by_column(table_obj)
self.assertEquals(result, "LOAD_CTL_KEY")
