authentik.tenants.tests.test_event_retention

Test event retention

 1"""Test event retention"""
 2
 3from django.test.client import RequestFactory
 4from django_tenants.utils import get_public_schema_name
 5from rest_framework.test import APITestCase
 6
 7from authentik.events.models import Event, EventAction
 8from authentik.lib.utils.time import timedelta_from_string
 9from authentik.tenants.models import Tenant
10
11
12class TestEventRetention(APITestCase):
13    """Test event retention"""
14
15    def test_event_retention(self):
16        """Test brand's event retention"""
17        default_tenant = Tenant.objects.get(schema_name=get_public_schema_name())
18        default_tenant.event_retention = "weeks=3"
19        default_tenant.save()
20        factory = RequestFactory()
21        request = factory.get("/")
22        request.tenant = default_tenant
23        event = Event.new(action=EventAction.SYSTEM_EXCEPTION, message="test").from_http(request)
24        self.assertEqual(event.expires.day, (event.created + timedelta_from_string("weeks=3")).day)
25        self.assertEqual(
26            event.expires.month,
27            (event.created + timedelta_from_string("weeks=3")).month,
28        )
29        self.assertEqual(
30            event.expires.year, (event.created + timedelta_from_string("weeks=3")).year
31        )
class TestEventRetention(rest_framework.test.APITestCase): 
13class TestEventRetention(APITestCase):
14    """Test event retention"""
15
16    def test_event_retention(self):
17        """Test brand's event retention"""
18        default_tenant = Tenant.objects.get(schema_name=get_public_schema_name())
19        default_tenant.event_retention = "weeks=3"
20        default_tenant.save()
21        factory = RequestFactory()
22        request = factory.get("/")
23        request.tenant = default_tenant
24        event = Event.new(action=EventAction.SYSTEM_EXCEPTION, message="test").from_http(request)
25        self.assertEqual(event.expires.day, (event.created + timedelta_from_string("weeks=3")).day)
26        self.assertEqual(
27            event.expires.month,
28            (event.created + timedelta_from_string("weeks=3")).month,
29        )
30        self.assertEqual(
31            event.expires.year, (event.created + timedelta_from_string("weeks=3")).year
32        )

Test event retention

def test_event_retention(self): 
16    def test_event_retention(self):
17        """Test brand's event retention"""
18        default_tenant = Tenant.objects.get(schema_name=get_public_schema_name())
19        default_tenant.event_retention = "weeks=3"
20        default_tenant.save()
21        factory = RequestFactory()
22        request = factory.get("/")
23        request.tenant = default_tenant
24        event = Event.new(action=EventAction.SYSTEM_EXCEPTION, message="test").from_http(request)
25        self.assertEqual(event.expires.day, (event.created + timedelta_from_string("weeks=3")).day)
26        self.assertEqual(
27            event.expires.month,
28            (event.created + timedelta_from_string("weeks=3")).month,
29        )
30        self.assertEqual(
31            event.expires.year, (event.created + timedelta_from_string("weeks=3")).year
32        )

Test brand's event retention