changes to add more examples for apply masking

README.md

@@ -246,6 +246,8 @@ This package was created using examples from [Serge](https://www.linkedin.com/in
 # References
 - https://docs.snowflake.com/en/user-guide/security-column-ddm-intro.html
 - https://getdbt.slack.com/archives/CJN7XRF1B/p1609177817234800
+- https://docs.snowflake.com/en/sql-reference/functions/policy_references.html
+- https://docs.snowflake.com/en/user-guide/security-column-ddm-use.html
 
 # Contributions
 Contributions to this package are welcomed. Please create issues for bugs or feature requests for enhancement ideas or PRs for any enhancement contributions.

integration_tests/dbt_project.yml

@@ -27,8 +27,7 @@ clean-targets:
 
 seeds:
   dbt_snow_mask_integration_tests:
-    sakila:
-      schema: raw_sakila
+    +schema: seeds
 
 models:
   pre-hook:
@@ -39,7 +38,7 @@ models:
 
   dbt_snow_mask_integration_tests:
     staging:
-      database: "DEV_ENTECHLOG_RAW_DB"
+      database: "DEV_ENTECHLOG_DEMO_DB"
       schema: staging
 
 on-run-start:

integration_tests/macros/snow-mask-ddl/create_masking_policy_mp_email.sql

@@ -0,0 +1,10 @@
+{% macro create_masking_policy_mp_email(node_database,node_schema) %}
+
+CREATE MASKING POLICY IF NOT EXISTS {{node_database}}.{{node_schema}}.mp_email AS (val string) 
+  RETURNS string ->
+      CASE WHEN CURRENT_ROLE() IN ('SYSADMIN') THEN val 
+           WHEN CURRENT_ROLE() IN ('DEVELOPER') THEN SHA2(val)
+      ELSE '**********'
+      END
+
+{% endmacro %}

integration_tests/models/sources/sakila/src_sakila.yml

@@ -1,7 +1,7 @@
 version: 2
 
 sources:
-  - name: raw_sakila
+  - name: seeds
     tables:
       - name: customer
         columns:

integration_tests/models/staging/pii/stg_contacts.sql

@@ -0,0 +1,19 @@
+WITH contacts AS (
+    SELECT
+        contact_id,
+        first_name,
+        last_name,
+        email,
+        gender,
+        ip_address,
+        ssn,phone
+
+    FROM {{ ref('contact') }}
+),
+
+final AS (
+    SELECT *
+    FROM contacts
+)
+
+SELECT * FROM final

integration_tests/models/staging/pii/stg_contacts.yml

@@ -0,0 +1,27 @@
+version: 2
+
+models:
+- name: stg_contacts
+  description: ""
+  columns:
+  - name: contact_id
+    description: ""
+  - name: first_name
+    description: ""
+  - name: last_name
+    description: ""
+  - name: email
+    description: "Email Address"
+    meta:
+      masking_policy: mp_email
+  - name: gender
+    description: ""
+
+  - name: ip_address
+    description: ""
+
+  - name: ssn
+    description: ""
+
+  - name: phone
+    description: ""

integration_tests/models/staging/sakila/stg_customer.sql

@@ -4,4 +4,4 @@
     )
 }}
 
-SELECT * FROM {{ source('raw_sakila', 'customer') }}
+SELECT * FROM {{ source('seeds', 'customer') }}