Fix to_config_dict implementation.

enthought · Jun 19, 2015 · 8e30699 · 8e30699
1 parent d33b84a
commit 8e30699
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/enstaller/auth/_impl.py b/enstaller/auth/_impl.py
@@ -193,7 +193,7 @@ def change_auth(self, filename):
             fp.write(data)
 
     def to_config_dict(self):
-        return {"kind": "token", "api_token": self.api_token}
+        return {"kind": "token", "api_token": self._api_token}
 
     @property
     def cant_login_message(self):

diff --git a/enstaller/auth/tests/test__impl.py b/enstaller/auth/tests/test__impl.py
@@ -232,3 +232,11 @@ def test_cant_login_message(self):
 
         # Then
         self.assertEqual(auth.cant_login_message, r_message)
+
+    def test_to_config_dict(self):
+        # When
+        auth = APITokenAuth("bad robot")
+
+        # Then
+        self.assertEqual(auth.to_config_dict(),
+                         {"kind": "token", "api_token": "bad robot"})