aa-structures 2.8.0__py3-none-any.whl → 2.9.1__py3-none-any.whl

structures/migrations/0005_add_notification_types.py

@@ -0,0 +1,135 @@
+# Generated by Django 4.0.10 on 2024-03-21 13:45
+
+import multiselectfield.db.fields
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("structures", "0004_improve_localization"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="webhook",
+            name="notification_types",
+            field=multiselectfield.db.fields.MultiSelectField(
+                choices=[
+                    ("OwnershipTransferred", "Upwell structure ownership transferred"),
+                    ("StructureAnchoring", "Upwell structure anchoring"),
+                    ("StructureDestroyed", "Upwell structure destroyed"),
+                    ("StructureFuelAlert", "Upwell structure fuel alert"),
+                    ("StructureJumpFuelAlert", "Upwell structure jump fuel alert"),
+                    ("StructureLostArmor", "Upwell structure lost armor"),
+                    ("StructureLostShields", "Upwell structure lost shields"),
+                    ("StructureOnline", "Upwell structure went online"),
+                    ("StructureRefueledExtra", "Upwell structure refueled"),
+                    (
+                        "StructuresReinforcementChanged",
+                        "Upwell structure reinforcement time changed",
+                    ),
+                    (
+                        "StructureServicesOffline",
+                        "Upwell structure services went offline",
+                    ),
+                    ("StructureUnanchoring", "Upwell structure unanchoring"),
+                    ("StructureUnderAttack", "Upwell structure is under attack"),
+                    ("StructureWentHighPower", "Upwell structure went high power"),
+                    ("StructureWentLowPower", "Upwell structure went low power"),
+                    ("OrbitalAttacked", "Customs office attacked"),
+                    ("OrbitalReinforced", "Customs office reinforced"),
+                    ("TowerAlertMsg", "Starbase attacked"),
+                    ("TowerResourceAlertMsg", "Starbase fuel alert"),
+                    ("TowerRefueledExtra", "Starbase refueled (BETA)"),
+                    ("TowerReinforcedExtra", "Starbase reinforced (BETA)"),
+                    (
+                        "MoonminingAutomaticFracture",
+                        "Moon mining automatic fracture triggered",
+                    ),
+                    (
+                        "MoonminingExtractionCancelled",
+                        "Moon mining extraction cancelled",
+                    ),
+                    ("MoonminingExtractionFinished", "Moon mining extraction finished"),
+                    ("MoonminingExtractionStarted", "Moon mining extraction started"),
+                    ("MoonminingLaserFired", "Moonmining laser fired"),
+                    ("SovAllClaimAquiredMsg", "Sovereignty claim acknowledgment"),
+                    ("SovAllClaimLostMsg", "Sovereignty lost"),
+                    ("AllAnchoringMsg", "Structure anchoring in alliance space"),
+                    ("EntosisCaptureStarted", "Sovereignty entosis capture started"),
+                    ("SovStructureDestroyed", "Sovereignty structure destroyed"),
+                    ("SovStructureReinforced", "Sovereignty structure reinforced"),
+                    (
+                        "SovCommandNodeEventStarted",
+                        "Sovereignty command node event started",
+                    ),
+                    ("AcceptedAlly", "War accepted ally"),
+                    (
+                        "AllWarCorpJoinedAllianceMsg",
+                        "Alliance war corporation joined alliance message",
+                    ),
+                    ("AllWarSurrenderMsg", "Alliance war surrender message"),
+                    ("AllyJoinedWarAggressorMsg", "War ally joined aggressor"),
+                    ("AllyJoinedWarAllyMsg", "War ally joined ally"),
+                    ("AllyJoinedWarDefenderMsg", "War ally joined defender"),
+                    ("CorpWarSurrenderMsg", "War party surrendered"),
+                    ("CorpBecameWarEligible", "War corporation became eligible"),
+                    ("CorpNoLongerWarEligible", "War corporation no longer eligible"),
+                    ("DeclareWar", "War declared"),
+                    ("WarHQRemovedFromSpace", "War HQ removed from space"),
+                    ("WarInvalid", "War invalid"),
+                    (
+                        "MercOfferedNegotiationMsg",
+                        "War mercenary offered negotiation message",
+                    ),
+                    ("MercOfferRetractedMsg", "War mercenary offer retracted message"),
+                    ("OfferedSurrender", "War offered surrender"),
+                    ("OfferedToAlly", "War offered to become ally"),
+                    ("WarAdopted", "War adopted"),
+                    ("WarDeclared", "War declared"),
+                    ("WarInherited", "War inherited"),
+                    ("WarRetractedByConcord", "War retracted by Concord"),
+                    ("WarSurrenderOfferMsg", "War surrender offered"),
+                    ("CharAppAcceptMsg", "Character joins corporation"),
+                    ("CorpAppInvitedMsg", "Character invited to join corporation"),
+                    ("CorpAppNewMsg", "Character submitted application"),
+                    ("CharAppRejectMsg", "Corp application rejected message"),
+                    (
+                        "CorpAppRejectCustomMsg",
+                        "Corp application rejected custom message",
+                    ),
+                    ("CharAppWithdrawMsg", "Character withdrew application"),
+                    ("CharLeftCorpMsg", "Character leaves corporation"),
+                    ("CorpAllBillMsg", "Corp alliance billing message"),
+                    ("BillOutOfMoneyMsg", "Bill out of money"),
+                    (
+                        "InfrastructureHubBillAboutToExpire",
+                        "I-HUB bill about to expire",
+                    ),
+                    ("IHubDestroyedByBillFailure", "I_HUB destroyed by bill failure"),
+                ],
+                default=[
+                    "StructureAnchoring",
+                    "StructureDestroyed",
+                    "StructureFuelAlert",
+                    "StructureLostArmor",
+                    "StructureLostShields",
+                    "StructureOnline",
+                    "StructureServicesOffline",
+                    "StructureUnderAttack",
+                    "StructureWentHighPower",
+                    "StructureWentLowPower",
+                    "OrbitalAttacked",
+                    "OrbitalReinforced",
+                    "TowerAlertMsg",
+                    "TowerResourceAlertMsg",
+                    "SovStructureReinforced",
+                    "SovStructureDestroyed",
+                ],
+                help_text="Select which type of notifications should be forwarded to this webhook",
+                max_length=1338,
+                verbose_name="notification types",
+            ),
+        ),
+    ]

