feat(operations): job, claim and worker lease models

operations/tests/test_models.py

@@ -0,0 +1,47 @@
+"""The job record: states, defaults, the per-user submission key, and the
+SQLite connection mode every process relies on."""
+
+from django.contrib.auth import get_user_model
+from django.db import IntegrityError, connection, transaction
+from django.test import TestCase
+
+from operations.models import Claim, Operation
+
+
+def new_op(user, key="k", **kw):
+    return Operation.objects.create(kind="test.noop", user=user, username=user.username, submission_key=key,
+                                    resource="instance:u1", resource_label="c/vm", **kw)
+
+
+class OperationModelTestCase(TestCase):
+    def setUp(self):
+        self.user = get_user_model().objects.create_user("jobs_user", "j@example.com", "pw")
+
+    def test_a_new_operation_is_queued_with_empty_records(self):
+        op = new_op(self.user)
+        op.refresh_from_db()
+        self.assertEqual(op.state, Operation.QUEUED)
+        self.assertEqual((op.params, op.intent, op.result), ({}, {}, {}))
+        self.assertIn(Operation.QUEUED, Operation.ACTIVE)
+        self.assertIn(Operation.UNCERTAIN, Operation.TERMINAL)
+
+    def test_a_submission_key_is_unique_per_user(self):
+        new_op(self.user)
+        with self.assertRaises(IntegrityError), transaction.atomic():
+            new_op(self.user)
+
+    def test_claims_go_with_their_operation(self):
+        op = new_op(self.user)
+        Claim.objects.create(operation=op, resource="instance:u1", mode=Claim.EXCLUSIVE)
+        op.delete()
+        self.assertFalse(Claim.objects.exists())
+
+
+class SqliteModeTestCase(TestCase):
+    def test_sqlite_runs_in_wal_with_immediate_transactions(self):
+        if connection.vendor != "sqlite":
+            self.skipTest("SQLite only")
+        with connection.cursor() as cursor:
+            cursor.execute("PRAGMA journal_mode")
+            self.assertEqual(cursor.fetchone()[0].lower(), "wal")
+        self.assertEqual(connection.transaction_mode, "IMMEDIATE")