aa-ledger 1.0.4__py3-none-any.whl → 2.0.0__py3-none-any.whl

ledger/tests/test_tasks.py

@@ -1,282 +0,0 @@
-# Standard Library
-import logging
-from unittest.mock import patch
-
-# Django
-from django.test import TestCase
-from django.test.utils import override_settings
-from django.utils import timezone
-
-# AA Ledger
-from ledger import tasks
-from ledger.app_settings import LEDGER_CACHE_KEY
-from ledger.models.planetary import CharacterPlanetDetails
-from ledger.tests.testdata.generate_characteraudit import (
-    create_characteraudit_from_evecharacter,
-    create_update_status,
-)
-from ledger.tests.testdata.generate_corporationaudit import (
-    create_corporation_update_status,
-    create_corporationaudit_from_evecharacter,
-)
-from ledger.tests.testdata.generate_planets import (
-    _planetary_data,
-    create_character_planet,
-    create_character_planet_details,
-)
-from ledger.tests.testdata.load_allianceauth import load_allianceauth
-from ledger.tests.testdata.load_eveuniverse import load_eveuniverse
-
-TASK_PATH = "ledger.tasks"
-
-
-@patch(TASK_PATH + ".update_character", spec=True)
-class TestUpdateAllCharacters(TestCase):
-    @classmethod
-    def setUpClass(cls):
-        super().setUpClass()
-        load_allianceauth()
-
-        cls.audit = create_characteraudit_from_evecharacter(1001)
-
-    def test_should_update_all_characters(self, mock_update_character):
-        # when
-        tasks.update_all_characters()
-        # then
-        self.assertTrue(mock_update_character.apply_async.called)
-
-    def test_should_update_subset_characters(self, mock_update_character):
-        # when
-        tasks.update_subset_characters()
-        # then
-        self.assertTrue(mock_update_character.apply_async.called)
-
-
-@patch(TASK_PATH + ".chain", spec=True)
-@patch(TASK_PATH + ".logger", spec=True)
-class TestUpdateCharacter(TestCase):
-    @classmethod
-    def setUpClass(cls):
-        super().setUpClass()
-        load_allianceauth()
-
-        cls.audit = create_characteraudit_from_evecharacter(1001)
-
-    def test_update_character_should_no_updated(self, mock_logger, __):
-        # given
-        for section in self.audit.UpdateSection.get_sections():
-            create_update_status(
-                self.audit,
-                section=section,
-                is_success=True,
-                error_message="",
-                has_token_error=False,
-                last_run_at=timezone.now(),
-                last_run_finished_at=timezone.now(),
-                last_update_at=timezone.now(),
-                last_update_finished_at=timezone.now(),
-            )
-
-        # when
-        tasks.update_character(self.audit.pk)
-        # then
-        mock_logger.info.assert_called_once_with(
-            "No updates needed for %s",
-            self.audit.eve_character.character_name,
-        )
-
-    def test_update_character_should_update(self, mock_logger, mock_chain):
-        # given
-        create_update_status(
-            self.audit,
-            section=self.audit.UpdateSection.WALLET_JOURNAL,
-            is_success=True,
-            error_message="",
-            has_token_error=False,
-            last_run_at=None,
-            last_run_finished_at=None,
-            last_update_at=None,
-            last_update_finished_at=None,
-        )
-
-        # when
-        tasks.update_character(self.audit.pk)
-        # then
-        mock_chain.assert_called_once()
-
-
-@patch(TASK_PATH + ".update_corporation", spec=True)
-class TestUpdateAllCorporations(TestCase):
-    @classmethod
-    def setUpClass(cls):
-        super().setUpClass()
-        load_allianceauth()
-
-        cls.audit = create_corporationaudit_from_evecharacter(1001)
-
-    def test_should_update_all_corporations(self, mock_update_corporation):
-        # when
-        tasks.update_all_corporations()
-        # then
-        self.assertTrue(mock_update_corporation.apply_async.called)
-
-    def test_should_update_subset_corporation(self, mock_update_corporation):
-        # when
-        tasks.update_subset_corporations()
-        # then
-        self.assertTrue(mock_update_corporation.apply_async.called)
-
-
-@patch(TASK_PATH + ".chain", spec=True)
-@patch(TASK_PATH + ".logger", spec=True)
-class TestUpdateCorporation(TestCase):
-    @classmethod
-    def setUpClass(cls):
-        super().setUpClass()
-        load_allianceauth()
-
-        cls.audit = create_corporationaudit_from_evecharacter(1001)
-
-    def test_update_corporation_should_no_updated(self, mock_logger, __):
-        # given
-        for section in self.audit.UpdateSection.get_sections():
-            create_corporation_update_status(
-                self.audit,
-                section=section,
-                is_success=True,
-                error_message="",
-                has_token_error=False,
-                last_run_at=timezone.now(),
-                last_run_finished_at=timezone.now(),
-                last_update_at=timezone.now(),
-                last_update_finished_at=timezone.now(),
-            )
-        # when
-        tasks.update_corporation(self.audit.pk)
-        # then
-        mock_logger.info.assert_called_once_with(
-            "No updates needed for %s",
-            self.audit.corporation.corporation_name,
-        )
-
-    def test_update_corporation_should_update(self, mock_logger, mock_chain):
-        # given
-        create_corporation_update_status(
-            self.audit,
-            section=self.audit.UpdateSection.WALLET_JOURNAL,
-            is_success=True,
-            error_message="",
-            has_token_error=False,
-            last_run_at=None,
-            last_run_finished_at=None,
-            last_update_at=None,
-            last_update_finished_at=None,
-        )
-
-        # when
-        tasks.update_corporation(self.audit.pk)
-        # then
-        mock_chain.assert_called_once()
-
-
-@override_settings(
-    CELERY_ALWAYS_EAGER=True,
-    CELERY_EAGER_PROPAGATES_EXCEPTIONS=True,
-    APP_UTILS_OBJECT_CACHE_DISABLED=True,
-)
-@patch(TASK_PATH + ".logger", spec=True)
-class TestClearEtag(TestCase):
-    @classmethod
-    def setUpClass(cls):
-        super().setUpClass()
-        load_allianceauth()
-
-    def test_clear_all_etag_return_no_etags(self, mock_logger):
-        # when
-        tasks.clear_all_etags()
-
-        # then
-        mock_logger.info.assert_any_call("Deleting %s etag keys", 0)
-        mock_logger.info.assert_any_call("No etag keys to delete")
-
-    def test_clear_all_etag_return_etags(self, mock_logger):
-        # given
-        # pylint: disable=import-outside-toplevel
-        # Third Party
-        from django_redis import get_redis_connection
-
-        _client = get_redis_connection("default")
-        keys = _client.keys(f":?:{LEDGER_CACHE_KEY}-*")
-
-        # when
-        tasks.clear_all_etags()
-
-        # then
-        mock_logger.info.assert_any_call("Deleting %s etag keys", len(keys))
-        mock_logger.info.assert_any_call("Deleted %s etag keys", len(keys))
-
-
-@override_settings(
-    CELERY_ALWAYS_EAGER=True,
-    CELERY_EAGER_PROPAGATES_EXCEPTIONS=True,
-    APP_UTILS_OBJECT_CACHE_DISABLED=True,
-)
-class TestCheckPlanetaryNotification(TestCase):
-    @classmethod
-    def setUpClass(cls):
-        super().setUpClass()
-        load_allianceauth()
-        load_eveuniverse()
-
-        cls.planet_params = {
-            "upgrade_level": 5,
-            "num_pins": 5,
-        }
-
-        cls.planetdetails_params = {
-            "pin_id": 1,
-            "pin_type": "Extractor",
-            "status": "Active",
-            "last_update": None,
-            "last_update_finished_at": None,
-        }
-
-        cls.audit = create_characteraudit_from_evecharacter(1001)
-
-        cls.audit2 = create_characteraudit_from_evecharacter(1002)
-
-        cls.planet_1 = create_character_planet(
-            cls.audit, planet_id=4001, **cls.planet_params
-        )
-
-        cls.planet_2 = create_character_planet(
-            cls.audit2, planet_id=4001, **cls.planet_params
-        )
-
-        cls.planet_3 = create_character_planet(
-            cls.audit2, planet_id=4002, **cls.planet_params
-        )
-
-        cls.planetdetails_1 = create_character_planet_details(
-            cls.planet_1, notification=True, **_planetary_data
-        )
-
-        cls.planetdetails_2 = create_character_planet_details(
-            cls.planet_2,
-            notification=True,
-        )
-
-        cls.planetdetails_3 = create_character_planet_details(
-            cls.planet_3, notification=True, **_planetary_data
-        )
-
-    def test_check_planetary_notification(self):
-        # when
-        tasks.check_planetary_alarms()
-        # then
-        self.assertEqual(
-            CharacterPlanetDetails.objects.filter(
-                notification_sent=True,
-            ).count(),
-            2,
-        )

