test(controller): add more container state tests

This commit tests more of the finite state machine's functionality
that was introduced in #699, including disabled direct modification
of the FSM and most of the expected state transitions.

closes #700

Author: Matthew Fisher

controller/api/tests/test_container.py

@@ -12,6 +12,8 @@
 from django.test import TransactionTestCase
 from django.test.utils import override_settings
 
+from django_fsm import TransitionNotAllowed
+
 from api.models import Container, App
 
 
@@ -30,8 +32,15 @@ def setUp(self):
         response = self.client.post('/api/clusters', json.dumps(body),
                                     content_type='application/json')
         self.assertEqual(response.status_code, 201)
+        # create a malicious scheduler as well
+        body['id'] = 'autotest2'
+        body['type'] = 'faulty'
+        response = self.client.post('/api/clusters', json.dumps(body),
+                                    content_type='application/json')
+        self.assertEqual(response.status_code, 201)
 
-    def test_container(self):
+    def test_container_state_good(self):
+        """Test that the finite state machine transitions with a good scheduler"""
         url = '/api/apps'
         body = {'cluster': 'autotest'}
         response = self.client.post(url, json.dumps(body), content_type='application/json')
@@ -44,13 +53,57 @@ def test_container(self):
                                      type='web',
                                      num=1)
         self.assertEqual(c.state, 'initialized')
+        # test an illegal transition
+        self.assertRaises(TransitionNotAllowed, lambda: c.start())
         c.create()
         self.assertEqual(c.state, 'created')
         c.start()
         self.assertEqual(c.state, 'up')
+        c.deploy(App.objects.get(id=app_id).release_set.latest())
+        self.assertEqual(c.state, 'up')
         c.destroy()
         self.assertEqual(c.state, 'destroyed')
 
+    def test_container_state_bad(self):
+        """Test that the finite state machine transitions with a faulty scheduler"""
+        url = '/api/apps'
+        body = {'cluster': 'autotest2'}
+        response = self.client.post(url, json.dumps(body), content_type='application/json')
+        self.assertEqual(response.status_code, 201)
+        app_id = response.data['id']
+        # create a container
+        c = Container.objects.create(owner=User.objects.get(username='autotest'),
+                                     app=App.objects.get(id=app_id),
+                                     release=App.objects.get(id=app_id).release_set.latest(),
+                                     type='web',
+                                     num=1)
+        self.assertEqual(c.state, 'initialized')
+        self.assertRaises(Exception, lambda: c.create())
+        self.assertEqual(c.state, 'initialized')
+        # test an illegal transition
+        self.assertRaises(TransitionNotAllowed, lambda: c.start())
+        self.assertEqual(c.state, 'initialized')
+        self.assertRaises(Exception, lambda: c.deploy(App.objects.get(id=app_id).release_set.latest()))
+        self.assertEqual(c.state, 'down')
+        self.assertRaises(Exception, lambda: c.destroy())
+        self.assertEqual(c.state, 'down')
+        self.assertRaises(Exception, lambda: c.run('echo hello world'))
+        self.assertEqual(c.state, 'down')
+
+    def test_container_state_protected(self):
+        """Test that you cannot directly modify the state"""
+        url = '/api/apps'
+        body = {'cluster': 'autotest'}
+        response = self.client.post(url, json.dumps(body), content_type='application/json')
+        self.assertEqual(response.status_code, 201)
+        app_id = response.data['id']
+        c = Container.objects.create(owner=User.objects.get(username='autotest'),
+                                     app=App.objects.get(id=app_id),
+                                     release=App.objects.get(id=app_id).release_set.latest(),
+                                     type='web',
+                                     num=1)
+        self.assertRaises(AttributeError, lambda: setattr(c, 'state', 'up'))
+
     def test_container_api(self):
         url = '/api/apps'
         body = {'cluster': 'autotest'}