Minor PR to organize migrations (#1851)

berinhard · web-flow · commit b19497d8c424 · 2021-08-27T08:10:27.000-04:00
* Add missing migration from contract updates

* Remove created file so tests can be idempotent
diff --git a/sponsors/migrations/0035_auto_20210826_1929.py b/sponsors/migrations/0035_auto_20210826_1929.py
@@ -0,0 +1,29 @@
+# Generated by Django 2.0.13 on 2021-08-26 19:29
+
+from django.db import migrations, models
+import sponsors.models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('sponsors', '0034_contract_document_docx'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='contract',
+            name='document',
+            field=models.FileField(blank=True, upload_to='sponsors/contracts/', verbose_name='Unsigned PDF'),
+        ),
+        migrations.AlterField(
+            model_name='contract',
+            name='document_docx',
+            field=models.FileField(blank=True, upload_to='sponsors/contracts/docx/', verbose_name='Unsigned Docx'),
+        ),
+        migrations.AlterField(
+            model_name='contract',
+            name='signed_document',
+            field=models.FileField(blank=True, upload_to=sponsors.models.signed_contract_random_path, verbose_name='Signed PDF'),
+        ),
+    ]
diff --git a/sponsors/tests/test_use_cases.py b/sponsors/tests/test_use_cases.py
@@ -1,6 +1,8 @@
+import os
 from unittest.mock import Mock, patch
 from model_bakery import baker
 from datetime import timedelta, date
+from pathlib import Path
 
 from django.conf import settings
 from django.test import TestCase
@@ -186,6 +188,14 @@ def setUp(self):
         self.file = SimpleUploadedFile("contract.txt", b"Contract content")
         self.contract = baker.make_recipe("sponsors.tests.empty_contract", status=Contract.DRAFT)
 
+    def tearDown(self):
+        try:
+            signed_file = Path(self.contract.signed_document.path)
+            if signed_file.exists():
+                os.remove(str(signed_file.resolve()))
+        except ValueError:
+            pass
+
     @patch("sponsors.models.uuid.uuid4", Mock(return_value="1234"))
     def test_execute_and_update_database_object(self):
         self.use_case.execute(self.contract, self.file)