ledger/tests/test_view_helpers/test_core.py

@@ -1,47 +0,0 @@
-# Django
-from django.test import RequestFactory, TestCase
-
-# AA Ledger
-from ledger.helpers.core import (
-    add_info_to_context,
-)
-from ledger.tests.testdata.generate_corporationaudit import (
-    add_corporationaudit_corporation_to_user,
-    create_user_from_evecharacter,
-)
-from ledger.tests.testdata.load_allianceauth import load_allianceauth
-from ledger.tests.testdata.load_eveentity import load_eveentity
-from ledger.tests.testdata.load_eveuniverse import load_eveuniverse
-
-MODULE_PATH = "ledger.view_helpers.core"
-
-
-class TestViewHelpers(TestCase):
-    @classmethod
-    def setUpClass(cls):
-        super().setUpClass()
-        load_allianceauth()
-        load_eveuniverse()
-        load_eveentity()
-
-        cls.factory = RequestFactory()
-        cls.user, cls.character_ownership = create_user_from_evecharacter(
-            1001, permissions=["ledger.basic_access"]
-        )
-        cls.audit = add_corporationaudit_corporation_to_user(
-            cls.user, cls.character_ownership.character.character_id
-        )
-
-    def test_add_info_to_context(self):
-        # given
-        request = self.factory.get("/")
-        request.user = self.user
-
-        context = {
-            "theme": None,
-            "issues": [],
-        }
-        # when
-        result = add_info_to_context(request, context)
-        # then
-        self.assertEqual(result, context)

