From: Alfredo Deza <alfredo@deza.pe>
Date: Tue, 10 Jul 2018 17:02:29 +0000 (-0400)
Subject: tests: ensure we can extend the remote env
X-Git-Tag: v0.0.31~3
X-Git-Url: http://git-server-git.apps.pok.os.sepia.ceph.com/?a=commitdiff_plain;h=4a35163e8787de99e827bb64222d5b0e59c8cad5;p=remoto.git

tests: ensure we can extend the remote env

Signed-off-by: Alfredo Deza <alfredo@deza.pe>
---

diff --git a/remoto/tests/test_process.py b/remoto/tests/test_process.py
index 3f53344..4588f5c 100644
--- a/remoto/tests/test_process.py
+++ b/remoto/tests/test_process.py
@@ -11,14 +11,23 @@ class TestExtendPath(object):
         fake_conn = Mock()
         fake_conn.gateway.remote_exec.return_value = fake_conn
         fake_conn.receive.return_value = {}
-        result = process.extend_path(fake_conn, {})
+        result = process.extend_env(fake_conn, {})
         assert result['env']['PATH'] == self.path
 
     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, {})
+        result = process.extend_env(fake_conn, {})
         new_path = result['env']['PATH']
         assert new_path.endswith(self.path)
         assert '/home/alfredo/bin' in new_path
+
+    def test_custom_env_var_extends_existing_env(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_env(fake_conn, {'extend_env': {'CEPH_VOLUME_DEBUG': '1'}})
+        new_path = result['env']['PATH']
+        assert result['env']['PATH'].endswith(self.path)
+        assert result['env']['CEPH_VOLUME_DEBUG'] == '1'