diff --git a/ecs/jskult-batch/tests/manager/test_jskult_hdke_tbl_manager.py b/ecs/jskult-batch/tests/manager/test_jskult_hdke_tbl_manager.py index 380dbe61..62bab46a 100644 --- a/ecs/jskult-batch/tests/manager/test_jskult_hdke_tbl_manager.py +++ b/ecs/jskult-batch/tests/manager/test_jskult_hdke_tbl_manager.py @@ -14,23 +14,24 @@ class TestJskultHdkeTblManager: # Setup db = Database.get_instance() db.connect() - backup_hdke_tbl = db.execute_select('SELECT * FROM src07.hdke_tbl')[0] + backup_hdke_tbl = db.execute_select('SELECT * FROM src07.hdke_tbl') db.execute('DELETE FROM src07.hdke_tbl') yield # Teardown - db.execute('DELETE FROM src07.hdke_tbl') - db.execute(""" - INSERT INTO src07.hdke_tbl - (syor_date, bch_actf, dump_sts_kbn, creater, create_date, updater, update_date) - VALUES - (:syor_date, :bch_actf, :dump_sts_kbn, CURRENT_USER(), CURRENT_TIMESTAMP(), NULL, NULL) - """, { - 'syor_date': backup_hdke_tbl['syor_date'], - 'bch_actf': backup_hdke_tbl['bch_actf'], - 'dump_sts_kbn': backup_hdke_tbl['dump_sts_kbn'], - }) + if len(backup_hdke_tbl) != 0: + db.execute('DELETE FROM src07.hdke_tbl') + db.execute(""" + INSERT INTO src07.hdke_tbl + (syor_date, bch_actf, dump_sts_kbn, creater, create_date, updater, update_date) + VALUES + (:syor_date, :bch_actf, :dump_sts_kbn, CURRENT_USER(), CURRENT_TIMESTAMP(), NULL, NULL) + """, { + 'syor_date': backup_hdke_tbl['syor_date'], + 'bch_actf': backup_hdke_tbl['bch_actf'], + 'dump_sts_kbn': backup_hdke_tbl['dump_sts_kbn'], + }) db.disconnect() def test_get_batch_statuses(self):