fix(firestore-bigquery-export): tests

Author: CorieW

firestore-bigquery-export/firestore-bigquery-change-tracker/src/__tests__/bigquery/snapshot.test.ts

@@ -29,6 +29,7 @@ import {
   FirestoreBigQueryEventHistoryTracker,
   RawChangelogViewSchema,
 } from "../../bigquery";
+import { LogLevel } from "../../logger";
 
 const fixturesDir = __dirname + "/../fixtures";
 const sqlDir = fixturesDir + "/sql";
@@ -55,6 +56,7 @@ const trackerInstance = new FirestoreBigQueryEventHistoryTracker({
   timePartitioningFirestoreField: undefined,
   clustering: null,
   bqProjectId: null,
+  logLevel: LogLevel.INFO,
 });
 
 async function readFormattedSQL(file: string): Promise<string> {

firestore-bigquery-export/firestore-bigquery-change-tracker/src/__tests__/fixtures/changeTracker.ts

@@ -4,6 +4,7 @@ import {
   FirestoreBigQueryEventHistoryTracker,
   FirestoreDocumentChangeEvent,
 } from "../..";
+import { LogLevel } from "../../logger";
 
 export const changeTracker = ({
   datasetId = "",
@@ -22,6 +23,7 @@ export const changeTracker = ({
   useIncrementalMaterializedView = false,
   maxStaleness = undefined,
   refreshIntervalMinutes = undefined,
+  logLevel = LogLevel.INFO,
 }): FirestoreBigQueryEventHistoryTracker => {
   return new FirestoreBigQueryEventHistoryTracker({
     datasetId,
@@ -40,6 +42,7 @@ export const changeTracker = ({
     useIncrementalMaterializedView,
     maxStaleness,
     refreshIntervalMinutes,
+    logLevel,
   });
 };
 

firestore-bigquery-export/functions/__tests__/__snapshots__/config.test.ts.snap

@@ -21,6 +21,7 @@ Object {
   "instanceId": undefined,
   "kmsKeyName": "test",
   "location": "us-central1",
+  "logLevel": "info",
   "maxDispatchesPerSecond": 10,
   "maxEnqueueAttempts": 3,
   "maxStaleness": undefined,

firestore-bigquery-export/functions/__tests__/config.test.ts

@@ -6,6 +6,7 @@ import * as yaml from "js-yaml";
 import mockedEnv from "mocked-env";
 
 import { clustering } from "../src/config";
+import { log } from "console";
 
 let restoreEnv;
 let extensionYaml;
@@ -19,6 +20,7 @@ const environment = {
   TRANSFORM_FUNCTION: "",
   CLUSTERING: "data,timestamp",
   KMS_KEY_NAME: "test",
+  LOG_LEVEL: "info",
 };
 
 //@ts-ignore
@@ -51,6 +53,7 @@ describe("extension config", () => {
       tableId: environment.TABLE_ID,
       clustering: clustering(environment.CLUSTERING),
       kmsKeyName: environment.KMS_KEY_NAME,
+      logLevel: environment.LOG_LEVEL,
     };
     expect(config()).toMatchSnapshot(env);
   });