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()