def test_run_rerun_runtime(self):
# Slow run
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1, "-v"
]
time_unix_ms, time_human = get_current_time()
start = time_unix_ms
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
time_unix_ms, time_human = get_current_time()
end = time_unix_ms
runtime1 = end - start
# Fast run run2<run1
time_unix_ms, time_human = get_current_time()
start = time_unix_ms
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
time_unix_ms, time_human = get_current_time()
end = time_unix_ms
runtime2 = end - start
self.assertGreater(runtime1 * 1.5, runtime2)
# Middle run: run3>run2
PathManager.unlink("outdir/output_file1.txt")
time_unix_ms, time_human = get_current_time()
start = time_unix_ms
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
time_unix_ms, time_human = get_current_time()
end = time_unix_ms
runtime3 = end - start
self.assertLess(runtime2, runtime3)
def test_dry_drun_skipping_all_but_one(self):
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1, "-v"
]
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
PathManager.unlink("outdir/output_file7.txt")
with self.assertRaises(SystemExit):
WopMars().run(cmd_line + ["-n"])
def test_dry_drun_skipping(self):
cmd_line = [
"python", "-D", self.__db_url, "-w",
self.__example_def_file2_only_files, "-v"
]
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
with self.assertRaises(SystemExit):
WopMars().run(cmd_line + ["-n"])
def test_clear_history(self):
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__right_def_file, "-vv",
"-p", "-d",
PathFinder.get_module_path(), "-c", "-F"
]
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
WopMars().run(cmd_line)
session = SQLManager.instance().get_session()
self.assertEqual(session.query(Execution).count(), 1)
def test_dry_drun_skipping_all_but_one(self):
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__right_def_file, "-vv",
"-p"
]
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
PathFinder.silentremove("test/output/output_File7.txt")
with self.assertRaises(SystemExit):
WopMars().run(cmd_line + ["-n"])
def test_clear_history(self):
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"-v", "-d", self.test_path
]
cmd_line_clear_history = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"-v", "-d", self.test_path, "--cleanup-metadata"
]
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
with self.assertRaises(SystemExit):
WopMars().run(cmd_line_clear_history)
session = SQLManager.instance().get_session()
self.assertEqual(session.query(Execution).count(), 0)
def test_run_error_in_command_line(self):
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"--dot", "/usr/"
]
with self.assertRaises(SystemExit) as SE:
WopMars().run(cmd_line)
self.assertEqual(SE.exception.code, 2)
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"--not-known"
]
with self.assertRaises(SystemExit) as SE:
WopMars().run(cmd_line)
self.assertEqual(SE.exception.code, 2)
def test_run_packaged_wrappers(self):
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file4,
"-v", "-d", self.test_path
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertEqual(se.exception.code, 0)
def test_04run_sourcerule_fail(self):
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__right_def_file, "-vv",
"-p", "--sourcerule", "failure"
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertEqual(se.exception.code, 1)
def test_run6(self):
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__right_def_file, "-p",
"-vv"
]
start = time.time()
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
end = time.time()
runtime1 = end - start
start = time.time()
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
end = time.time()
runtime2 = end - start
self.assertGreater(runtime1 * 1.5, runtime2)
def test_run_target_rule(self):
# SQLManager.instance().create_all()
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"-v", "--until", "rule3"
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertEqual(se.exception.code, 0)
def test_03run_that_fail(self):
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__def_file_never_ready,
"-vv", "-p", "-d",
PathFinder.get_module_path()
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertEqual(se.exception.code, 1)
def test_01run(self):
cmd_line = [
"python", "-l", "-D", self.__db_url, "-w", self.__right_def_file,
"-v", "-p", "-d",
PathFinder.get_module_path()
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertEqual(se.exception.code, 0)
def test_run_input_file_not_ready(self):
# SQLManager.instance().create_all()
cmd_line = [
"python", "-D", self.__db_url, "-w",
self.__example_def_file_input_not_ready, "-v"
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertEqual(se.exception.code, 1)
def test_05run_skiping_steps_time_check(self):
cmd_line = [
"python", "-D", self.__db_url, "-w",
self.__right_def_file_only_files, "-vv", "-p"
]
start = time.time()
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
end = time.time()
runtime1 = end - start
start = time.time()
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
end = time.time()
runtime2 = end - start
self.assertGreater(runtime1 * 1.5, runtime2)
PathFinder.silentremove("test/output/output_File1.txt")
def test_core(self):
cmd_line = [
"python", "tool", "test.resource.wrapper.FooWrapperCore", "-o",
"{'table': {'FooBase': 'test.resource.model.FooBase'}}", "-vv",
"-p", "-D", self.__db_url, "-d",
PathFinder.get_module_path()
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertEqual(se.exception.code, 0)
def test_run_one_tool(self):
cmd_line = [
"python", "tool", "test.resource.wrapper.FooWrapper4", "-i",
"{'file': {'input1': 'test/resource/input_files/input_File1.txt'}}",
"-o", "{'file': {'output1': 'test/output/output1.txt'}}", "-D",
self.__db_url, "-vv", "-p", "-d",
PathFinder.get_module_path()
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertEqual(se.exception.code, 0)
def test_run_that_fails(self):
cmd_line = [
"python", "-D", self.__db_url, "-w",
self.__example_def_file5_never_ready, "-vv", "-d", self.test_path
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertFalse(
os.path.exists(
os.path.join(self.test_path, 'resources/never_done.txt')))
self.assertEqual(se.exception.code, 1)
def test_run_sourcerule_fail(self):
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"-v", "--since", "failure"
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertFalse(
os.path.exists(
os.path.join(self.test_path, 'outdir/output_file1.txt')))
self.assertEqual(se.exception.code, 1)
def test_run_one_tool(self):
cmd_line = [
"python", "tool", "wopmars.tests.resource.wrapper.FooWrapper4",
"-i",
"{'file': {'input1': 'resource/input_files/input_file1.txt'}}",
"-o", "{'file': {'output1': 'outdir/output1.txt'}}", "-D",
self.__db_url, "-v", "-d", self.test_path
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertEqual(se.exception.code, 0)
def test_run_skipping_steps_time_check(self):
cmd_line = [
"python", "-D", self.__db_url, "-w",
self.__example_def_file2_only_files, "-v"
]
time_unix_ms, time_human = get_current_time()
start = time_unix_ms
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
time_unix_ms, time_human = get_current_time()
end = time_unix_ms
runtime1 = end - start
time_unix_ms, time_human = get_current_time()
start = time_unix_ms
with self.assertRaises(SystemExit):
WopMars().run(cmd_line)
time_unix_ms, time_human = get_current_time()
end = time_unix_ms
runtime2 = end - start
self.assertGreater(runtime1 * 1.5, runtime2)
# pathlib.Path('outdir/output_file1.txt').unlink()
PathManager.unlink("outdir/output_file1.txt")
def test_forceall_dryrun(self):
"""This tests carries out a normal a forceall and a dryrun and stores the modification times.
Then it will tests that the modification time of the two first outputs are different and the
second and third equal"""
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"-v", "-d", self.test_path
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
# Test that modification date of first run is different from second forced run
mtime1 = os.path.getmtime(
os.path.join(self.test_path, 'outdir/output_file7.txt'))
# Run forceall
cmd_line_force = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"-v", "-d", self.test_path, "--forceall"
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line_force)
mtime2 = os.path.getmtime(
os.path.join(self.test_path, 'outdir/output_file7.txt'))
# Run dry-run
cmd_line_dry_run = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"-v", "-d", self.test_path, "--dry-run"
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line_dry_run)
mtime3 = os.path.getmtime(
os.path.join(self.test_path, 'outdir/output_file7.txt'))
# Assert first two modification times are different
# Assert second and third modification times same
self.assertTrue(mtime1 != mtime2)
self.assertTrue(mtime2 == mtime3)
self.assertEqual(se.exception.code, 0)
def test_run(self):
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"-v", "-d", self.test_path
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertTrue(
os.path.exists(
os.path.join(self.test_path, 'outdir/output_file1.txt')))
self.assertTrue(
os.path.exists(
os.path.join(self.test_path, 'outdir/output_file2.txt')))
self.assertTrue(
os.path.exists(
os.path.join(self.test_path, 'outdir/output_file7.txt')))
def test_05run_sourcerule_succeed(self):
p = subprocess.Popen(["touch", "test/output/output_File1.txt"])
p.wait()
p = subprocess.Popen(["touch", "test/output/output_File2.txt"])
p.wait()
p = subprocess.Popen(["touch", "test/output/output_File3.txt"])
p.wait()
p = subprocess.Popen(["touch", "test/output/output_File4.txt"])
p.wait()
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__right_def_file, "-d",
PathFinder.get_module_path(), "-vv", "-p", "--sourcerule", "rule2"
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertEqual(se.exception.code, 0)
def test_dry_run(self):
cmd_line = [
"python", "--dry-run", "-D", self.__db_url, "-w",
self.__example_def_file1, "-v", "-d", self.test_path
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
# The tests is that these files do not exist
self.assertFalse(
os.path.exists(
os.path.join(self.test_path, 'outdir/output_file1.txt')))
self.assertFalse(
os.path.exists(
os.path.join(self.test_path, 'outdir/output_file2.txt')))
self.assertFalse(
os.path.exists(
os.path.join(self.test_path, 'outdir/output_file7.txt')))
self.assertEqual(se.exception.code, 0)
def test_example(self):
# pip.main(['install', '{}/.'.format(self.example_dir_path), '--upgrade']) # working in travis
subprocess.run([
sys.executable, '-m', 'pip', 'install',
'{}/.'.format(self.example_dir_path), '--upgrade'
])
cmd_args = [
None, "-v", "--database", self.db_url, "--wopfile", self.wopfile,
"--directory", self.working_directory
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_args)
self.assertEqual(se.exception.code, 0)
conn = sqlite3.connect(self.db)
cursor = conn.execute("SELECT * FROM 'Piece';")
row_list = sorted(cursor.fetchall())
self.assertEqual(20, len(row_list))
conn.close()
def test_run_sourcerule_succeed(self):
p = subprocess.Popen(
["touch",
os.path.join(self.test_path, "outdir/output_file1.txt")])
p.wait()
p = subprocess.Popen(
["touch",
os.path.join(self.test_path, "outdir/output_file2.txt")])
p.wait()
p = subprocess.Popen(
["touch",
os.path.join(self.test_path, "outdir/output_file3.txt")])
p.wait()
p = subprocess.Popen(
["touch",
os.path.join(self.test_path, "outdir/output_file4.txt")])
p.wait()
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"-d", self.test_path, "-v", "--since", "rule2"
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
self.assertEqual(se.exception.code, 0)
def test_run_touch(self):
"""This is a tests for the touch function"""
################################################################################################################
#
# Normal run until rule 2
#
################################################################################################################
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"-v", "-d", self.test_path, "-U", "rule2"
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
# self.assertTrue(os.path.exists(os.path.join(self.test_path, 'outdir/output_file1.txt')))
# self.assertEqual(se.exception.code, 0)
output_file1_mtime = os.path.getmtime(
os.path.join(self.test_path, 'outdir/output_file1.txt'))
foobase_mtime = SQLManager.instance().get_session().query(TableModificationTime)\
.filter_by(table_name='FooBase').first().mtime_epoch_millis
################################################################################################################
#
# Dry-run
#
################################################################################################################
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"-v", "-d", self.test_path, "-U", "rule2 --dry-run"
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
# self.assertTrue(os.path.exists(os.path.join(self.test_path, 'outdir/output_file1.txt')))
# self.assertEqual(se.exception.code, 0)
output_file1_mtime_after_dry = os.path.getmtime(
os.path.join(self.test_path, 'outdir/output_file1.txt'))
foobase_mtime_after_dry = SQLManager.instance().get_session().query(TableModificationTime)\
.filter_by(table_name='FooBase').first().mtime_epoch_millis
################################################################################################################
#
# Touch output files
#
################################################################################################################
cmd_line = [
"python", "-D", self.__db_url, "-w", self.__example_def_file1,
"-v", "-d", self.test_path, "-U", "rule2", "--touch"
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
output_file1_mtime_after_touch = os.path.getmtime(
os.path.join(self.test_path, 'outdir/output_file1.txt'))
foobase_mtime_after_touch = SQLManager.instance().get_session().query(TableModificationTime)\
.filter_by(table_name='FooBase').first().mtime_epoch_millis
################################################################################################################
#
# Assert that modification times have been updated
#
################################################################################################################
self.assertEqual(output_file1_mtime, output_file1_mtime_after_dry)
self.assertEqual(foobase_mtime, foobase_mtime_after_dry)
self.assertLess(output_file1_mtime, output_file1_mtime_after_touch)
self.assertLess(foobase_mtime, foobase_mtime_after_touch)
def test_db_forceall(self):
"""Tests the forceall option in the database. Will carry out different runs in either default or forceall mode
that will either increase or keep same the number of rows of the FooBase table"""
pathlib.Path(os.path.join(self.test_path,
"outdir/output_file1.txt")).touch()
################################################################################################################
#
# Run 1: normal mode. Results in 1000 rows
#
################################################################################################################
cmd_line = [
"python", "-D", self.__db_url, "-w",
self.__example_def_file1_only_database, "-v", "-d", self.test_path
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
foobase_model = Base.metadata.tables['FooBase']
row_count = SQLManager.instance().get_session().query(
foobase_model).count()
self.assertTrue(row_count == 1000)
self.assertEqual(se.exception.code, 0)
################################################################################################################
#
# Run 2: normal mode. Still 1000 rows, because it does not execute
#
################################################################################################################
cmd_line = [
"python", "-D", self.__db_url, "-w",
self.__example_def_file1_only_database, "-v", "-d", self.test_path
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
row_count = SQLManager.instance().get_session().query(
foobase_model).count()
self.assertTrue(row_count == 1000)
self.assertEqual(se.exception.code, 0)
################################################################################################################
#
# Run 3: forceall mode. Now 2000 rows, because it is forced to execute
#
################################################################################################################
cmd_line = [
"python", "-D", self.__db_url, "-w",
self.__example_def_file1_only_database, "-v", "-d", self.test_path,
"--forceall"
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
row_count = SQLManager.instance().get_session().query(
foobase_model).count()
self.assertTrue(row_count == 2000)
self.assertEqual(se.exception.code, 0)
################################################################################################################
#
# Run 4: normal mode. Again 2000 rows, because it does not execute
#
################################################################################################################
cmd_line = [
"python", "-D", self.__db_url, "-w",
self.__example_def_file1_only_database, "-v", "-d", self.test_path
]
with self.assertRaises(SystemExit) as se:
WopMars().run(cmd_line)
row_count = SQLManager.instance().get_session().query(
foobase_model).count()
self.assertTrue(row_count == 2000)
self.assertEqual(se.exception.code, 0)
