Commite9de21a

committed

minor

1 parent2e12935 commite9de21aCopy full SHA for e9de21a

File tree

-10

lines changed

-10

lines changed

Lines changed: 0 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,3 @@`
`1`		`-#!/usr/bin/env python3`
`2`		`-`
`3`	`1`	`"""`
`4`	`2`	`Test script for Databricks SQL token federation with GitHub Actions OIDC tokens.`
`5`	`3`

Lines changed: 2 additions & 8 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,9 +1,3 @@`
`1`		`-#!/usr/bin/env python3`
`2`		`-`
`3`		`-"""`
`4`		`-Unit tests for token federation functionality in the Databricks SQL connector.`
`5`		`-"""`
`6`		`-`
`7`	`1`	`importpytest`
`8`	`2`	`fromunittest.mockimportMagicMock,patch`
`9`	`3`	`fromdatetimeimportdatetime,timezone,timedelta`
`@@ -134,7 +128,7 @@ def mock_discover_token_endpoint(self):`
`134`	`128`	`withpatch(`
`135`	`129`	`"databricks.sql.auth.oidc_utils.OIDCDiscoveryUtil.discover_token_endpoint"`
`136`	`130`	`)asmock:`
`137`		`-mock.return_value="https://databricks.com/token"`
	`131`	`+mock.return_value="https://databricks.com/oidc/v1/token"`
`138`	`132`	`yieldmock`
`139`	`133`
`140`	`134`	`@pytest.fixture`
`@@ -375,7 +369,7 @@ def test_exchange_token(`
`375`	`369`	`mock_request_post.return_value=mock_response`
`376`	`370`
`377`	`371`	`# Set the token endpoint`
`378`		`-federation_provider.token_endpoint="https://databricks.com/token"`
	`372`	`+federation_provider.token_endpoint="https://databricks.com/oidc/v1/token"`
`379`	`373`
`380`	`374`	`# Call the method`
`381`	`375`	`token=federation_provider._exchange_token("original_token")`

Comments

(0)