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()
