fluent-plugin-kusto 0.0.1.beta → 0.0.2.beta

data/test/plugin/test_wif_tokenprovider.rb

@@ -0,0 +1,136 @@
+# frozen_string_literal: true
+
+require 'test/unit'
+require 'mocha/test_unit'
+require_relative '../../lib/fluent/plugin/auth/wif_tokenprovider'
+
+class DummyConfigForWIF
+  attr_reader :kusto_endpoint, :workload_identity_client_id, :workload_identity_tenant_id, :workload_identity_token_file_path
+
+  def initialize(kusto_endpoint, client_id, tenant_id, token_file_path = nil)
+    @kusto_endpoint = kusto_endpoint
+    @workload_identity_client_id = client_id
+    @workload_identity_tenant_id = tenant_id
+    @workload_identity_token_file_path = token_file_path
+  end
+
+  def logger
+    require 'logger'
+    Logger.new($stdout)
+  end
+end
+
+class WorkloadIdentityTest < Test::Unit::TestCase
+  def setup
+    @resource = 'https://kusto.kusto.windows.net'
+    @client_id = '074c3c54-29e2-4230-a81f-333868b8d6ca'
+    @tenant_id = '12345678-1234-1234-1234-123456789abc'
+    @token_file = '/tmp/test-token'
+  end
+
+  def test_initialize_with_custom_token_file
+    config = DummyConfigForWIF.new(@resource, @client_id, @tenant_id, @token_file)
+    provider = WorkloadIdentity.new(config)
+    
+    # Verify instance variables are set correctly
+    assert_equal @resource, provider.instance_variable_get(:@kusto_endpoint)
+    assert_equal @client_id, provider.instance_variable_get(:@client_id)
+    assert_equal @tenant_id, provider.instance_variable_get(:@tenant_id)
+    assert_equal @token_file, provider.instance_variable_get(:@token_file)
+    assert_equal "#{@resource}/.default", provider.instance_variable_get(:@scope)
+  end
+
+  def test_initialize_with_default_token_file
+    config = DummyConfigForWIF.new(@resource, @client_id, @tenant_id, nil)
+    provider = WorkloadIdentity.new(config)
+    
+    # Verify default token file is used
+    expected_default = '/var/run/secrets/azure/tokens/azure-identity-token'
+    assert_equal expected_default, provider.instance_variable_get(:@token_file)
+  end
+
+  def test_fetch_token_success
+    config = DummyConfigForWIF.new(@resource, @client_id, @tenant_id, @token_file)
+    provider = WorkloadIdentity.new(config)
+    
+    # Mock successful token acquisition
+    mock_response = {
+      'access_token' => 'wif-access-token',
+      'expires_in' => 7200
+    }
+    
+    provider.stubs(:acquire_workload_identity_token).returns(mock_response)
+    
+    result = provider.send(:fetch_token)
+    assert_equal 'wif-access-token', result[:access_token]
+    assert_equal 7200, result[:expires_in]
+  end
+
+  def test_acquire_workload_identity_token_success
+    config = DummyConfigForWIF.new(@resource, @client_id, @tenant_id, @token_file)
+    provider = WorkloadIdentity.new(config)
+    
+    # Mock file read and HTTP request
+    File.stubs(:read).with(@token_file).returns('fake-oidc-token')
+    
+    mock_response = mock
+    mock_response.stubs(:is_a?).with(Net::HTTPSuccess).returns(true)
+    mock_response.stubs(:body).returns('{"access_token":"wif-token","expires_in":7200}')
+    
+    mock_http = mock
+    mock_http.expects(:use_ssl=).with(true)
+    mock_http.expects(:request).returns(mock_response)
+    
+    Net::HTTP.stubs(:new).returns(mock_http)
+    
+    result = provider.send(:acquire_workload_identity_token)
+    assert_equal 'wif-token', result['access_token']
+    assert_equal 7200, result['expires_in']
+  end
+
+  def test_acquire_workload_identity_token_failure
+    config = DummyConfigForWIF.new(@resource, @client_id, @tenant_id, @token_file)
+    provider = WorkloadIdentity.new(config)
+    
+    # Mock file read and HTTP request failure
+    File.stubs(:read).with(@token_file).returns('fake-oidc-token')
+    
+    mock_response = mock
+    mock_response.stubs(:is_a?).with(Net::HTTPSuccess).returns(false)
+    mock_response.stubs(:code).returns(400)
+    mock_response.stubs(:body).returns('Bad Request')
+    
+    mock_http = mock
+    mock_http.expects(:use_ssl=).with(true)
+    mock_http.expects(:request).returns(mock_response)
+    
+    Net::HTTP.stubs(:new).returns(mock_http)
+    
+    assert_raise(RuntimeError, 'Failed to get access token: 400 Bad Request') do
+      provider.send(:acquire_workload_identity_token)
+    end
+  end
+
+  def test_oauth2_endpoint_formation
+    config = DummyConfigForWIF.new(@resource, @client_id, @tenant_id, @token_file)
+    provider = WorkloadIdentity.new(config)
+    
+    # Test that the endpoint is formatted correctly with tenant_id
+    File.stubs(:read).with(@token_file).returns('fake-oidc-token')
+    
+    # Mock the rest of the HTTP request to avoid actual network call
+    mock_response = mock
+    mock_response.stubs(:is_a?).with(Net::HTTPSuccess).returns(true)
+    mock_response.stubs(:body).returns('{"access_token":"test","expires_in":3600}')
+    
+    mock_http = mock
+    mock_http.expects(:use_ssl=).with(true)
+    mock_http.expects(:request).returns(mock_response)
+    
+    Net::HTTP.stubs(:new).returns(mock_http)
+    
+    # Just verify it doesn't crash - the important thing is that setup_config was called
+    result = provider.send(:acquire_workload_identity_token)
+    assert_equal 'test', result['access_token']
+  end
+end

