MC-14937: Complete Page Builder Analytics data collection

- Swap query if staging is enabled

Author: davemacaulay

dev/tests/integration/_files/Magento/TestModuleCmsPageBuilderAnalytics/etc/reports.xml

@@ -6,7 +6,7 @@
  */
 -->
 <config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="urn:magento:module:Magento_Analytics:etc/reports.xsd">
-    <report name="pagebuilder_page_test" connection="default">
+    <report name="pagebuilder_page_cms_test" connection="default">
         <source name="cms_page">
             <attribute name="content"/>
             <link-source name="cms_page_store" link-type="inner">
@@ -20,4 +20,18 @@
             </filter>
         </source>
     </report>
+    <report name="pagebuilder_page_cms_staging_test" connection="default">
+        <source name="cms_page">
+            <attribute name="content"/>
+            <link-source name="cms_page_store" link-type="inner">
+                <using glue="and">
+                    <condition attribute="row_id" operator="eq" type="identifier">row_id</condition>
+                    <condition attribute="store_id" operator="in">0,1</condition>
+                </using>
+            </link-source>
+            <filter glue="and">
+                <condition attribute="identifier" operator="eq">page-builder-analytics-test-page</condition>
+            </filter>
+        </source>
+    </report>
 </config>

dev/tests/integration/testsuite/Magento/CmsPageBuilderAnalytics/Model/ContentTypeUsageReportProviderTest.php

@@ -37,7 +37,13 @@ public function testGetReport($expectedReportData, $ignoredContentTypes)
                 'connectionFactory' => $connectionFactoryMock
             ]
         );
-        $reportData = $contentTypeUsageReportProvider->getReport('pagebuilder_page_test');
+        $moduleManager = Bootstrap::getObjectManager()->get(\Magento\Framework\Module\Manager::class);
+        // If CMS Staging is enabled we need to use an alternative query
+        if ($moduleManager->isEnabled('Magento_CmsStaging')) {
+            $reportData = $contentTypeUsageReportProvider->getReport('pagebuilder_page_cms_staging_test');
+        } else {
+            $reportData = $contentTypeUsageReportProvider->getReport('pagebuilder_page_cms_test');
+        }
 
         foreach ($reportData->getInnerIterator() as $reportItem) {
             // Ignore the header within the report