structures/models/eveuniverse.py

@@ -1,6 +1,5 @@
 """Eve Universe models for Structures."""
 
-
 from django.db import models
 from django.utils.translation import gettext_lazy as _
 from eveuniverse.models import EveSolarSystem

structures/models/notifications.py

@@ -409,7 +409,7 @@ class NotificationBase(models.Model):
         logger.info("%s: Trying to sent to webhook: %s", self, webhook)
         try:
             embed, ping_type = self._generate_embed(webhook.language_code)
-        except (OSError, NotImplementedError) as ex:
+        except OSError as ex:
             logger.warning("%s: Failed to generate embed: %s", self, ex, exc_info=True)
             return False
 

structures/models/owners.py

@@ -235,7 +235,9 @@ class Owner(models.Model):
                 is_alliance_main=False
             )
             if "update_fields" in kwargs:
-                kwargs["update_fields"].append("is_alliance_main")
+                u = set(kwargs["update_fields"])
+                u.add("is_alliance_main")
+                kwargs["update_fields"] = u
         super().save(*args, **kwargs)
 
     @property

structures/tests/core/notification_embeds/test_main.py

@@ -51,7 +51,8 @@ class TestBilType(TestCase):
 
 class TestNotificationEmbeds(TestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         cls.owner = OwnerFactory()
 
     def test_should_create_obj_from_notification(self):
@@ -69,15 +70,6 @@ class TestNotificationEmbeds(TestCase):
         )
         self.assertEqual(str(notification_embed), "1000000999:CorpBecameWarEligible")
 
