test(controller): add unit tests for ps:restart

Author: Matthew Fisher

controller/api/tests/test_container.py

@@ -636,3 +636,39 @@ def test_modified_procfile_from_build_removes_containers(self):
                                     HTTP_AUTHORIZATION='token {}'.format(self.token))
         self.assertEqual(response.status_code, 201)
         self.assertEqual(Container.objects.filter(type='web').count(), 0)
+
+    def test_restart_containers(self):
+        url = '/v1/apps'
+        response = self.client.post(url, HTTP_AUTHORIZATION='token {}'.format(self.token))
+        self.assertEqual(response.status_code, 201)
+        app_id = response.data['id']
+        # post a new build
+        build_url = "/v1/apps/{app_id}/builds".format(**locals())
+        body = {'image': 'autotest/example', 'sha': 'a'*40,
+                'procfile': json.dumps({'web': 'node server.js', 'worker': 'node worker.js'})}
+        response = self.client.post(build_url, json.dumps(body), content_type='application/json',
+                                    HTTP_AUTHORIZATION='token {}'.format(self.token))
+        url = "/v1/apps/{app_id}/scale".format(**locals())
+        body = {'web': 4, 'worker': 8}
+        response = self.client.post(url, json.dumps(body), content_type='application/json',
+                                    HTTP_AUTHORIZATION='token {}'.format(self.token))
+        self.assertEqual(response.status_code, 204)
+        container_set = App.objects.get(id=app_id).container_set.all()
+        # restart all containers
+        response = self.client.post('/v1/apps/{}/containers/restart'.format(app_id),
+                                    content_type='application/json',
+                                    HTTP_AUTHORIZATION='token {}'.format(self.token))
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(len(response.data), container_set.count())
+        # restart only the workers
+        response = self.client.post('/v1/apps/{}/containers/worker/restart'.format(app_id),
+                                    content_type='application/json',
+                                    HTTP_AUTHORIZATION='token {}'.format(self.token))
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(len(response.data), container_set.filter(type='worker').count())
+        # restart only web.2
+        response = self.client.post('/v1/apps/{}/containers/web/1/restart'.format(app_id),
+                                    content_type='application/json',
+                                    HTTP_AUTHORIZATION='token {}'.format(self.token))
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(len(response.data), container_set.filter(type='web', num=1).count())