From 6d01b290626a82c2cca7fdb9ba4909cf049fa19d Mon Sep 17 00:00:00 2001
From: Robert Astel <rob.astel@gmail.com>
Date: Thu, 25 May 2023 09:27:27 -0400
Subject: [PATCH] Preserve quotes round trip
---
dbt_invoke/internal/_utils.py | 2 ++
dbt_invoke/internal/_version.py | 2 +-
tests/test_properties.py | 7 +++++++
.../test_property_files/customers_preserve_quotes.yml | 11 +++++++++++
.../customers_preserve_quotes_expected.yml | 11 +++++++++++
5 files changed, 32 insertions(+), 1 deletion(-)
create mode 100644 tests/test_property_files/customers_preserve_quotes.yml
create mode 100644 tests/test_property_files/customers_preserve_quotes_expected.yml
diff --git a/dbt_invoke/internal/_utils.py b/dbt_invoke/internal/_utils.py
index 28d5599..05178e9 100644
--- a/dbt_invoke/internal/_utils.py
+++ b/dbt_invoke/internal/_utils.py
@@ -83,6 +83,7 @@ def parse_yaml(location):
:return: The contents of the yaml file
"""
yaml = YAML(typ="rt")
+ yaml.preserve_quotes = True
with open(location, 'r') as stream:
try:
parsed_yaml = yaml.load(stream)
@@ -101,6 +102,7 @@ def write_yaml(location, data, mode='w'):
:return: None
"""
yaml = YAML(typ="rt")
+ yaml.preserve_quotes = True
try:
with open(location, mode) as stream:
yaml.dump(data, stream)
diff --git a/dbt_invoke/internal/_version.py b/dbt_invoke/internal/_version.py
index 020ed73..d93b5b2 100644
--- a/dbt_invoke/internal/_version.py
+++ b/dbt_invoke/internal/_version.py
@@ -1 +1 @@
-__version__ = '0.2.2'
+__version__ = '0.2.3'
diff --git a/tests/test_properties.py b/tests/test_properties.py
index f4bd918..855df4e 100644
--- a/tests/test_properties.py
+++ b/tests/test_properties.py
@@ -211,6 +211,13 @@ def test_multiline_quotes(self):
expected_file="customers_multiline_quotes_expected.yml",
)
+ def test_preserve_quotes(self):
+ self.edit_update_compare(
+ "customers_preserve_quotes.yml",
+ target_model="customers",
+ expected_file="customers_preserve_quotes_expected.yml",
+ )
+
def test_long_string(self):
self.edit_update_compare(
"customers_long_string.yml",
diff --git a/tests/test_property_files/customers_preserve_quotes.yml b/tests/test_property_files/customers_preserve_quotes.yml
new file mode 100644
index 0000000..b5d5677
--- /dev/null
+++ b/tests/test_property_files/customers_preserve_quotes.yml
@@ -0,0 +1,11 @@
+version: 2
+models:
+- name: customers
+ description: A description
+ config:
+ labels: {'contains_pii': 'no', 'created_by': 'somebody'}
+ columns:
+ - name: customer_id
+ description: ''
+ - name: created_at
+ description: ''
diff --git a/tests/test_property_files/customers_preserve_quotes_expected.yml b/tests/test_property_files/customers_preserve_quotes_expected.yml
new file mode 100644
index 0000000..b5d5677
--- /dev/null
+++ b/tests/test_property_files/customers_preserve_quotes_expected.yml
@@ -0,0 +1,11 @@
+version: 2
+models:
+- name: customers
+ description: A description
+ config:
+ labels: {'contains_pii': 'no', 'created_by': 'somebody'}
+ columns:
+ - name: customer_id
+ description: ''
+ - name: created_at
+ description: ''