From 56811ecf8ef8db659a824d9b5a3c13e79429dc64 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Th=C3=A9ophile=20Madet?= <theo.madet@posteo.net>
Date: Mon, 23 Sep 2024 11:57:14 +0200
Subject: [PATCH] Added to test data generation the creation of fake update
 logs.

---
 .../commands/generate_test_data_functions.py  | 71 ++++++++++++++++++-
 tapir/utils/shortcuts.py                      |  6 +-
 2 files changed, 73 insertions(+), 4 deletions(-)

diff --git a/tapir/utils/management/commands/generate_test_data_functions.py b/tapir/utils/management/commands/generate_test_data_functions.py
index 27d179ecb..8a680b355 100644
--- a/tapir/utils/management/commands/generate_test_data_functions.py
+++ b/tapir/utils/management/commands/generate_test_data_functions.py
@@ -8,10 +8,13 @@
 from django.utils import timezone
 from django_auth_ldap.config import LDAPSearch
 from fabric.testing.fixtures import connection
+from faker import Faker
 from ldap.ldapobject import LDAPObject
 
 from tapir import settings
-from tapir.accounts.models import TapirUser
+from tapir.accounts.models import TapirUser, UpdateTapirUserLogEntry
+from tapir.accounts.tests.factories.factories import TapirUserFactory
+from tapir.accounts.tests.factories.user_data_factory import UserDataFactory
 from tapir.coop.models import (
     ShareOwner,
     ShareOwnership,
@@ -20,6 +23,7 @@
     NewMembershipsForAccountingRecap,
     ExtraSharesForAccountingRecap,
     MemberStatus,
+    MembershipResignation,
 )
 from tapir.coop.services.MemberInfoService import MemberInfoService
 from tapir.coop.services.MembershipPauseService import MembershipPauseService
@@ -107,6 +111,7 @@ def generate_tapir_users(json_users):
         tapir_user.is_active = True
         tapir_user.date_joined = json_user.date_joined
         tapir_user.password = tapir_user.username
+        tapir_user.co_purchaser = Faker().name() if random.random() > 0.5 else ""
         result.append(tapir_user)
 
     tapir_users = [tapir_user for tapir_user in result if tapir_user is not None]
@@ -403,13 +408,14 @@ def clear_ldap():
             connection.delete_s(build_ldap_group_dn(group_name))
 
     for tapir_user in tapir_users:
-        if tapir_user.get_ldap_user("check delete"):
+        if tapir_user.get_ldap_user():
             connection.delete_s(tapir_user.build_ldap_dn())
 
 
 def clear_django_db():
     classes = [
         LogEntry,
+        MembershipResignation,
         ShiftAttendance,
         ShiftCycleEntry,
         ShiftAccountEntry,
@@ -512,7 +518,64 @@ def generate_purchase_baskets():
         )
 
     PurchaseBasket.objects.bulk_create(baskets)
-    print("Done")
+
+
+def generate_log_updates():
+    print("Generating history of changes")
+    tapir_users = [tapir_user for tapir_user in TapirUser.objects.all()]
+    logs = {}
+    now = timezone.now()
+    fields = UserDataFactory.ATTRIBUTES + ["co_purchaser"]
+
+    nb_logs_to_create = TapirUser.objects.count() * 2
+    for _ in range(nb_logs_to_create):
+        tapir_user = random.choice(tapir_users)
+        if (now - tapir_user.date_joined).days < 7:
+            continue
+
+        if tapir_user in logs.keys():
+            oldest_log = logs[tapir_user][-1]
+            reference_date = oldest_log.created_date
+        else:
+            reference_date = now
+
+        date_range = round((reference_date - tapir_user.date_joined).days / 2.0)
+        le_random = random.randint(1, date_range)
+        log_date = now - datetime.timedelta(days=le_random)
+
+        log = UpdateTapirUserLogEntry(
+            created_date=log_date, user=tapir_user, actor=random.choice(tapir_users)
+        )
+        if tapir_user not in logs.keys():
+            logs[tapir_user] = []
+
+        old_values_reference = TapirUserFactory.build()
+        nb_fields = random.randint(1, 5)
+        fields_in_log = set()
+        for _ in range(nb_fields):
+            fields_in_log.add(random.choice(fields))
+
+        log.old_values = {}
+        log.new_values = {}
+        for field in fields_in_log:
+            old_value = getattr(old_values_reference, field)
+
+            new_value = None
+            for newer_log in logs[tapir_user]:
+                if field in newer_log.new_values.keys():
+                    new_value = newer_log.new_values.get(field)
+                    break
+            if new_value is None:
+                new_value = getattr(tapir_user, field)
+            if old_value != new_value:
+                log.old_values[field] = old_value
+                log.new_values[field] = new_value
+
+        log.save()
+        # the date gets overriden on creating since the field has auto_now_add
+        log.created_date = log_date
+        log.save()
+        logs[tapir_user].append(log)
 
 
 def reset_all_test_data():
@@ -524,3 +587,5 @@ def reset_all_test_data():
     generate_test_users()
     generate_test_applicants()
     generate_purchase_baskets()
+    generate_log_updates()
+    print("Done")
diff --git a/tapir/utils/shortcuts.py b/tapir/utils/shortcuts.py
index 9070ad49f..704e44ed6 100644
--- a/tapir/utils/shortcuts.py
+++ b/tapir/utils/shortcuts.py
@@ -58,7 +58,11 @@ def get_timezone_aware_datetime(
     date: datetime.date, time: datetime.time
 ) -> datetime.datetime:
     result = datetime.datetime.combine(date, time)
-    return timezone.make_aware(result) if timezone.is_naive(result) else result
+    return (
+        timezone.make_aware(result, is_dst=False)
+        if timezone.is_naive(result)
+        else result
+    )
 
 
 def setup_ssh_for_biooffice_storage():