firebase-js-sdk/packages/data-connect/test/util.ts

/**
 * @license
 * Copyright 2024 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import {
  connectDataConnectEmulator,
  ConnectorConfig,
  DataConnect,
  getDataConnect,
  makeMemoryCacheProvider
} from '../src';

export const EMULATOR_PORT = process.env.DC_EMULATOR_PORT;
const USE_EMULATOR = !!EMULATOR_PORT;
export const CONNECTOR_NAME = 'tests';
export const LOCATION_NAME = 'us-west2';
export const SERVICE_NAME = 'fdc-service';
export const PROJECT_ID = USE_EMULATOR ? 'p' : 'jscore-sandbox-141b5';
export function getConnectionConfig(): ConnectorConfig {
  return {
    connector: CONNECTOR_NAME,
    location: LOCATION_NAME,
    service: SERVICE_NAME
  };
}

// Seed the database to have the proper fields to query, such as a list of tasks.
export function initDatabase(): DataConnect {
  const instance = getDataConnect(getConnectionConfig(), {
    cacheSettings: {
      cacheProvider: makeMemoryCacheProvider()
    }
  });
  if (USE_EMULATOR) {
    connectDataConnectEmulator(instance, 'localhost', Number(EMULATOR_PORT));
  }
  return instance;
}