metadata

@@ -1,58 +1,77 @@
 --- !ruby/object:Gem::Specification
 name: fluent-plugin-kusto
 version: !ruby/object:Gem::Version
-  version: 0.0.1.beta
+  version: 0.0.2.beta
 platform: ruby
 authors:
 - Komal Rani
 - Kusto OSS IDC Team
-autorequire: 
 bindir: bin
 cert_chain: []
-date: 2025-08-20 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 2.6.9
+        version: '2.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 2.6.9
+        version: '2.0'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 13.2.1
+        version: '13.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 13.2.1
+        version: '13.0'
 - !ruby/object:Gem::Dependency
   name: test-unit
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 3.6.7
+        version: '3.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 3.6.7
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.0'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.0'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '2'
 - !ruby/object:Gem::Dependency
   name: fluentd
   requirement: !ruby/object:Gem::Requirement
@@ -126,21 +145,23 @@ files:
 - lib/fluent/plugin/kusto_query.rb
 - lib/fluent/plugin/out_kusto.rb
 - test/helper.rb
+- test/plugin/e2e_kusto.rb
 - test/plugin/test_azcli_tokenprovider.rb
 - test/plugin/test_e2e_kusto.rb
+- test/plugin/test_mi_tokenprovider.rb
 - test/plugin/test_out_kusto_config.rb
 - test/plugin/test_out_kusto_format.rb
 - test/plugin/test_out_kusto_process.rb
 - test/plugin/test_out_kusto_start.rb
 - test/plugin/test_out_kusto_try_write.rb
 - test/plugin/test_out_kusto_write.rb
+- test/plugin/test_wif_tokenprovider.rb
 homepage: https://github.com/Azure/azure-kusto-fluentd
 licenses:
 - Apache-2.0
 metadata:
   fluentd_plugin: 'true'
   fluentd_group: output
-post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -148,24 +169,26 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '2.5'
+      version: 2.7.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - ">"
+  - - ">="
     - !ruby/object:Gem::Version
-      version: 1.3.1
+      version: '0'
 requirements: []
-rubygems_version: 3.0.3.1
-signing_key: 
+rubygems_version: 3.7.1
 specification_version: 4
 summary: A custom Fluentd output plugin for Azure Kusto ingestion.
 test_files:
 - test/helper.rb
+- test/plugin/e2e_kusto.rb
 - test/plugin/test_azcli_tokenprovider.rb
 - test/plugin/test_e2e_kusto.rb
+- test/plugin/test_mi_tokenprovider.rb
 - test/plugin/test_out_kusto_config.rb
 - test/plugin/test_out_kusto_format.rb
 - test/plugin/test_out_kusto_process.rb
 - test/plugin/test_out_kusto_start.rb
 - test/plugin/test_out_kusto_try_write.rb
 - test/plugin/test_out_kusto_write.rb
+- test/plugin/test_wif_tokenprovider.rb