MNTOR-3893 - Fix QA Custom tool for mocked data brokers

src/app/(proper_react)/(redesign)/(authenticated)/admin/qa-customs/onerepConfig.tsx

@@ -6,6 +6,8 @@
 
 import React, { useState, useEffect } from "react";
 import styles from "./ConfigPage.module.scss";
+import { RemovalStatus } from "../../../../../functions/universal/scanResult";
+import { DataBrokerRemovalStatus } from "../../../../../functions/universal/dataBroker";
 
 interface QaBrokerDataCounts {
   onerep_scan_result_id?: number;
@@ -23,6 +25,9 @@ interface QaBrokerDataCounts {
   status: string;
   manually_resolved: string;
   optout_attempts?: number;
+  scan_result_status: RemovalStatus;
+  broker_status: DataBrokerRemovalStatus;
+  url: string;
 }
 
 const endpointBase = "/api/v1/admin/qa-customs/onerep";
@@ -63,6 +68,9 @@ const OnerepConfigPage = (props: Props) => {
     last_name: "",
     status: "new",
     manually_resolved: "false",
+    scan_result_status: "new",
+    broker_status: "active",
+    url: "",
   };
 
   // Temporary state to hold form input for a new broker
@@ -383,6 +391,25 @@ const OnerepConfigPage = (props: Props) => {
                 </select>
               </label>
 
+              <label>
+                Broker status:
+                <select
+                  name="status"
+                  value={newBroker.broker_status}
+                  onChange={(e) => void handleChange(e)}
+                >
+                  <option value="active">Active</option>
+                  <option value="on_hold">In Progress</option>
+                  <option value="ceased_operation">Ceased operation</option>
+                  <option value="scan_under_maintenance">
+                    Scan under maintenance
+                  </option>
+                  <option value="removal_under_maintenance">
+                    Removal under maintenance
+                  </option>
+                </select>
+              </label>
+
               <label className={styles.label}>
                 Optout Attempts:
                 <input

src/app/(proper_react)/(redesign)/(authenticated)/user/(dashboard)/dashboard/[[...slug]]/page.tsx

@@ -14,6 +14,7 @@ import {
 } from "../../../../../../../functions/universal/user";
 import {
   getLatestScanForProfileByReason,
+  getMockedScanResults,
   getScanResultsWithBroker,
   getScansCountForProfile,
 } from "../../../../../../../../db/tables/onerep_scans";
@@ -46,6 +47,7 @@ import {
   getL10n,
 } from "../../../../../../../functions/l10n/serverComponents";
 import { getDataBrokerRemovalTimeEstimates } from "../../../../../../../functions/server/getDataBrokerRemovalTimeEstimates";
+import { getQaToggleRow } from "../../../../../../../../db/tables/qa_customs";
 
 const dashboardTabSlugs = ["action-needed", "fixed"];
 
@@ -112,6 +114,18 @@ export default async function DashboardPage(props: Props) {
     hasPremium(session.user),
   );
 
+  const mockScanResultsData = await getMockedScanResults(profileId ?? 100);
+
+  const qaToggles = await getQaToggleRow(profileId);
+  console.log({ qaToggles });
+  let showCustomBrokers = false;
+
+  if (qaToggles) {
+    showCustomBrokers = qaToggles.show_custom_brokers;
+  }
+
+  const brokerData = showCustomBrokers ? mockScanResultsData : latestScan;
+
   const scanCount =
     typeof profileId === "number"
       ? await getScansCountForProfile(profileId)
@@ -161,7 +175,7 @@ export default async function DashboardPage(props: Props) {
       user={session.user}
       isEligibleForPremium={userIsEligibleForPremium}
       isEligibleForFreeScan={userIsEligibleForFreeScan}
-      userScanData={latestScan}
+      userScanData={brokerData}
       userBreaches={subBreaches}
       enabledFeatureFlags={enabledFeatureFlags}
       monthlySubscriptionUrl={`${monthlySubscriptionUrl}&${additionalSubplatParams.toString()}`}

src/app/(proper_react)/(redesign)/(authenticated)/user/(dashboard)/dashboard/fix/data-broker-profiles/removal-under-maintenance/page.tsx

@@ -11,6 +11,7 @@ import {
 import { getCountryCode } from "../../../../../../../../../functions/server/getCountryCode";
 import { headers } from "next/headers";
 import {
+  getMockedScanResults,
   getScanResultsWithBroker,
   getScanResultsWithBrokerUnderMaintenance,
 } from "../../../../../../../../../../db/tables/onerep_scans";
@@ -20,6 +21,7 @@ import { getSubscriberEmails } from "../../../../../../../../../functions/server
 import { RemovalUnderMaintenanceView } from "./RemovalUnderMaintenanceView";
 import { hasPremium } from "../../../../../../../../../functions/universal/user";
 import { getEnabledFeatureFlags } from "../../../../../../../../../../db/tables/featureFlags";
+import { getQaToggleRow } from "../../../../../../../../../../db/tables/qa_customs";
 
 export default async function RemovalUnderMaintenance() {
   const session = await getServerSession();
@@ -64,19 +66,30 @@ export default async function RemovalUnderMaintenance() {
     "DataBrokerManualRemoval",
   );
 
-  if (
-    scansWithRemovalUnderMaintenance?.results.length === 0 ||
-    !scansWithRemovalUnderMaintenance
-  ) {
+  const mockScanResultsData = await getMockedScanResults(profileId ?? 100);
+  const qaToggles = await getQaToggleRow(profileId);
+  let showCustomBrokers = false;
+
+  if (qaToggles) {
+    showCustomBrokers = qaToggles.show_custom_brokers;
+  }
+
+  const brokerData = showCustomBrokers
+    ? mockScanResultsData
+    : scansWithRemovalUnderMaintenance;
+
+  if (brokerData?.results.length === 0 || !brokerData) {
     redirect(getNextStep.href);
   }
 
+  console.log({ brokerData });
+
   const subscriberEmails = await getSubscriberEmails(session.user);
 
   return (
     <RemovalUnderMaintenanceView
       stepDeterminationData={data}
-      data={scansWithRemovalUnderMaintenance}
+      data={brokerData}
       subscriberEmails={subscriberEmails}
       enabledFeatureFlags={enabledFeatureFlags}
     />

src/app/(proper_react)/(redesign)/(authenticated)/user/(dashboard)/dashboard/getUserDashboardState.tsx

@@ -33,6 +33,7 @@ const isMatchingContent = (
 ) => {
   const { hasExposures, hasUnresolvedBreaches, hasUnresolvedBrokers } =
     contentConditions;
+
   // If a user does not have any exposures it’s also not possible to have unresolved ones.
   // This check is meant to avoid adding invalid conditions in `getUserDashboardState`.
   /* c8 ignore next 8 */

src/app/api/v1/admin/qa-customs/onerep/route.ts

@@ -91,6 +91,9 @@ export async function POST(req: NextRequest) {
   const status = body.status || "new";
   const manually_resolved = body.manually_resolved || false;
   const optout_attempts = body.optout_attempts || null;
+  const scan_result_status = body.scan_result_status || "new";
+  const broker_status = body.broker_status || "active";
+  const url = body.url || "";
 
   const brokerData: QaBrokerData = {
     onerep_profile_id,
@@ -107,6 +110,9 @@ export async function POST(req: NextRequest) {
     status,
     manually_resolved,
     optout_attempts,
+    scan_result_status,
+    broker_status,
+    url,
   };
   try {
     await addQaCustomBroker(brokerData);

src/db/tables/onerep_scans.ts

@@ -18,7 +18,11 @@ import {
 } from "knex/types/tables";
 import { RemovalStatus } from "../../app/functions/universal/scanResult.js";
 import { CONST_DAY_MILLISECONDS } from "../../constants.ts";
-import { getQaCustomBrokers, getQaToggleRow } from "./qa_customs.ts";
+import {
+  getAllQaCustomBrokers,
+  getQaCustomBrokers,
+  getQaToggleRow,
+} from "./qa_customs.ts";
 
 const knex = createDbConnection();
 
@@ -450,6 +454,23 @@ async function getScanResultsWithBrokerUnderMaintenance(
   return { results: scanResults } as LatestOnerepScanData;
 }
 
+export async function getMockedScanResults(
+  onerepProfileId: number,
+): Promise<LatestOnerepScanData> {
+  if (onerepProfileId === null) {
+    return {
+      scan: null,
+      results: [],
+    } as LatestOnerepScanData;
+  }
+
+  const scan = await getLatestOnerepScan(onerepProfileId);
+  const scanResults: OnerepScanResultDataBrokerRow[] | OnerepScanResultRow[] =
+    await getAllQaCustomBrokers(onerepProfileId);
+
+  return { scan: scan ?? null, results: scanResults } as LatestOnerepScanData;
+}
+
 async function getScanResultsWithBroker(
   onerepProfileId: number | null,
   hasPremium: boolean | null,

src/db/tables/qa_customs.ts

@@ -2,11 +2,16 @@
  * License, v. 2.0. If a copy of the MPL was not distributed with this
  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
 
-import { OnerepScanResultRow } from "knex/types/tables";
+import {
+  OnerepScanResultDataBrokerRow,
+  OnerepScanResultRow,
+} from "knex/types/tables";
 import { logger } from "../../app/functions/server/logging";
 import createDbConnection from "../connect";
 import { getOnerepProfileId } from "./subscribers";
 import { HibpLikeDbBreach } from "../../utils/hibp";
+import { RemovalStatus } from "../../app/functions/universal/scanResult";
+import { DataBrokerRemovalStatus } from "../../app/functions/universal/dataBroker";
 
 const knex = createDbConnection();
 
@@ -25,6 +30,9 @@ interface QaBrokerData {
   status: string;
   manually_resolved: boolean;
   optout_attempts: number;
+  scan_result_status: RemovalStatus;
+  broker_status: DataBrokerRemovalStatus;
+  url: string;
 }
 
 interface QaBreachData {
@@ -123,10 +131,10 @@ async function addQaCustomBroker(brokerData: QaBrokerData): Promise<void> {
 
 async function getAllQaCustomBrokers(
   onerep_profile_id: number,
-): Promise<QaBrokerData[]> {
+): Promise<OnerepScanResultDataBrokerRow[] | OnerepScanResultRow[]> {
   const res = (await knex("qa_custom_brokers")
     .where("onerep_profile_id", onerep_profile_id)
-    .select("*")) as QaBrokerData[];
+    .select("*")) as OnerepScanResultDataBrokerRow[];
   return res;
 }
 
@@ -190,11 +198,13 @@ async function getQaToggleRow(emailHashOrOneRepId: string | number | null) {
   if (emailHashOrOneRepId === null || envIsProd()) {
     return null;
   }
+  // for email sha
   if (typeof emailHashOrOneRepId === "string") {
     return (await knex("qa_custom_toggles")
       .select("*")
       .where("email_hash", emailHashOrOneRepId)
       .first()) as QaToggleRow;
+    // for onerep ID
   } else if (typeof emailHashOrOneRepId === "number") {
     return (await knex("qa_custom_toggles")
       .select("*")