Reverting tests, now that I'm coercing these numbers to strings.

ethyca · Jul 17, 2024 · e93e9ab · e93e9ab
1 parent b27bf03
commit e93e9ab
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/tests/fideslang/test_models.py b/tests/fideslang/test_models.py
@@ -115,7 +115,7 @@ def test_system_valid(self) -> None:
             ],
             meta={"some": "meta stuff"},
             name="Test System",
-            organization_fides_key="1",
+            organization_fides_key=1,
             cookies=[{"name": "test_cookie"}],
             privacy_declarations=[
                 PrivacyDeclaration(
@@ -209,7 +209,7 @@ def test_system_valid_nested_meta(self) -> None:
                 },
             },
             name="Test System",
-            organization_fides_key="1",
+            organization_fides_key=1,
             privacy_declarations=[
                 PrivacyDeclaration(
                     data_categories=[],
@@ -256,7 +256,7 @@ def test_system_valid_no_meta(self) -> None:
             ],
             # purposefully omitting the `meta` property to ensure it's effectively optional
             name="Test System",
-            organization_fides_key="1",
+            organization_fides_key=1,
             privacy_declarations=[
                 PrivacyDeclaration(
                     data_categories=[],
@@ -278,7 +278,7 @@ def test_system_valid_no_egress_or_ingress(self) -> None:
             fides_key="test_system",
             meta={"some": "meta stuff"},
             name="Test System",
-            organization_fides_key="1",
+            organization_fides_key=1,
             privacy_declarations=[
                 PrivacyDeclaration(
                     data_categories=[],
@@ -305,7 +305,7 @@ def test_system_no_egress(self) -> None:
                 ],
                 meta={"some": "meta stuff"},
                 name="Test System",
-                organization_fides_key="1",
+                organization_fides_key=1,
                 privacy_declarations=[
                     PrivacyDeclaration(
                         data_categories=[],
@@ -338,7 +338,7 @@ def test_system_no_ingress(self) -> None:
                 fides_key="test_system",
                 meta={"some": "meta stuff"},
                 name="Test System",
-                organization_fides_key="1",
+                organization_fides_key=1,
                 privacy_declarations=[
                     PrivacyDeclaration(
                         data_categories=[],
@@ -370,7 +370,7 @@ def test_system_user_ingress_valid(self) -> None:
             ],
             meta={"some": "meta stuff"},
             name="Test System",
-            organization_fides_key="1",
+            organization_fides_key=1,
             privacy_declarations=[
                 PrivacyDeclaration(
                     data_categories=[],
@@ -387,7 +387,7 @@ def test_system_user_ingress_valid(self) -> None:
     def test_expanded_system(self):
         system = System(
             fides_key="test_system",
-            organization_fides_key="1",
+            organization_fides_key=1,
             tags=["some", "tags"],
             name="Exponential Interactive, Inc d/b/a VDX.tv",
             description="My system test",

diff --git a/tests/fideslang/test_validation.py b/tests/fideslang/test_validation.py
@@ -106,7 +106,7 @@ def test_built_from_dict_with_empty_versions(self, TaxonomyClass) -> None:
         """Try building from a dictionary with explicit None values."""
         TaxonomyClass.model_validate(
             {
-                "organization_fides_key": "1",
+                "organization_fides_key": 1,
                 "fides_key": "user",
                 "name": "Custom Test Data",
                 "description": "Custom Test Data Category",