Add auto assertion database override to core test (dataform-co#1602)

moker-spaghetti · Dec 4, 2023 · a22783a · a22783a
1 parent 362e85e
commit a22783a
Showing 1 changed file with 39 additions and 4 deletions.
diff --git a/core/main_test.ts b/core/main_test.ts
@@ -211,7 +211,11 @@ vars:
 config {
   type: "table",
   database: dataform.projectConfig.vars.var1,
-  description: dataform.projectConfig.vars.var2
+  schema: "tableSchema",
+  description: dataform.projectConfig.vars.var2,
+  assertions: {
+    nonNull: [dataform.projectConfig.vars.var3],
+  }
 }
 select 1 AS \${dataform.projectConfig.vars.var3}`
         );
@@ -223,6 +227,31 @@ select 1 AS \${dataform.projectConfig.vars.var3}`
 
         expect(asPlainObject(result.compile.compiledGraph)).deep.equals(
           asPlainObject({
+            assertions: [
+              {
+                canonicalTarget: {
+                  name: "tableSchema_file_assertions_rowConditions"
+                },
+                dependencyTargets: [
+                  {
+                    database: "value1",
+                    name: "file",
+                    schema: "tableSchema"
+                  }
+                ],
+                fileName: "definitions/file.sqlx",
+                parentAction: {
+                  database: "value1",
+                  name: "file",
+                  schema: "tableSchema"
+                },
+                query:
+                  "\nSELECT\n  'value3 IS NOT NULL' AS failing_row_condition,\n  *\nFROM `value1.tableSchema.file`\nWHERE NOT (value3 IS NOT NULL)\n",
+                target: {
+                  name: "tableSchema_file_assertions_rowConditions"
+                }
+              }
+            ],
             dataformCoreVersion: "3.0.0",
             graphErrors: {},
             projectConfig: {
@@ -241,23 +270,29 @@ select 1 AS \${dataform.projectConfig.vars.var3}`
                 },
                 canonicalTarget: {
                   database: "value1",
-                  name: "file"
+                  name: "file",
+                  schema: "tableSchema"
                 },
                 disabled: false,
                 enumType: "TABLE",
                 fileName: "definitions/file.sqlx",
                 query: "\n\nselect 1 AS value3",
                 target: {
                   database: "value1",
-                  name: "file"
+                  name: "file",
+                  schema: "tableSchema"
                 },
                 type: "table"
               }
             ],
             targets: [
               {
                 database: "value1",
-                name: "file"
+                name: "file",
+                schema: "tableSchema"
+              },
+              {
+                name: "tableSchema_file_assertions_rowConditions"
               }
             ]
           })