tech: better test_cmd

Author: stephenprn

core/management/commands/test_cmd.py

@@ -1,7 +1,11 @@
+import json
+from time import sleep
 from django.core.management.base import BaseCommand
 
 from core.utils.logs_helpers import log_command_event
 
+from django.db import connection
+
 
 def log_event(info: str):
     log_command_event(command_name="test_cmd", info=info)
@@ -32,6 +36,22 @@ def handle(self, *args, **options):
         test_int_not_required = options["test_int_not_required"]
         test_array = options["test_array"]
 
+        log_event("writing args in temp schema")
+
+        with connection.cursor() as cursor:
+            test_table = "test_cmd"
+            cursor.execute(
+                f"CREATE TABLE IF NOT EXISTS public.{test_table} (id SERIAL PRIMARY KEY, created_at TIMESTAMP DEFAULT NOW(), args JSONB);"
+            )
+            cursor.execute(
+                f"INSERT INTO public.{test_table} (data) VALUES ('{json.dumps(options)}');"
+            )
+
+        waiting_sec = 120
+        log_event(f"waiting for {waiting_sec} seconds")
+        sleep(waiting_sec)
+        log_event(f"waited for {waiting_sec} seconds")
+
         log_event("args retrieved")
 
         with open("test_cmd.txt", "w") as f: