def test_run_script_does_not_delete_workunit(self):
conn = hpycc.Connection("user", test_conn=False)
check_wu_ecl = "ecl getwuid -u user -pw 1234 -s localhost -n test_run_script_deletes_workunit"
good_script = ("#WORKUNIT('name','test_run_script_deletes_workunit');"
"OUTPUT(2);")
with TemporaryDirectory() as d:
p = os.path.join(d, "test.ecl")
with open(p, "w+") as file:
file.write(good_script)
hpycc.run_script(conn, p, delete_workunit=False)
res = conn._run_command(check_wu_ecl)
self.assertRegex(res.stdout, "W[0-9]{8}-[0-9]{6}")
def test_run_script_does_nothing_when_not_using_stored(self):
conn = hpycc.Connection("user", test_conn=False)
file_name = "test_run_script_does_nothing_when_not_using_stored_"
good_script = "str := 'abc' : STORED('str');" \
" z := DATASET([{{str + str}}]," \
" {{STRING str;}}); OUTPUT(z,,'~{}', " \
"EXPIRE(1));".format(file_name)
with TemporaryDirectory() as d:
p = os.path.join(d, "test.ecl")
with open(p, "w+") as file:
file.write(good_script)
hpycc.run_script(conn, p, stored={'b': 'Shouldbethecorrectoutput'})
res = hpycc.get_thor_file(conn, file_name)
expected = pd.DataFrame({"str": ["abcabc"], "__fileposition__": [0]})
pd.testing.assert_frame_equal(expected.sort_index(axis=1), res.sort_index(axis=1), check_dtype=False)
def test_run_script_returns_true(self):
conn = hpycc.Connection("user", test_conn=False)
good_script = "OUTPUT(2);"
with TemporaryDirectory() as d:
p = os.path.join(d, "test.ecl")
with open(p, "w+") as file:
file.write(good_script)
res = hpycc.run_script(conn, p, delete_workunit=False)
self.assertTrue(res)
def test_run_script_uses_default_parameters(self, mock):
conn = Connection("user", test_conn=False)
run_script(conn, "abc.ecl")
mock.assert_called_with("abc.ecl", True, True, None)
def test_run_script_uses_custom_parameters(self, mock):
conn = Connection("user", test_conn=False)
run_script(conn, "abc.ecl", False, False, {'a': 'Testing'})
mock.assert_called_with("abc.ecl", False, False, {'a': 'Testing'})