fix(limits): always set default

Author: duanhongyi

rootfs/api/models/app.py

@@ -738,19 +738,27 @@ def _scale_pods(self, scale_types, release, app_settings):
             self.log(err, logging.ERROR)
             raise ServiceUnavailable(err) from e
 
-    def _set_default_config(self, config=None, procfile_types=None):
-        procfile_types = self.procfile_types if procfile_types is None else procfile_types
+    def _set_default_limit(self, config, procfile_type):
+        if procfile_type not in config.limits:
+            plan = LimitPlan.get_default()
+            config.limits[procfile_type] = plan.id
+            config.save(update_fields=['limits'])
+        return config
+
+    def _set_default_config(self):
         plan = LimitPlan.get_default()
         limits = {PROCFILE_TYPE_WEB: plan.id, PROCFILE_TYPE_RUN: plan.id}
         try:
-            config = self.config_set.latest() if config is None else config
-            for procfile_type in procfile_types:
-                limits[procfile_type] = config.limits.get(procfile_type, plan.id)
-            if limits != config.limits:
-                config.limits = limits
-                config.save(update_fields=['limits'])
+            config = self.config_set.latest()
+            limits[PROCFILE_TYPE_WEB] = config.limits.get(PROCFILE_TYPE_WEB, plan.id)
+            limits[PROCFILE_TYPE_RUN] = config.limits.get(PROCFILE_TYPE_RUN, plan.id)
         except Config.DoesNotExist:
             config = Config.objects.create(owner=self.owner, app=self, limits=limits)
+        for procfile_type in self.procfile_types:
+            limits[procfile_type] = config.limits.get(procfile_type, plan.id)
+        if limits != config.limits:
+            config.limits = limits
+            config.save(update_fields=['limits'])
         return config
 
     def _create_default_ingress(self, target_port):
@@ -993,9 +1001,8 @@ def _gather_app_settings(self, release, app_settings, procfile_type, replicas, v
         """
 
         envs = self._build_env_vars(release)
-        config = release.config
         # Obtain a limit plan that must exist, if raise error here, it must be a bug
-        self._set_default_config(config, procfile_types=[procfile_type])
+        config = self._set_default_limit(release.config, procfile_type)
         limit_plan = LimitPlan.objects.get(id=config.limits.get(procfile_type))
 
         # see if the app config has deploy batch preference, otherwise use global

rootfs/api/tests/test_config.py

@@ -473,7 +473,6 @@ def test_unset_limits_error(self, mock_requests):
         # dockerfile + procfile worflow
         app = App.objects.get(id=app_id)
         user = User.objects.get(username='autotest')
-        build = Build.objects.create(owner=user, app=app, image="qwerty")
         build = Build.objects.create(
             owner=user,
             app=app,
@@ -545,3 +544,45 @@ def test_measure_config(self, *args, **kwargs):
         out = self.call_command()
         self.assertIn(out, "done\n")
         self.assertEqual(response.status_code, 201)
+
+    def test_set_config_limits_run(self, *args, **kwargs):
+        # create
+        app_id = self.create_app()
+        # dockerfile + procfile worflow
+        app = App.objects.get(id=app_id)
+        user = User.objects.get(username='autotest')
+        build = Build.objects.create(
+            owner=user,
+            app=app,
+            image="qwerty",
+            procfile={
+                'web': 'node server.js',
+                'worker': 'node worker.js'
+            },
+            dockerfile='foo',
+            sha='somereallylongsha'
+        )
+        # create an initial release
+        release = Release.objects.create(
+            version=3,
+            owner=user,
+            app=app,
+            config=app.config_set.latest(),
+            build=build
+        )
+        # deploy
+        app.pipeline(release)
+        body = {
+            'values': json.dumps({'PORT': 5000}),
+            'limits': {
+                  PROCFILE_TYPE_RUN: 'std1.large.c2m4',
+                  PROCFILE_TYPE_WEB: 'std1.large.c2m4',
+            },
+        }
+        url = f"/v2/apps/{app_id}/config"
+        response = self.client.post(url, body)
+        url = "/v2/apps/{app_id}/config".format(**locals())
+        response = self.client.get(url)
+        self.assertEqual(response.status_code, 200, response.data)
+        expect = {'run': 'std1.large.c2m4', 'web': 'std1.large.c2m4', 'worker': 'std1.large.c1m1'}
+        self.assertEqual(expect, response.json()["limits"], response.data)