def test_param_a_to_script(stc): stc_sys = CStcSystem.Instance() project = stc_sys.GetObject('Project') sequencer = stc_sys.GetObject("Sequencer") ctor = CScriptableCreator() pkg = "spirent.methodology" tags = project.GetObject('Tags') tagA = ctor.Create("tag", tags) tagA.Set("Name", "tagA") cmd = ctor.Create(pkg + ".RunPyScriptCommand", sequencer) RunPyScriptCommand.get_this_cmd = MagicMock(return_value=cmd) cmd2 = ctor.Create(pkg + ".LoadTemplateCommand", sequencer) cmd2.AddObject(tagA, RelationType("UserTag")) script_folder = mmutils.get_scripts_home_dir() filename = os.path.join(script_folder, "unit_test_file_param_a_to_script.py") try: if not os.path.exists(script_folder): os.makedirs(script_folder) with open(filename, "w") as f: f.write(unit_test_file_param_a_to_script()) RunPyScriptCommand.run("unit_test_file_param_a_to_script", "run", "tagA", {'a': 4}) assert cmd2.Get("CopiesPerParent") == 4 finally: remove_files(filename) return
def hold_test_error_returned(stc): stc_sys = CStcSystem.Instance() project = stc_sys.GetObject('Project') sequencer = stc_sys.GetObject("Sequencer") ctor = CScriptableCreator() pkg = "spirent.methodology" tags = project.GetObject('Tags') tagA = ctor.Create("tag", tags) tagA.Set("Name", "tagA") cmd = ctor.Create(pkg + ".RunPyScriptCommand", sequencer) RunPyScriptCommand.get_this_cmd = MagicMock(return_value=cmd) script_folder = mmutils.get_scripts_home_dir() filename = os.path.join(script_folder, "unit_test_file_error_returned.py") try: if not os.path.exists(script_folder): os.makedirs(script_folder) with open(filename, "w") as f: f.write(unit_test_file_error_returned()) rc = RunPyScriptCommand.run("unit_test_file_error_returned", "errorout", "tagA", {}) assert rc is False assert cmd.Get("ErrorMsg") != "" finally: remove_files(filename) return
def test_validate_missing_file(): stc_sys = CStcSystem.Instance() sequencer = stc_sys.GetObject("Sequencer") ctor = CScriptableCreator() pkg = "spirent.methodology" cmd = ctor.Create(pkg + ".RunPyScriptCommand", sequencer) RunPyScriptCommand.get_this_cmd = MagicMock(return_value=cmd) err = RunPyScriptCommand.validate("unit_test_file_missing_script", "run", "tagA", "") assert err != ""
def test_validate_valid_file(): stc_sys = CStcSystem.Instance() sequencer = stc_sys.GetObject("Sequencer") ctor = CScriptableCreator() pkg = "spirent.methodology" cmd = ctor.Create(pkg + ".RunPyScriptCommand", sequencer) RunPyScriptCommand.get_this_cmd = MagicMock(return_value=cmd) script_folder = mmutils.get_scripts_home_dir() filename = os.path.join(script_folder, "unit_test_file_valid_script.py") try: if not os.path.exists(script_folder): os.makedirs(script_folder) with open(filename, "w") as f: f.write(unit_test_file_valid_script()) err = RunPyScriptCommand.validate("unit_test_file_valid_script", "run", "tagA", "") assert err == "" err = RunPyScriptCommand.validate("unit_test_file_valid_script.py", "run", "tagA", "") assert err == "" finally: remove_files(filename) return