def test_validate_cwd_default(self): initial_cwd = os.getcwd() config["agent_chdir"] = None data = CommandData("") data.validate() self.assertEqual(data.cwd, os.getcwd()) self.assertEqual(initial_cwd, os.getcwd())
def test_validate_cwd_config(self): initial_cwd = os.getcwd() testdir, _ = self.create_directory(count=0) config["agent_chdir"] = testdir data = CommandData("") data.validate() self.assertEqual(data.cwd, testdir) self.assertEqual(initial_cwd, os.getcwd())
def test_set_default_environment(self): data = CommandData("", env=None) data.set_default_environment({"a": "b"}) self.assertEqual(data.env, {"a": "b"})
def test_set_default_environment_noop(self): data = CommandData("", env={"foo": "bar"}) data.set_default_environment({"a": "b"}) self.assertEqual(data.env, {"foo": "bar"})
def test_validate_cwd_invalid_type(self): data = CommandData("", cwd=1) with self.assertRaises(TypeError): data.validate()
def test_validate_cwd_does_not_exist(self): data = CommandData("", cwd=os.urandom(4).encode("hex")) with self.assertRaises(OSError): data.validate()