Merge pull request #6 from tanukifk/sandbox

specify a time zone in CURRENT_DATE() in data_freshness_assertion

Author: Hugo Rialan

definitions/example.js

@@ -29,11 +29,13 @@ const commonAssertionsResult = commonAssertions({
       "dateColumn": "updated_date",
       "timeUnit": "DAY",
       "delayCondition": 1,
+      "timeZone": "America/Los_Angeles"
     },
     "second_table": {
       "dateColumn": "updated_date",
       "timeUnit": "MONTH",
       "delayCondition": 3,
+      "timeZone": "-08"
     }
   },
   dataCompletenessConditions: {

includes/data_freshness_assertions.js

@@ -17,7 +17,7 @@
 
 const assertions = [];
 
-const createDataFreshnessAssertion = (globalParams, tableName, delayCondition, timeUnit, dateColumn) => {
+const createDataFreshnessAssertion = (globalParams, tableName, delayCondition, timeUnit, dateColumn, timeZone = "UTC") => {
   const assertion = assert(`assert_freshness_${tableName}`)
     .database(globalParams.database)
     .schema(globalParams.schema)
@@ -27,7 +27,7 @@ const createDataFreshnessAssertion = (globalParams, tableName, delayCondition, t
                 WITH
                     freshness AS (
                         SELECT
-                            DATE_DIFF(CURRENT_DATE(), MAX(${dateColumn}), ${timeUnit}) AS delay
+                            DATE_DIFF(CURRENT_DATE("${timeZone}"), MAX(${dateColumn}), ${timeUnit}) AS delay
                         FROM
                             ${ctx.ref(tableName)}
                     )
@@ -53,9 +53,10 @@ module.exports = (globalParams, freshnessConditions) => {
     const {
       delayCondition,
       timeUnit,
-      dateColumn
+      dateColumn,
+      timeZone
     } = freshnessConditions[tableName];
-    createDataFreshnessAssertion(globalParams, tableName, delayCondition, timeUnit, dateColumn);
+    createDataFreshnessAssertion(globalParams, tableName, delayCondition, timeUnit, dateColumn, timeZone);
   }
 
   return assertions;