replacing hyphens in MockTableMeta cte_name (#32)

Author: mcnick

CHANGELOG.md

@@ -16,6 +16,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Changed
 
 ### Fixed
+* Fixed generation of CTE names from references with hyphens
 
 ## [0.5.0]
 

src/sql_mock/table_mocks.py

@@ -343,4 +343,4 @@ class MockTableMeta(BaseModel):
     @property
     def cte_name(self):
         if getattr(self, "table_ref", None):
-            return self.table_ref.replace('"', "").replace(".", "__")
+            return self.table_ref.replace('"', "").replace(".", "__").replace("-", "_")

tests/sql_mock/test_table_mocks.py

@@ -1,7 +1,7 @@
 import pytest
 
 from sql_mock.column_mocks import ColumnMock
-from sql_mock.table_mocks import BaseMockTable, table_meta
+from sql_mock.table_mocks import BaseMockTable, MockTableMeta, table_meta
 
 
 class IntTestColumn(ColumnMock):
@@ -180,3 +180,11 @@ def test_to_sql_model_multiple_provided(self):
             ")"
         )
         assert sql_model == expected_sql_model
+
+
+def test_cte_name():
+    mock_table_meta = MockTableMeta(table_ref='"my-project.schema.table_name"')
+
+    expected_cte_name = "my_project__schema__table_name"
+
+    assert mock_table_meta.cte_name == expected_cte_name