-    def test_should_raise_exception_for_unsupported_notif_types(self):
-        # given
-        notification = NotificationFactory(
-            owner=self.owner, notif_type="XXXUnsupportedNotificationTypeXXX"
-        )
-        # when / then
-        with self.assertRaises(NotImplementedError):
-            NotificationBaseEmbed.create(notification)
-
     def test_should_require_notification_for_init(self):
         with self.assertRaises(TypeError):
             NotificationBaseEmbed(notification="dummy")
@@ -89,7 +81,8 @@ class TestNotificationEmbeds(TestCase):
 
 class TestNotificationEmbedsGenerate(TestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         load_eve_entities()
         alliance = EveAllianceInfoFactory(alliance_id=3001)
@@ -145,38 +138,44 @@ class TestNotificationEmbedsGenerate(TestCase):
         for notif_type in NotificationType.esi_notifications():
             with self.subTest(notif_type=notif_type):
                 # given
-                notification = (
+                notif = (
                     Notification.objects.select_related("owner", "sender")
                     .filter(notif_type=notif_type)
                     .first()
                 )
-                self.assertIsInstance(notification, Notification)
-                notification_embed = NotificationBaseEmbed.create(notification)
+                self.assertIsInstance(notif, Notification)
+                notif_embed = NotificationBaseEmbed.create(notif)
+
                 # when
-                discord_embed = notification_embed.generate_embed()
+                obj = notif_embed.generate_embed()
+
                 # then
-                self.assertIsInstance(discord_embed, dhooks_lite.Embed)
+                self.assertIsInstance(obj, dhooks_lite.Embed)
 
     def test_should_generate_embed_for_all_supported_esi_notification_types_normal(
         self,
     ):
-        # Structures references in notifications are pre-generated
+        # Pre-generate structures referenced in notifications
         StructureFactory(owner=self.owner, id=1000000000001, eve_type_name="Astrahus")
         StructureFactory(owner=self.owner, id=1000000000002, eve_type_name="Athanor")
+
         for notif_type in NotificationType.esi_notifications():
             with self.subTest(notif_type=notif_type):
                 # given
-                notification = (
+                notif = (
                     Notification.objects.select_related("owner", "sender")
                     .filter(notif_type=notif_type)
                     .first()
                 )
-                self.assertIsInstance(notification, Notification)
-                notification_embed = NotificationBaseEmbed.create(notification)
+                self.assertIsInstance(notif, Notification)
+                notif_embed = NotificationBaseEmbed.create(notif)
+
                 # when
-                discord_embed = notification_embed.generate_embed()
+                obj = notif_embed.generate_embed()
+
                 # then
-                self.assertIsInstance(discord_embed, dhooks_lite.Embed)
+                self.assertIsInstance(obj, dhooks_lite.Embed)
+                self.assertTrue(obj.description)
 
     def test_should_set_ping_everyone_for_color_danger(self):
         # given
@@ -244,7 +243,8 @@ class TestNotificationEmbedsGenerate(TestCase):
 
 class TestNotificationEmbedsClasses(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         user = UserMainDefaultOwnerFactory()
         cls.owner = OwnerFactory(user=user)
@@ -302,7 +302,8 @@ class TestNotificationEmbedsClasses(NoSocketsTestCase):
 
 class TestGeneratedNotification(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
 
     def test_should_create_tower_reinforced_embed(self):
@@ -327,7 +328,8 @@ class TestGeneratedNotification(NoSocketsTestCase):
 
 class TestEveNotificationEmbeds(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         cls.owner = OwnerFactory()
 
@@ -343,7 +345,7 @@ class TestEveNotificationEmbeds(NoSocketsTestCase):
         # when
         obj = embed.generate_embed()
         # then
-        self.assertIsInstance(obj, dhooks_lite.Embed)
+        self.assertTrue(obj.description)
 
     def test_should_create_sov_embed_without_sender(self):
         # given
@@ -357,4 +359,4 @@ class TestEveNotificationEmbeds(NoSocketsTestCase):
         # when
         obj = embed.generate_embed()
         # then
-        self.assertIsInstance(obj, dhooks_lite.Embed)
+        self.assertTrue(obj.description)

structures/tests/core/test_notification_structuretimers.py

@@ -43,7 +43,8 @@ if "structuretimers" in app_labels():
     @patch(MODULE_PATH + ".STRUCTURES_MOON_EXTRACTION_TIMERS_ENABLED", True)
     class TestTimersForStructureTimers(NoSocketsTestCase):
         @classmethod
-        def setUpTestData(cls):
+        def setUpClass(cls):
+            super().setUpClass()
             load_eveuniverse()
 
         def test_should_create_timer_for_reinforced_structure(self):

structures/tests/core/test_notifications_timerboard.py

@@ -29,7 +29,8 @@ if "timerboard" in app_labels():
     @patch("structuretimers.models.STRUCTURETIMERS_NOTIFICATIONS_ENABLED", False)
     class TestNotificationAddToTimerboard(NoSocketsTestCase):
         @classmethod
-        def setUpTestData(cls):
+        def setUpClass(cls):
+            super().setUpClass()
             load_eveuniverse()
             load_eve_entities()
             cls.owner = OwnerFactory(is_alliance_main=True)

structures/tests/core/test_serializers.py

@@ -28,7 +28,8 @@ def to_dict(lst: List[dict], key="id"):
 
 class TestStructureListSerializer(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         cls.factory = RequestFactory()
         load_eveuniverse()
         cls.user = UserMainDefaultFactory()
@@ -138,7 +139,8 @@ class TestStructureListSerializer(NoSocketsTestCase):
 
 class TestPocoListSerializer(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         cls.factory = RequestFactory()
         load_eveuniverse()
         alliance = EveAllianceInfoFactory(

structures/tests/core/test_starbases.py

@@ -8,7 +8,8 @@ from structures.tests.testdata.load_eveuniverse import load_eveuniverse
 
 class TestStarbases(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         cls.type_astrahus = EveType.objects.get(id=35832)
         cls.type_poco = EveType.objects.get(id=2233)
@@ -58,7 +59,8 @@ class TestStarbases(NoSocketsTestCase):
 
 class TestStarbasesFuelDuration(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
 
     def test_can_calculate_for_large_tower(self):

structures/tests/integration/test_tasks.py

@@ -58,7 +58,8 @@ TASKS_PATH = "structures.tasks"
 @patch(OWNERS_PATH + ".esi")
 class TestTasks(TestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
 
     def test_should_fetch_new_upwell_structure_from_esi(

structures/tests/integration/test_views.py

@@ -25,7 +25,8 @@ COMMON_PATH = "structures.views.common"
 
 class TestStructureListView(TestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         cls.user = UserMainDefaultFactory()
         cls.owner = OwnerFactory(user=cls.user)
@@ -56,7 +57,8 @@ class TestStructureListView(TestCase):
 
 class TestStatisticsView(TestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         cls.user = UserMainDefaultFactory()
         cls.owner = OwnerFactory(user=cls.user)
@@ -80,7 +82,8 @@ class TestStatisticsView(TestCase):
 
 class TestPocoView(TestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         character = EveCharacterFactory(character_name="Bruce Wayne")
         cls.user = UserMainBasicFactory(main_character__character=character)

structures/tests/models/test_eveuniverse.py

@@ -22,7 +22,8 @@ class TestEveSovereigntyMap(NoSocketsTestCase):
 
 class TestEveSpaceType(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
 
     def test_space_type_highsec(self):

structures/tests/models/test_notifications_1.py

@@ -28,7 +28,8 @@ MODULE_PATH = "structures.models.notifications"
 
 class TestNotification(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         load_eve_entities()
         cls.owner = OwnerFactory()
@@ -99,7 +100,7 @@ class TestNotification(NoSocketsTestCase):
                     self.assertTrue(notif.can_be_rendered)
 
     def test_can_be_rendered_3(self):
-        for notif_type in ["DeclareWar"]:
+        for notif_type in ["UnknownNotificationType"]:
             with self.subTest(notification_type=notif_type):
                 notif = Notification.objects.filter(notif_type=notif_type).first()
                 if notif:
@@ -191,7 +192,8 @@ class TestNotificationFilterForAllianceLevel(NoSocketsTestCase):
 
 class TestNotificationCreateFromStructure(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         EveEntityCorporationFactory(id=1000137, name="DED")
         cls.owner = OwnerFactory()
@@ -256,7 +258,8 @@ class TestNotificationCreateFromStructure(NoSocketsTestCase):
 
 class TestNotificationRelevantWebhooks(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         load_eve_entities()
 
@@ -407,7 +410,8 @@ class TestNotificationRelevantWebhooks(NoSocketsTestCase):
 
 class TestNotificationSendToConfiguredWebhooks(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
 
     @patch(MODULE_PATH + ".Notification.send_to_webhook")
@@ -546,7 +550,8 @@ class TestNotificationSendToConfiguredWebhooks(NoSocketsTestCase):
 @patch(MODULE_PATH + ".Webhook.send_message")
 class TestNotificationSendToWebhook(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         EveEntityCorporationFactory(id=1000137, name="DED")
 
@@ -612,18 +617,7 @@ class TestNotificationSendMessage(NoSocketsTestCase):
         self.assertIsNotNone(kwargs["content"])
         self.assertIsNotNone(kwargs["embeds"])
 
-    def test_should_ignore_unsupported_notif_types(self, mock_send_message):
-        # given
-        mock_send_message.return_value = 1
-        obj = NotificationFactory(
-            owner=self.owner, notif_type="XXXUnsupportedNotificationTypeXXX"
-        )
-        # when
-        result = obj.send_to_webhook(self.webhook)
-        # then
-        self.assertFalse(result)
-
-    def test_mark_notification_as_sent_when_successful(self, mock_send_message):
+    def test_should_mark_notification_as_sent_when_successful(self, mock_send_message):
         # given
         mock_send_message.return_value = True
         obj = NotificationFactory(owner=self.owner)
@@ -633,7 +627,7 @@ class TestNotificationSendMessage(NoSocketsTestCase):
         obj.refresh_from_db()
         self.assertTrue(obj.is_sent)
 
-    def test_dont_mark_notification_as_sent_when_error(self, mock_send_message):
+    def test_should_not_mark_notification_as_sent_when_error(self, mock_send_message):
         # given
         mock_send_message.return_value = 0
         obj = NotificationFactory(owner=self.owner)
@@ -643,7 +637,7 @@ class TestNotificationSendMessage(NoSocketsTestCase):
         obj.refresh_from_db()
         self.assertFalse(obj.is_sent)
 
-    def test_send_to_webhook_all_notification_types(self, mock_send_message):
+    def test_can_send_all_notification_types(self, mock_send_message):
         # given
         mock_send_message.return_value = 1
         types_tested = set()
@@ -736,7 +730,8 @@ class TestNotificationSendMessage(NoSocketsTestCase):
 @patch(MODULE_PATH + ".Webhook.send_message", spec=True)
 class TestNotificationPings(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         corporation = EveEntityCorporationFactory()
         cls.notif_params = {

structures/tests/models/test_notifications_2.py

@@ -363,7 +363,8 @@ class TestStructureFuelAlerts(NoSocketsTestCase):
 @patch(MODULE_PATH + ".Webhook.send_message", spec=True)
 class TestJumpFuelAlerts(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         EveEntityCorporationFactory(id=1000137, name="DED")
 
@@ -592,7 +593,8 @@ class TestJumpFuelAlerts(NoSocketsTestCase):
 
 class TestNotificationRelatedStructures(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         load_eve_entities()
         cls.owner = OwnerFactory()

structures/tests/models/test_notifications_3.py

@@ -22,7 +22,8 @@ MODULE_PATH = "structures.models.notifications"
 
 class TestGeneratedNotification(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
 
     def test_should_have_str(self):
@@ -53,7 +54,8 @@ class TestGeneratedNotification(NoSocketsTestCase):
 
 class TestGeneratedNotificationManagerCreatePosReinforced(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         cls.owner = OwnerFactory()
 
@@ -149,7 +151,8 @@ class TestGeneratedNotificationManagerCreatePosReinforced(NoSocketsTestCase):
 @mock.patch("structures.core.notification_timers.add_or_remove_timer")
 class TestProcessTimers(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         cls.owner = OwnerFactory()
 
     @mock.patch(MODULE_PATH + ".STRUCTURES_ADD_TIMERS", True)

structures/tests/models/test_notifications_discord.py

@@ -29,7 +29,8 @@ if "discord" in app_labels():
     @patch(MODULE_PATH + ".Webhook.send_message", spec=True)
     class TestGroupPings(NoSocketsTestCase):
         @classmethod
-        def setUpTestData(cls):
+        def setUpClass(cls):
+            super().setUpClass()
             load_eveuniverse()
             load_eve_entities()
             cls.group_1 = Group.objects.create(name="Dummy Group 1")

structures/tests/models/test_owners_1.py

@@ -27,7 +27,8 @@ MODULE_PATH = "structures.models.owners"
 
 class TestOwner(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         cls.alliance = EveAllianceInfoFactory(
             alliance_name="Wayne Enterprises", alliance_ticker="WYE"
         )
@@ -189,6 +190,26 @@ class TestOwner(NoSocketsTestCase):
         self.owner.refresh_from_db()
         self.assertFalse(self.owner.is_alliance_main)
 
+    def test_should_ensure_is_alliance_main_is_set_for_update_fields_1(self):
+        # given
+        corporation = EveCorporationInfoFactory(alliance=self.alliance)
+        owner = OwnerFactory(corporation=corporation, is_alliance_main=True)
+        # when
+        owner.save(update_fields={"is_active"})
+        # then
+        owner.refresh_from_db()
+        self.assertTrue(owner.is_alliance_main)
+
+    def test_should_ensure_is_alliance_main_is_set_for_update_fields_2(self):
+        # given
+        corporation = EveCorporationInfoFactory(alliance=self.alliance)
+        owner = OwnerFactory(corporation=corporation, is_alliance_main=True)
+        # when
+        owner.save(update_fields=["is_active"])
+        # then
+        owner.refresh_from_db()
+        self.assertTrue(owner.is_alliance_main)
+
     def test_should_allow_mains_from_other_alliances(self):
         # given
         self.owner.is_alliance_main = True
@@ -223,7 +244,8 @@ class TestOwner(NoSocketsTestCase):
 
 class TestOwnerHasSov(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         cls.owner = OwnerFactory()
         EveSovereigntyMapFactory(
@@ -474,7 +496,8 @@ class TestOwnerFetchToken(NoSocketsTestCase):
 
 class TestOwnerCharacters(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         cls.owner = OwnerFactory()
 
     def test_should_return_str(self):
@@ -544,7 +567,8 @@ class TestOwnerCharacters(NoSocketsTestCase):
 @patch(MODULE_PATH + ".notify_admins", spec=True)
 class TestOwnerDeleteCharacter(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         cls.owner = OwnerFactory(characters=False)
 
     def test_should_delete_character_and_notify(self, mock_notify_admins, mock_notify):

structures/tests/models/test_owners_2.py

@@ -30,7 +30,8 @@ MODULE_PATH = "structures.models.owners"
 @patch(MODULE_PATH + ".esi")
 class TestUpdateStructuresEsi(NoSocketsTestCase):
     @classmethod
-    def setUpTestData(cls):
+    def setUpClass(cls):
+        super().setUpClass()
         load_eveuniverse()
         cls.user = UserMainDefaultOwnerFactory()
         cls.owner = OwnerFactory(user=cls.user, structures_last_update_at=None)