dockercloud

tests/test_auth.py

@@ -0,0 +1,89 @@
+from __future__ import absolute_import
+
+import os
+import tempfile
+import unittest
+
+import unittest.mock as mock
+
+import dockercloud
+from .fake_api import *
+
+
+class AuthTestCase(unittest.TestCase):
+    def setUp(self):
+        self.dockercloud_auth = dockercloud.dockercloud_auth
+        self.basic_auth = dockercloud.basic_auth
+
+    def tearDown(self):
+        dockercloud.dockercloud_auth = self.dockercloud_auth
+        dockercloud.basic_auth = self.basic_auth
+
+    @mock.patch('dockercloud.api.auth.verify_credential')
+    def test_auth_authenticate(self, mock_verify_credential):
+        dockercloud.auth.authenticate(FAKE_USER, FAKE_PASSWORD)
+        mock_verify_credential.assert_called_with(FAKE_USER, FAKE_PASSWORD)
+        self.assertEqual(dockercloud.basic_auth, FAKE_BASIC_AUTH)
+        self.tearDown()
+
+    def test_auth_is_authenticated(self):
+        dockercloud.dockercloud_auth = FAKE_DOCKERCLOUD_AUTH
+        dockercloud.basic_auth = FAKE_BASIC_AUTH
+        self.assertTrue(dockercloud.auth.is_authenticated())
+
+        dockercloud.dockercloud_auth = None
+        dockercloud.basic_auth = FAKE_BASIC_AUTH
+        self.assertTrue(dockercloud.auth.is_authenticated())
+
+        dockercloud.dockercloud_auth = FAKE_DOCKERCLOUD_AUTH
+        dockercloud.basic_auth = None
+        dockercloud.apikey_auth = None
+        self.assertTrue(dockercloud.auth.is_authenticated())
+
+        dockercloud.dockercloud_auth = None
+        dockercloud.basic_auth = None
+        self.assertFalse(dockercloud.auth.is_authenticated())
+
+    def test_auth_logout(self):
+        dockercloud.dockercloud_auth = FAKE_DOCKERCLOUD_AUTH
+        dockercloud.basic_auth = FAKE_BASIC_AUTH
+        dockercloud.auth.logout()
+        self.assertIsNone(dockercloud.dockercloud_auth)
+        self.assertIsNone(dockercloud.basic_auth)
+
+    def test_auth_load_from_file(self):
+        temp = tempfile.NamedTemporaryFile('w', delete=False)
+        with temp as f:
+            f.write('''{
+	"auths": {
+		"https://index.docker.io/v1/": {
+			"auth": "%s",
+			"email": "tifayuki@gmail.com"
+		}
+	}
+}''' % FAKE_BASIC_AUTH)
+        basic_auth = dockercloud.auth.load_from_file(f.name)
+        self.assertEqual(basic_auth, FAKE_BASIC_AUTH)
+        os.remove(temp.name)
+
+    def test_auth_load_from_file_with_exception(self):
+        basic_auth = dockercloud.auth.load_from_file('abc')
+        self.assertIsNone(basic_auth)
+
+    def test_auth_get_auth_header(self):
+        dockercloud.dockercloud_auth = FAKE_DOCKERCLOUD_AUTH
+        dockercloud.basic_auth = FAKE_BASIC_AUTH
+        self.assertEqual({'Authorization': FAKE_DOCKERCLOUD_AUTH}, dockercloud.auth.get_auth_header())
+
+        print "===================="
+        dockercloud.dockercloud_auth = None
+        dockercloud.basic_auth = FAKE_BASIC_AUTH
+        self.assertEqual({'Authorization': 'Basic %s' % (FAKE_BASIC_AUTH)}, dockercloud.auth.get_auth_header())
+
+        dockercloud.dockercloud_auth = FAKE_DOCKERCLOUD_AUTH
+        dockercloud.basic_auth = None
+        self.assertEqual({'Authorization': FAKE_DOCKERCLOUD_AUTH}, dockercloud.auth.get_auth_header())
+
+        dockercloud.dockercloud_auth = None
+        dockercloud.basic_auth = None
+        self.assertEqual({}, dockercloud.auth.get_auth_header())