plz fix

tdrs-backend/conftest.py

@@ -0,0 +1,6 @@
+"""Sets an env variable to indicate that we are running tests. This is used to run custom migrations during testing."""
+import os
+
+def pytest_sessionstart(session):
+    """Set PYTEST env variable to indicate that we are running tests."""
+    os.environ['PYTEST'] = 'True'

tdrs-backend/tdpservice/stts/migrations/0005_stt_stt_code.py

@@ -15,4 +15,9 @@ class Migration(migrations.Migration):
             name='stt_code',
             field=models.PositiveIntegerField(blank=True, null=True),
         ),
+        migrations.AddField(
+            model_name='stt',
+            name='ssp',
+            field=models.BooleanField(default=False),
+        ),
     ]

tdrs-backend/tdpservice/stts/migrations/0007_stt_ssp.py

@@ -1,17 +1,47 @@
-# Generated by Django 3.2.15 on 2022-11-02 16:39 edited by Cameron
+# Generated by Cameron
+import csv
+from pathlib import Path
+from ..models import STT
+import os
 
 from django.db import migrations, models
 
+
+def _update(path):
+    data_dir = Path(__file__).resolve().parent.parent  /"management" / "commands" / "data"
+
+    with open(data_dir / path) as csvfile:
+        reader = csv.DictReader(csvfile)
+
+        for row in reader:
+            stt = STT.objects.get(code=row["Code"])
+            stt.ssp = row["SSP"] == "0"
+            stt.save()
+
+def _update_stts(apps, schema_editor):
+    _update("states.csv")
+    _update("territories.csv")
+    _update("tribes.csv")
+
+
 class Migration(migrations.Migration):
 
     dependencies = [
         ('stts', '0006_alter_stt_filenames'),
     ]
 
-    operations = [
-        migrations.AddField(
-            model_name='stt',
-            name='ssp',
-            field=models.BooleanField(default=False),
-        ),
-    ]
+    is_pytest = os.environ.get('PYTEST')
+
+    if is_pytest:
+        operations = [
+            migrations.RunPython(_update_stts),
+        ]
+    else:
+        operations = [
+            migrations.AddField(
+                model_name='stt',
+                name='ssp',
+                field=models.BooleanField(default=False),
+            ),
+            migrations.RunPython(_update_stts),
+        ]