test(controller): mock import_repository task

Author: Matthew Fisher

controller/api/tasks.py

@@ -38,8 +38,7 @@ def deploy_release(app, release):
 
 @task
 def import_repository(source, target_repository):
-    """Imports an image from a remote into our own private registry"""
-
+    """Imports an image from a remote registry into our own private registry"""
     data = {
         'src': source,
     }

controller/api/tests/test_build.py

@@ -7,13 +7,22 @@
 from __future__ import unicode_literals
 
 import json
+import mock
+import requests
 
 from django.test import TransactionTestCase
 from django.test.utils import override_settings
 
 from api.models import Build
 
 
+def mock_import_repository_task(*args, **kwargs):
+    resp = requests.Response()
+    resp.status_code = 200
+    resp._content_consumed = True
+    return resp
+
+
 @override_settings(CELERY_ALWAYS_EAGER=True)
 class BuildTest(TransactionTestCase):
 
@@ -30,6 +39,7 @@ def setUp(self):
                                     content_type='application/json')
         self.assertEqual(response.status_code, 201)
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_build(self):
         """
         Test that a null build is created and that users can post new builds
@@ -71,6 +81,7 @@ def test_build(self):
         self.assertEqual(self.client.patch(url).status_code, 405)
         self.assertEqual(self.client.delete(url).status_code, 405)
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_build_default_containers(self):
         url = '/api/apps'
         body = {'cluster': 'autotest'}
@@ -152,6 +163,7 @@ def test_build_default_containers(self):
         self.assertEqual(container['type'], 'web')
         self.assertEqual(container['num'], 1)
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_build_str(self):
         """Test the text representation of a build."""
         url = '/api/apps'

controller/api/tests/test_config.py

@@ -7,13 +7,22 @@
 from __future__ import unicode_literals
 
 import json
+import mock
+import requests
 
 from django.test import TransactionTestCase
 from django.test.utils import override_settings
 
 from api.models import Config
 
 
+def mock_import_repository_task(*args, **kwargs):
+    resp = requests.Response()
+    resp.status_code = 200
+    resp._content_consumed = True
+    return resp
+
+
 @override_settings(CELERY_ALWAYS_EAGER=True)
 class ConfigTest(TransactionTestCase):
 
@@ -30,6 +39,7 @@ def setUp(self):
                                     content_type='application/json')
         self.assertEqual(response.status_code, 201)
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_config(self):
         """
         Test that config is auto-created for a new app and that
@@ -94,6 +104,7 @@ def test_config(self):
         self.assertEqual(self.client.delete(url).status_code, 405)
         return config5
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_config_set_same_key(self):
         """
         Test that config sets on the same key function properly
@@ -116,6 +127,7 @@ def test_config_set_same_key(self):
         self.assertIn('PORT', json.loads(response.data['values']))
         self.assertEqual(json.loads(response.data['values'])['PORT'], '5001')
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_config_str(self):
         """Test the text representation of a node."""
         config5 = self.test_config()

controller/api/tests/test_container.py

@@ -7,6 +7,8 @@
 from __future__ import unicode_literals
 
 import json
+import mock
+import requests
 
 from django.contrib.auth.models import User
 from django.test import TransactionTestCase
@@ -17,9 +19,15 @@
 from api.models import Container, App
 
 
+def mock_import_repository_task(*args, **kwargs):
+    resp = requests.Response()
+    resp.status_code = 200
+    resp._content_consumed = True
+    return resp
+
+
 @override_settings(CELERY_ALWAYS_EAGER=True)
 class ContainerTest(TransactionTestCase):
-
     """Tests creation of containers on nodes"""
 
     fixtures = ['tests.json']
@@ -174,6 +182,7 @@ def test_container_api_heroku(self):
         response = self.client.get(url)
         self.assertEqual(response.status_code, 200)
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_container_api_docker(self):
         url = '/api/apps'
         body = {'cluster': 'autotest'}
@@ -233,6 +242,7 @@ def test_container_api_docker(self):
         response = self.client.get(url)
         self.assertEqual(response.status_code, 200)
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_container_release(self):
         url = '/api/apps'
         body = {'cluster': 'autotest'}

controller/api/tests/test_hooks.py

@@ -7,13 +7,22 @@
 from __future__ import unicode_literals
 
 import json
+import mock
+import requests
 
 from django.test import TransactionTestCase
 from django.test.utils import override_settings
 
 from django.conf import settings
 
 
+def mock_import_repository_task(*args, **kwargs):
+    resp = requests.Response()
+    resp.status_code = 200
+    resp._content_consumed = True
+    return resp
+
+
 @override_settings(CELERY_ALWAYS_EAGER=True)
 class HookTest(TransactionTestCase):
 
@@ -95,6 +104,7 @@ def test_push_abuse(self):
                                     HTTP_X_DEIS_BUILDER_AUTH=settings.BUILDER_KEY)
         self.assertEqual(response.status_code, 403)
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_build_hook(self):
         """Test creating a Build via an API Hook"""
         url = '/api/apps'

controller/api/tests/test_release.py

@@ -7,13 +7,22 @@
 from __future__ import unicode_literals
 
 import json
+import mock
+import requests
 
 from django.test import TransactionTestCase
 from django.test.utils import override_settings
 
 from api.models import Release
 
 
+def mock_import_repository_task(*args, **kwargs):
+    resp = requests.Response()
+    resp.status_code = 200
+    resp._content_consumed = True
+    return resp
+
+
 @override_settings(CELERY_ALWAYS_EAGER=True)
 class ReleaseTest(TransactionTestCase):
 
@@ -30,6 +39,7 @@ def setUp(self):
                                     content_type='application/json')
         self.assertEqual(response.status_code, 201)
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_release(self):
         """
         Test that a release is created when a cluster is created, and
@@ -101,6 +111,7 @@ def test_release(self):
         self.assertEqual(self.client.delete(url).status_code, 405)
         return release3
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_release_rollback(self):
         url = '/api/apps'
         body = {'cluster': 'autotest'}
@@ -187,12 +198,14 @@ def test_release_rollback(self):
         self.assertIn('NEW_URL1', values)
         self.assertEqual('http://localhost:8080/', values['NEW_URL1'])
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_release_str(self):
         """Test the text representation of a release."""
         release3 = self.test_release()
         release = Release.objects.get(uuid=release3['uuid'])
         self.assertEqual(str(release), "{}-v3".format(release3['app']))
 
+    @mock.patch('requests.post', mock_import_repository_task)
     def test_release_summary(self):
         """Test the text summary of a release."""
         release3 = self.test_release()

controller/dev_requirements.txt

@@ -26,3 +26,6 @@ pyflakes==0.7.3
 
 # Python expect is used to automate client-driven tests
 pexpect>=3.1
+
+# Used for mocking endpoints
+mock==1.0.1