def testSQLCostTaskToXML(self): query = 'select * from dual' task = SQLCostTask(query=query) to_xml = task.serialize() right_xml = sql_cost_template % {'sql': query} self.assertEqual(to_str(to_xml), to_str(right_xml)) task = Task.parse(None, to_xml) self.assertIsInstance(task, SQLCostTask)
def testMergeTaskToXML(self): task = MergeTask('task_1', table='table_name') task.update_settings({'odps.merge.cross.paths': True}) to_xml = task.serialize() right_xml = merge_template % dict(name='task_1', table='table_name') self.assertEqual(to_str(to_xml), to_str(right_xml)) task = Task.parse(None, to_xml) self.assertIsInstance(task, MergeTask)
def testCupidTaskToXML(self): task = CupidTask('task_1', 'plan_text', {'odps.cupid.wait.am.start.time': 600}) to_xml = task.serialize() right_xml = cupid_template self.assertEqual(to_str(to_xml), to_str(right_xml)) task = Task.parse(None, to_xml) self.assertIsInstance(task, CupidTask)
def testSQLTaskToXML(self): query = 'select * from dual' task = SQLTask(query=query) to_xml = task.serialize() right_xml = template % {'sql': query} self.assertEqual(to_str(to_xml), to_str(right_xml)) task = Task.parse(None, to_xml) self.assertIsInstance(task, SQLTask)