table test

Author: hectorcast-db

tests/test_fieldmask.py

@@ -3,161 +3,62 @@
 from databricks.sdk.common.types.fieldmask import FieldMask
 
 
-def test_to_json_string_basic():
-    """Test ToJsonString with basic list of paths."""
+@pytest.mark.parametrize("input_paths,expected_result,description", [
+    (["field1", "field2", "field3"], "field1,field2,field3", "basic list of paths"),
+    (["single_field"], "single_field", "single path"),
+    ([], "", "empty paths list"),
+    (["user.name", "user.email", "address.street"], "user.name,user.email,address.street", "nested field paths"),
+])
+def test_to_json_string(input_paths, expected_result, description):
+    """Test ToJsonString with various path configurations."""
     field_mask = FieldMask()
-    field_mask.paths = ["field1", "field2", "field3"]
+    field_mask.paths = input_paths
 
     result = field_mask.ToJsonString()
 
-    assert result == "field1,field2,field3"
+    assert result == expected_result
 
 
-def test_to_json_string_single_path():
-    """Test ToJsonString with single path."""
+@pytest.mark.parametrize("input_string,expected_paths,description", [
+    ("field1,field2,field3", ["field1", "field2", "field3"], "basic comma-separated string"),
+    ("single_field", ["single_field"], "single field"),
+    ("", [], "empty string"),
+    ("user.name,user.email,address.street", ["user.name", "user.email", "address.street"], "nested field paths"),
+    ("field1, field2 , field3", ["field1", " field2 ", " field3"], "spaces around commas"),
+])
+def test_from_json_string_success_cases(input_string, expected_paths, description):
+    """Test FromJsonString with various valid input strings."""
     field_mask = FieldMask()
-    field_mask.paths = ["single_field"]
 
-    result = field_mask.ToJsonString()
-
-    assert result == "single_field"
-
-
-def test_to_json_string_empty_paths():
-    """Test ToJsonString with empty paths list."""
-    field_mask = FieldMask()
-    field_mask.paths = []
-
-    result = field_mask.ToJsonString()
-
-    assert result == ""
-
-
-def test_to_json_string_nested_paths():
-    """Test ToJsonString with nested field paths."""
-    field_mask = FieldMask()
-    field_mask.paths = ["user.name", "user.email", "address.street"]
-
-    result = field_mask.ToJsonString()
-
-    assert result == "user.name,user.email,address.street"
-
-
-def test_from_json_string_basic():
-    """Test FromJsonString with basic comma-separated string."""
-    field_mask = FieldMask()
-
-    field_mask.FromJsonString("field1,field2,field3")
-
-    assert field_mask.paths == ["field1", "field2", "field3"]
-
-
-def test_from_json_string_single_field():
-    """Test FromJsonString with single field."""
-    field_mask = FieldMask()
-
-    field_mask.FromJsonString("single_field")
-
-    assert field_mask.paths == ["single_field"]
-
-
-def test_from_json_string_empty_string():
-    """Test FromJsonString with empty string."""
-    field_mask = FieldMask()
-
-    field_mask.FromJsonString("")
-
-    assert field_mask.paths == []
-
-
-def test_from_json_string_nested_paths():
-    """Test FromJsonString with nested field paths."""
-    field_mask = FieldMask()
-
-    field_mask.FromJsonString("user.name,user.email,address.street")
-
-    assert field_mask.paths == ["user.name", "user.email", "address.street"]
-
-
-def test_from_json_string_with_spaces():
-    """Test FromJsonString with spaces around commas."""
-    field_mask = FieldMask()
-
-    field_mask.FromJsonString("field1, field2 , field3")
+    field_mask.FromJsonString(input_string)
 
-    assert field_mask.paths == ["field1", " field2 ", " field3"]
+    assert field_mask.paths == expected_paths
 
 
-def test_from_json_string_non_string_input():
-    """Test FromJsonString raises ValueError for non-string input."""
+@pytest.mark.parametrize("invalid_input,expected_error_substring,description", [
+    (123, "FieldMask JSON value not a string: 123", "non-string integer input"),
+    (None, "FieldMask JSON value not a string: None", "None input"),
+    (["field1", "field2"], "FieldMask JSON value not a string:", "list input"),
+    ({"field": "value"}, "FieldMask JSON value not a string:", "dict input"),
+])
+def test_from_json_string_error_cases(invalid_input, expected_error_substring, description):
+    """Test FromJsonString raises ValueError for invalid input types."""
     field_mask = FieldMask()
 
     with pytest.raises(ValueError) as exc_info:
-        field_mask.FromJsonString(123)
+        field_mask.FromJsonString(invalid_input)
 
-    assert "FieldMask JSON value not a string: 123" in str(exc_info.value)
+    assert expected_error_substring in str(exc_info.value)
 
 
-def test_from_json_string_none_input():
-    """Test FromJsonString raises ValueError for None input."""
-    field_mask = FieldMask()
-
-    with pytest.raises(ValueError) as exc_info:
-        field_mask.FromJsonString(None)
-
-    assert "FieldMask JSON value not a string: None" in str(exc_info.value)
-
-
-def test_from_json_string_list_input():
-    """Test FromJsonString raises ValueError for list input."""
-    field_mask = FieldMask()
-
-    with pytest.raises(ValueError) as exc_info:
-        field_mask.FromJsonString(["field1", "field2"])
-
-    assert "FieldMask JSON value not a string:" in str(exc_info.value)
-
-
-def test_from_json_string_dict_input():
-    """Test FromJsonString raises ValueError for dict input."""
-    field_mask = FieldMask()
-
-    with pytest.raises(ValueError) as exc_info:
-        field_mask.FromJsonString({"field": "value"})
-
-    assert "FieldMask JSON value not a string:" in str(exc_info.value)
-
-
-def test_roundtrip_conversion():
+@pytest.mark.parametrize("original_paths,description", [
+    (["user.name", "user.email", "profile.settings"], "multiple nested fields"),
+    (["single_field"], "single field"),
+    ([], "empty paths"),
+])
+def test_roundtrip_conversion(original_paths, description):
     """Test that ToJsonString and FromJsonString are inverse operations."""
     field_mask = FieldMask()
-    original_paths = ["user.name", "user.email", "profile.settings"]
-    field_mask.paths = original_paths
-
-    # Convert to string and back.
-    json_string = field_mask.ToJsonString()
-    field_mask.FromJsonString(json_string)
-
-    assert field_mask.paths == original_paths
-
-
-def test_roundtrip_conversion_single_field():
-    """Test roundtrip conversion with single field."""
-    field_mask = FieldMask()
-    original_paths = ["single_field"]
-    field_mask.paths = original_paths
-
-    # Convert to string and back.
-    json_string = field_mask.ToJsonString()
-    field_mask.FromJsonString(json_string)
-
-    assert field_mask.paths == original_paths
-
-
-def test_roundtrip_conversion_empty():
-    """Test roundtrip conversion with empty paths."""
-    field_mask = FieldMask()
-    original_paths = []
     field_mask.paths = original_paths
 
     # Convert to string and back.