fix: add code import as step audit (#1649)

Please ensure your pull request adheres to the following guidelines:
- [ ] make sure to link the related issues in this description
- [ ] when merging / squashing, make sure the fixed issue references are
visible in the commits, for easy compilation of release notes
- [ ] If data sources for any opportunity has been updated/added, please
update the
[wiki](https://wiki.corp.adobe.com/display/AEMSites/Data+Sources+for+Opportunities)
for same opportunity.

## Related Issues


Thanks for contributing!

Author: dmaurya929

src/accessibility/handler.js

@@ -413,6 +413,19 @@ export function createProcessAccessibilityOpportunitiesWithDevice(deviceType) {
   };
 }
 
+export async function codeImportStep(context) {
+  const {
+    log, site,
+  } = context;
+
+  log.info(`[A11yAudit] [Site Id: ${site.getId()}] starting code import step`);
+
+  return {
+    type: 'code',
+    siteId: site.getId(),
+  };
+}
+
 export default new AuditBuilder()
   .withUrlResolver((site) => site.resolveFinalURL())
   .addStep(
@@ -425,5 +438,6 @@ export default new AuditBuilder()
     scrapeAccessibilityData,
     AUDIT_STEP_DESTINATIONS.CONTENT_SCRAPER,
   )
+  .addStep('codeImport', codeImportStep, AUDIT_STEP_DESTINATIONS.IMPORT_WORKER)
   .addStep('processAccessibilityOpportunities', processAccessibilityOpportunities)
   .build();

src/accessibility/utils/data-processing.js

@@ -984,7 +984,7 @@ export async function sendRunImportMessage(
  * @param {Object} context - The context object containing log, s3Client, env
  * @returns {Promise<Object|null>} Object containing codeBucket and codePath, or null if should skip
  */
-async function getCodeInfo(site, opportunityType, context) {
+export async function getCodeInfo(site, opportunityType, context) {
   const { log, s3Client, env } = context;
   const siteId = site.getId();
   const deliveryType = site.getDeliveryType();

src/accessibility/utils/generate-individual-opportunities.js

@@ -20,7 +20,7 @@ import {
 import {
   successCriteriaLinks, accessibilityOpportunitiesMap, URL_SOURCE_SEPARATOR, issueTypesForCodeFix,
 } from './constants.js';
-import { getAuditData } from './data-processing.js';
+import { getAuditData, getCodeInfo } from './data-processing.js';
 import { processSuggestionsForMystique } from '../guidance-utils/mystique-data-processing.js';
 import { isAuditEnabledForSite } from '../../common/audit-utils.js';
 import { saveMystiqueValidationMetricsToS3, saveOpptyWithRetry } from './scrape-utils.js';
@@ -173,81 +173,45 @@ async function sendMystiqueMessage({
   log,
   context,
 }) {
+  // Create base message
+  const message = createDirectMystiqueMessage({
+    url,
+    issuesList,
+    opportunity,
+    siteId,
+    auditId,
+    deliveryType,
+    aggregationKey,
+  });
+
   // Check if code fix flow is enabled for this site
   const autoFixEnabled = await isAuditEnabledForSite('a11y-mystique-auto-fix', context.site, context);
   const useCodeFixFlow = autoFixEnabled && shouldUseCodeFixFlow(issuesList);
 
+  // Add code info if code fix flow is enabled
   if (useCodeFixFlow) {
-    const forwardPayload = createMystiqueForwardPayload({
-      url,
-      issuesList,
-      opportunity,
-      aggregationKey,
-      siteId,
-      auditId,
-      deliveryType,
-    });
+    const codeInfo = await getCodeInfo(context.site, 'accessibility', context);
+    if (codeInfo?.codeBucket && codeInfo?.codePath) {
+      message.data.codeBucket = codeInfo.codeBucket;
+      message.data.codePath = codeInfo.codePath;
+    }
+  }
 
-    const message = {
-      type: 'code',
-      siteId,
-      allowCache: true,
-      data: {},
-      forward: {
-        queue: env.QUEUE_SPACECAT_TO_MYSTIQUE,
-        payload: forwardPayload,
-      },
+  try {
+    await sqs.sendMessage(env.QUEUE_SPACECAT_TO_MYSTIQUE, message);
+    return {
+      success: true,
+      url,
     };
-
-    try {
-      await sqs.sendMessage(env.IMPORT_WORKER_QUEUE_URL, message);
-      log.info(
-        `[A11yIndividual] Sent message to import worker for code fix and forwarding to Mystique for url ${url}`,
-      );
-      return {
-        success: true,
-        url,
-      };
-    } catch (error) {
-      log.error(
-        `[A11yIndividual][A11yProcessingError] Failed to send message to import worker for url ${url} with error: ${error.message}`,
-      );
-      return {
-        success: false,
-        url,
-        error: error.message,
-      };
-    }
-  } else {
-    // Legacy flow: Send directly to Mystique without code injection
-    const message = createDirectMystiqueMessage({
+  } catch (error) {
+    log.error(
+      `[A11yIndividual][A11yProcessingError] Failed to send message to Mystique for url ${url} with error: ${error.message}`,
+    );
+    return {
+      success: false,
       url,
-      issuesList,
-      opportunity,
-      siteId,
-      auditId,
-      deliveryType,
-    });
-
-    try {
-      await sqs.sendMessage(env.QUEUE_SPACECAT_TO_MYSTIQUE, message);
-      log.info(
-        `[A11yIndividual] Sent message directly to Mystique (legacy flow) for url ${url}`,
-      );
-      return {
-        success: true,
-        url,
-      };
-    } catch (error) {
-      log.error(
-        `[A11yIndividual][A11yProcessingError] Failed to send message to Mystique for url ${url}, message: ${JSON.stringify(message, null, 2)} with error: ${error.message}`,
-      );
-      return {
-        success: false,
-        url,
-        error: error.message,
-      };
-    }
+      error: error.message,
+    };
   }
 }
 

test/audits/accessibility.test.js

@@ -30,6 +30,7 @@ describe('Accessibility Audit Handler', () => {
   let processAccessibilityOpportunities;
   let createProcessAccessibilityOpportunitiesWithDevice;
   let processImportStep;
+  let codeImportStep;
   let getUrlsForAuditStub;
   let aggregateAccessibilityDataStub;
   let generateReportOpportunitiesStub;
@@ -106,6 +107,7 @@ describe('Accessibility Audit Handler', () => {
     processAccessibilityOpportunities = accessibilityModule.processAccessibilityOpportunities;
     createProcessAccessibilityOpportunitiesWithDevice = accessibilityModule.createProcessAccessibilityOpportunitiesWithDevice;
     processImportStep = accessibilityModule.processImportStep;
+    codeImportStep = accessibilityModule.codeImportStep;
   });
 
   afterEach(() => {
@@ -1482,6 +1484,83 @@ describe('Accessibility Audit Handler', () => {
 
   });
 
+  describe('codeImportStep', () => {
+    it('should return correct step result with type and siteId', async () => {
+      // Arrange
+      const context = {
+        log: mockContext.log,
+        site: mockSite,
+      };
+
+      // Act
+      const result = await codeImportStep(context);
+
+      // Assert
+      expect(result).to.deep.equal({
+        type: 'code',
+        siteId: 'test-site-id',
+      });
+    });
+
+    it('should log the correct info message', async () => {
+      // Arrange
+      const context = {
+        log: mockContext.log,
+        site: mockSite,
+      };
+
+      // Act
+      await codeImportStep(context);
+
+      // Assert
+      expect(mockContext.log.info).to.have.been.calledWith(
+        '[A11yAudit] [Site Id: test-site-id] starting code import step',
+      );
+    });
+
+    it('should call site.getId() twice', async () => {
+      // Arrange
+      const context = {
+        log: mockContext.log,
+        site: mockSite,
+      };
+
+      // Reset the stub to track calls
+      mockSite.getId.resetHistory();
+
+      // Act
+      await codeImportStep(context);
+
+      // Assert
+      expect(mockSite.getId).to.have.been.calledTwice;
+    });
+
+    it('should handle different site IDs correctly', async () => {
+      // Arrange
+      const customSiteId = 'different-site-id';
+      const customMockSite = {
+        getId: sandbox.stub().returns(customSiteId),
+        getBaseURL: sandbox.stub().returns('https://different.com'),
+      };
+      const context = {
+        log: mockContext.log,
+        site: customMockSite,
+      };
+
+      // Reset log history to check specific calls
+      mockContext.log.info.resetHistory();
+
+      // Act
+      const result = await codeImportStep(context);
+
+      // Assert
+      expect(result.siteId).to.equal(customSiteId);
+      expect(mockContext.log.info).to.have.been.calledWith(
+        `[A11yAudit] [Site Id: ${customSiteId}] starting code import step`,
+      );
+    });
+  });
+
   describe('createProcessAccessibilityOpportunitiesWithDevice', () => {
     beforeEach(() => {
       // Reset context to include AWS_ENV for tests