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")