From 8f18bb6a5e588ffb010810afd30e358df02913a2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gon=C3=A9ri=20Le=20Bouder?= Date: Mon, 16 Sep 2024 10:12:29 -0400 Subject: [PATCH] AAP-31326: Add OrgName to the Marketing report --- ansible_ai_connect/users/reports/generators.py | 2 ++ ansible_ai_connect/users/reports/tests/test_generators.py | 5 ++++- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/ansible_ai_connect/users/reports/generators.py b/ansible_ai_connect/users/reports/generators.py index e3ced8c74..dd92ee3b2 100644 --- a/ansible_ai_connect/users/reports/generators.py +++ b/ansible_ai_connect/users/reports/generators.py @@ -125,6 +125,7 @@ def generate( "First name", "Last name", "Email", + "Organization name", "Plan name", "Trial started", "Trial expired_at", @@ -140,6 +141,7 @@ def generate( user["given_name"], user["family_name"], user["email"], + organization["name"], plan["plan"]["name"], plan["created_at"], plan["expired_at"], diff --git a/ansible_ai_connect/users/reports/tests/test_generators.py b/ansible_ai_connect/users/reports/tests/test_generators.py index baaaab934..63dacbc54 100644 --- a/ansible_ai_connect/users/reports/tests/test_generators.py +++ b/ansible_ai_connect/users/reports/tests/test_generators.py @@ -209,7 +209,10 @@ def tearDown(self): super().cleanup() def get_report_header(self) -> str: - return "OrgId,UUID,First name,Last name,Email,Plan name,Trial started,Trial expired_at" + return ( + "OrgId,UUID,First name,Last name,Email,Organization name," + "Plan name,Trial started,Trial expired_at" + ) def get_report_generator(self) -> BaseGenerator: return UserMarketingReportGenerator()