Merge pull request #52 from glassflow/kerberos-support

ETL-291: add kerberos parameters

Author: PabloPardoGarcia

VERSION

@@ -1 +1 @@
-3.3.0
+3.4.0

src/glassflow/etl/models/source.py

@@ -17,6 +17,7 @@ class KafkaMechanism(CaseInsensitiveStrEnum):
     SCRAM_SHA_256 = "SCRAM-SHA-256"
     SCRAM_SHA_512 = "SCRAM-SHA-512"
     PLAIN = "PLAIN"
+    GSSAPI = "GSSAPI"
 
 
 class SchemaField(BaseModel):
@@ -135,6 +136,10 @@ class KafkaConnectionParams(BaseModel):
     username: Optional[str] = Field(default=None)
     password: Optional[str] = Field(default=None)
     root_ca: Optional[str] = Field(default=None)
+    kerberos_service_name: Optional[str] = Field(default=None)
+    kerberos_keytab: Optional[str] = Field(default=None)
+    kerberos_realm: Optional[str] = Field(default=None)
+    kerberos_config: Optional[str] = Field(default=None)
     skip_auth: bool = Field(default=False)
 
     @model_validator(mode="before")

tests/test_pipeline.py

@@ -21,7 +21,9 @@ def test_create_success(self, pipeline, mock_success):
             mock_request.assert_called_once_with(
                 "POST",
                 pipeline.ENDPOINT,
-                json=pipeline.config.model_dump(mode="json", by_alias=True),
+                json=pipeline.config.model_dump(
+                    mode="json", by_alias=True, exclude_none=True
+                ),
             )
             assert result == pipeline
             assert pipeline.status == models.PipelineStatus.CREATED
@@ -325,7 +327,7 @@ def test_from_json(self, pipeline):
     def test_to_dict(self, pipeline):
         """Test pipeline to dictionary."""
         assert pipeline.to_dict() == pipeline.config.model_dump(
-            mode="json", by_alias=True
+            mode="json", by_alias=True, exclude_none=True
         )
 
         pipeline = Pipeline(host="http://localhost:8080", pipeline_id="test-pipeline")