ledger/tests/test_views/corporation/test_corporation.py

@@ -1,267 +0,0 @@
-"""TestView class."""
-
-# Standard Library
-from http import HTTPStatus
-from unittest.mock import Mock, patch
-
-# Django
-from django.contrib.sessions.middleware import SessionMiddleware
-from django.test import RequestFactory, TestCase, override_settings
-from django.urls import reverse
-from django.utils import timezone
-
-# AA Ledger
-from ledger.models.general import EveEntity
-from ledger.tests.testdata.generate_corporationaudit import (
-    create_corporationaudit_from_user,
-    create_user_from_evecharacter,
-)
-from ledger.tests.testdata.generate_walletjournal import (
-    create_division,
-    create_wallet_journal_entry,
-)
-from ledger.tests.testdata.load_allianceauth import load_allianceauth
-from ledger.tests.testdata.load_eveentity import load_eveentity
-from ledger.tests.testdata.load_eveuniverse import load_eveuniverse
-from ledger.views.corporation.corporation_ledger import (
-    corporation_data_export_generate,
-    corporation_data_export_run_update,
-)
-
-MODULE_PATH = "ledger.views.corporation.corporation_ledger"
-
-
-@patch(MODULE_PATH + ".messages")
-@patch(MODULE_PATH + ".tasks")
-@override_settings(CELERY_ALWAYS_EAGER=True, CELERY_EAGER_PROPAGATES_EXCEPTIONS=True)
-class TestCorporationLedgerView(TestCase):
-    @classmethod
-    def setUpClass(cls):
-        super().setUpClass()
-        load_allianceauth()
-        load_eveuniverse()
-        load_eveentity()
-
-        cls.factory = RequestFactory()
-        cls.user, cls.character_ownership = create_user_from_evecharacter(
-            1001,
-            permissions=[
-                "ledger.basic_access",
-                "ledger.advanced_access",
-                "ledger.manage_access",
-            ],
-        )
-
-        cls.user_no_permissions, cls.character_ownership = (
-            create_user_from_evecharacter(
-                1002,
-                permissions=[
-                    "ledger.basic_access",
-                    "ledger.advanced_access",
-                    "ledger.manage_access",
-                ],
-            )
-        )
-        cls.audit = create_corporationaudit_from_user(cls.user)
-
-        cls.eve_character_first_party = EveEntity.objects.get(eve_id=2001)
-        cls.eve_character_second_party = EveEntity.objects.get(eve_id=1001)
-
-        cls.division = create_division(
-            corporation=cls.audit, name="MEGA KONTO", balance=1000000, division_id=1
-        )
-
-        cls.journal_entry = create_wallet_journal_entry(
-            journal_type="corporation",
-            division=cls.division,
-            context_id=1,
-            entry_id=10,
-            amount=1000,
-            balance=2000,
-            date=timezone.datetime.replace(
-                timezone.now(),
-                year=2016,
-                month=10,
-                day=29,
-                hour=0,
-                minute=0,
-                second=0,
-                microsecond=0,
-            ),
-            description="Test Journal",
-            first_party=cls.eve_character_first_party,
-            second_party=cls.eve_character_second_party,
-            ref_type="player_donation",
-        )
-
-    def test_corporation_data_export_generate(self, mock_tasks, mock_messages):
-        """Test should generate corporation data export"""
-        corporation_id = self.audit.corporation.corporation_id
-        mock_tasks.export_data_ledger.apply_async = Mock()
-
-        form_data = {
-            "year": "2016",
-        }
-
-        request = self.factory.post(
-            reverse("ledger:corporation_data_export_generate", args=[corporation_id]),
-            data=form_data,
-        )
-
-        request.user = self.user
-
-        response = corporation_data_export_generate(
-            request, corporation_id=corporation_id
-        )
-
-        self.assertEqual(response.status_code, HTTPStatus.FOUND)
-        mock_messages.info.assert_called_once_with(
-            request,
-            f"Data export for {corporation_id} has been started. This can take a couple of minutes. You will get a notification once it is completed.",
-        )
-        mock_tasks.export_data_ledger.apply_async.assert_called_once()
-
-    def test_corporation_data_export_generate_no_permission(
-        self, mock_tasks, mock_messages
-    ):
-        """Test should not generate corporation data export due to no permission"""
-        corporation_id = self.audit.corporation.corporation_id
-        mock_tasks.export_data_ledger.apply_async = Mock()
-
-        form_data = {
-            "year": "2016",
-        }
-
-        request = self.factory.post(
-            reverse("ledger:corporation_data_export_generate", args=[corporation_id]),
-            data=form_data,
-        )
-
-        request.user = self.user_no_permissions
-
-        response = corporation_data_export_generate(
-            request, corporation_id=corporation_id
-        )
-
-        self.assertEqual(response.status_code, HTTPStatus.FOUND)
-        mock_messages.error.assert_called_once_with(request, "Permission Denied")
-        mock_tasks.export_data_ledger.apply_async.assert_not_called()
-
-    def test_corporation_data_export_generate_corporation_not_found(
-        self, mock_tasks, mock_messages
-    ):
-        """Test should not generate corporation data export due to corporation not found"""
-        corporation_id = 9999  # Non-existent corporation ID
-        mock_tasks.export_data_ledger.apply_async = Mock()
-
-        form_data = {
-            "year": "2016",
-        }
-
-        request = self.factory.post(
-            reverse("ledger:corporation_data_export_generate", args=[corporation_id]),
-            data=form_data,
-        )
-
-        request.user = self.user
-
-        response = corporation_data_export_generate(
-            request, corporation_id=corporation_id
-        )
-
-        self.assertEqual(response.status_code, HTTPStatus.FOUND)
-        mock_messages.info.assert_called_once_with(request, "Corporation not found")
-        mock_tasks.export_data_ledger.apply_async.assert_not_called()
-
-    def test_corporation_data_export_generate_invalid_form(
-        self, mock_tasks, mock_messages
-    ):
-        """Test should not generate corporation data export due to invalid form"""
-        corporation_id = self.audit.corporation.corporation_id
-        mock_tasks.export_data_ledger.apply_async = Mock()
-
-        form_data = {
-            "year": "invalid_year",  # Invalid year
-        }
-
-        request = self.factory.post(
-            reverse("ledger:corporation_data_export_generate", args=[corporation_id]),
-            data=form_data,
-        )
-
-        request.user = self.user
-
-        response = corporation_data_export_generate(
-            request, corporation_id=corporation_id
-        )
-
-        self.assertEqual(response.status_code, HTTPStatus.FOUND)
-        mock_messages.error.assert_called_once_with(request, "Invalid form submission.")
-        mock_tasks.export_data_ledger.apply_async.assert_not_called()
-
-    @patch(MODULE_PATH + ".data_exporter.LedgerCSVExporter.decoder")
-    def test_corporation_data_export_run_update(
-        self, mock_decoder, mock_tasks, mock_messages
-    ):
-        """Test should run corporation data export update task"""
-        corporation_id = self.audit.corporation.corporation_id
-        hashcode = "testhashcode"
-        mock_tasks.export_data_ledger.apply_async = Mock()
-        mock_decoder.return_value = (2001, 1, 2016, 1)
-
-        request = self.factory.get(
-            reverse("ledger:corporation_data_export_run_update", args=[hashcode]),
-        )
-
-        request.user = self.user
-
-        response = corporation_data_export_run_update(request, hash_code=hashcode)
-
-        self.assertEqual(response.status_code, HTTPStatus.FOUND)
-        mock_messages.info.assert_called_once_with(
-            request,
-            f"Data export for {corporation_id} has been started. This can take a couple of minutes. You will get a notification once it is completed.",
-        )
-        mock_tasks.export_data_ledger.apply_async.assert_called_once()
-
-    @patch(MODULE_PATH + ".data_exporter.LedgerCSVExporter.decoder")
-    def test_corporation_data_export_run_update_no_permission(
-        self, mock_decoder, mock_tasks, mock_messages
-    ):
-        """Test should not run corporation data export update task due to no permission"""
-        hashcode = "testhashcode"
-        mock_tasks.export_data_ledger.apply_async = Mock()
-        mock_decoder.return_value = (2001, 1, 2016, 1)
-
-        request = self.factory.get(
-            reverse("ledger:corporation_data_export_run_update", args=[hashcode]),
-        )
-
-        request.user = self.user_no_permissions
-
-        response = corporation_data_export_run_update(request, hash_code=hashcode)
-
-        self.assertEqual(response.status_code, HTTPStatus.FOUND)
-        mock_messages.error.assert_called_once_with(request, "Permission Denied")
-        mock_tasks.export_data_ledger.apply_async.assert_not_called()
-
-    @patch(MODULE_PATH + ".data_exporter.LedgerCSVExporter.decoder")
-    def test_corporation_data_export_run_update_corporation_not_found(
-        self, mock_decoder, mock_tasks, mock_messages
-    ):
-        """Test should not run corporation data export update task due to corporation not found"""
-        hashcode = "testhashcode"
-        mock_tasks.export_data_ledger.apply_async = Mock()
-        mock_decoder.return_value = (9999, 1, 2016, 1)
-
-        request = self.factory.get(
-            reverse("ledger:corporation_data_export_run_update", args=[hashcode]),
-        )
-
-        request.user = self.user
-
-        response = corporation_data_export_run_update(request, hash_code=hashcode)
-
-        self.assertEqual(response.status_code, HTTPStatus.FOUND)
-        mock_messages.info.assert_called_once_with(request, "Corporation not found")
-        mock_tasks.export_data_ledger.apply_async.assert_not_called()