run-ccsp2-build works

metrics_utility/__init__.py

@@ -15,8 +15,8 @@ def manage():
         sys.path.append(awx_path)
         spec = importlib.util.find_spec('awx')
         if spec is None:
-            sys.stderr.write(f"Automation Controller modules not found in {awx_path}\n")
-            exit(1)
+            sys.stderr.write(f"Automation Controller modules not found in {awx_path} (AWX_PATH)\n")
+            sys.path.append(os.path.normpath(os.path.join(os.path.dirname(__file__), '..', 'mock_awx')))
 
     import django
     from awx import prepare_env

metrics_utility/automation_controller_billing/collector.py

@@ -2,9 +2,9 @@
 import json
 import logging
 
+import importlib.util
+
 import insights_analytics_collector as base
-from awx.main.utils import datetime_hook
-from awx.main.utils.pglock import advisory_lock
 from django.conf import settings
 from django.core.serializers.json import DjangoJSONEncoder
 from django.db import connection
@@ -13,8 +13,17 @@
 # from awx.main.models import Job
 # from awx.main.access import access_registry
 # from rest_framework.exceptions import PermissionDenied
-from metrics_utility.automation_controller_billing.package.factory import \
-    Factory as PackageFactory
+
+from metrics_utility.automation_controller_billing.package.factory import Factory as PackageFactory
+
+from awx.main.utils import datetime_hook
+
+if importlib.util.find_spec('ansible_base') is None:
+    # 2.4
+    from awx.main.utils.pglock import advisory_lock
+else:
+    # 2.5
+    from ansible_base.lib.utils.db import advisory_lock
 
 logger = logging.getLogger('metrics_utility.collector')
 

mock_awx/awx/__init__.py

@@ -0,0 +1,4 @@
+import os
+
+def prepare_env():
+    os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'settings')

mock_awx/awx/main/utils/__init__.py

@@ -0,0 +1,19 @@
+from django.utils.dateparse import parse_datetime
+
+def get_awx_version():
+    "24.6.123"
+
+def get_awx_http_client_headers():
+    return {
+        'Content-Type': 'application/json',
+        'User-Agent': '{} {} ({})'.format('Red Hat Ansible Automation Platform', get_awx_version(), 'UNLICENSED'),
+    }
+
+def datetime_hook(d):
+    new_d = {}
+    for key, value in d.items():
+        try:
+            new_d[key] = parse_datetime(value)
+        except TypeError:
+            new_d[key] = value
+    return new_d

mock_awx/awx/main/utils/pglock.py

@@ -0,0 +1 @@
+from django_pglocks import advisory_lock

run-ccsp2-build

@@ -0,0 +1,23 @@
+#!/bin/sh
+set -e
+
+export METRICS_UTILITY_PRICE_PER_NODE=11.55 # in USD
+export METRICS_UTILITY_REPORT_COMPANY_NAME="Partner A"
+export METRICS_UTILITY_REPORT_EMAIL="[email protected]"
+export METRICS_UTILITY_REPORT_END_USER_CITY="Springfield"
+export METRICS_UTILITY_REPORT_END_USER_COMPANY_NAME="Customer A"
+export METRICS_UTILITY_REPORT_END_USER_COUNTRY="US"
+export METRICS_UTILITY_REPORT_END_USER_STATE="TX"
+export METRICS_UTILITY_REPORT_H1_HEADING="CCSP NA Direct Reporting Template"
+export METRICS_UTILITY_REPORT_PO_NUMBER="123"
+export METRICS_UTILITY_REPORT_RHN_LOGIN="test_login"
+export METRICS_UTILITY_REPORT_SKU="MCT3752MO"
+export METRICS_UTILITY_REPORT_SKU_DESCRIPTION="EX: Red Hat Ansible Automation Platform, Full Support (1 Managed Node, Dedicated, Monthly)"
+export METRICS_UTILITY_REPORT_TYPE="CCSPv2"
+export METRICS_UTILITY_SHIP_PATH=./metrics_utility/test/test_data/
+export METRICS_UTILITY_SHIP_TARGET=directory
+
+uv run ./manage.py build_report --month=2024-04 --force "$@"
+
+set -x
+ls -l metrics_utility/test/test_data/reports/2024/04/