def test_run(self): create_item = CommandItem("create_item", 'echo hello>test.txt') if platform.system().lower() == "windows": delete_item = CommandItem("delete_item", "del test.txt") expected_contents = "hello \n" else: delete_item = CommandItem("delete_item", "rm test.txt") expected_contents = "hello\n" create_item.action() self.assertEqual(create_item.get_return(), 0) self.assertTrue(os.path.isfile("test.txt")) with open("test.txt", 'r') as text: self.assertEqual(text.read(), expected_contents) delete_item.action() self.assertEqual(delete_item.get_return(), 0) self.assertFalse(os.path.isfile("test.txt"))
def test_return(self): if platform.system().lower() == "windows": return_command_item = CommandItem("return_command_item", "exit 1") else: return_command_item = CommandItem("return_command_item", "return 1") return_command_item.action() self.assertEqual(return_command_item.get_return(), 1)