Add public flag in Program (#1181)

* add RuntimeJob table

Signed-off-by: Akihiko Kuroda <akihikokuroda2020@gmail.com>

gateway/api/migrations/0015_program_public.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.2 on 2024-01-21 14:30
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("api", "0014_runtimejob"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="program",
+            name="public",
+            field=models.BooleanField(default=False),
+        ),
+    ]

gateway/api/models.py

@@ -69,6 +69,7 @@ class Program(ExportModelOperationsMixin("program"), models.Model):
         settings.AUTH_USER_MODEL,
         on_delete=models.CASCADE,
     )
+    public = models.BooleanField(null=False, default=False)
 
     arguments = models.TextField(null=False, blank=True, default="{}")
     env_vars = models.TextField(null=False, blank=True, default="{}")

gateway/api/v1/serializers.py

@@ -17,6 +17,7 @@ class Meta(serializers.ProgramSerializer.Meta):
             "artifact",
             "dependencies",
             "arguments",
+            "public",
         ]
 
 

gateway/tests/api/test_v1_program.py

@@ -81,3 +81,37 @@ def test_run_existing(self):
         self.assertEqual(job.config.max_workers, 5)
         self.assertEqual(job.config.workers, None)
         self.assertEqual(job.config.auto_scaling, True)
+
+    def test_public(self):
+        """Tests public flag."""
+        auth = reverse("rest_login")
+        response = self.client.post(
+            auth, {"username": "test_user", "password": "123"}, format="json"
+        )
+        token = response.data.get("access")
+        self.client.credentials(HTTP_AUTHORIZATION="Bearer " + token)
+
+        programs_response = self.client.get(
+            "/api/v1/programs/1a7947f9-6ae8-4e3d-ac1e-e7d608deec82/",
+            format="json",
+        )
+        print(programs_response.json())
+        public = programs_response.json()["public"]
+        self.assertEqual(public, False)
+
+        programs_response = self.client.patch(
+            "/api/v1/programs/1a7947f9-6ae8-4e3d-ac1e-e7d608deec82/",
+            data={
+                "public": True,
+            },
+            format="json",
+        )
+        self.assertEqual(programs_response.status_code, status.HTTP_200_OK)
+
+        programs_response = self.client.get(
+            "/api/v1/programs/1a7947f9-6ae8-4e3d-ac1e-e7d608deec82/",
+            format="json",
+        )
+        print(programs_response.json())
+        public = programs_response.json()["public"]
+        self.assertEqual(public, True)