def test_custom_path_does_not_get_overridden(self):
fake_conn = Mock()
fake_conn.gateway.remote_exec.return_value = fake_conn
fake_conn.receive.return_value = {'PATH': '/home/alfredo/bin'}
result = process.extend_path(fake_conn, {})
new_path = result['env']['PATH']
assert new_path.endswith(self.path)
assert '/home/alfredo/bin' in new_path
def test_no_environment_sets_path(self):
fake_conn = Mock()
fake_conn.gateway.remote_exec.return_value = fake_conn
fake_conn.receive.return_value = {}
result = process.extend_path(fake_conn, {})
assert result['env']['PATH'] == self.path