AAP-31326: Add OrgName to the Marketing report

ansible · Sep 16, 2024 · 8f18bb6 · 8f18bb6
1 parent 04e87cb
commit 8f18bb6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git 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
@@ -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()