Skip to content

Add OBO credential flow integration tests#352

Open
dhruv0811 wants to merge 8 commits intomainfrom
obo-integration-tests
Open

Add OBO credential flow integration tests#352
dhruv0811 wants to merge 8 commits intomainfrom
obo-integration-tests

Conversation

@dhruv0811
Copy link
Contributor

@dhruv0811 dhruv0811 commented Mar 2, 2026

Summary

  • Integration tests verifying OBO identity forwarding for both Model Serving and Databricks Apps CUJs
  • Uses two service principals and a whoami() UC function to assert the end user's identity is correctly propagated
  • Also fixes a typo in the ModelServingUserCredentials docstring (credential_strategycredentials_strategy)

Test plan

dhruv0811 and others added 6 commits March 2, 2026 12:00
Test that identity is forwarded correctly through both the Model Serving
(ModelServingUserCredentials) and Databricks Apps (direct token) OBO paths
using two different service principals and a whoami() UC function.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
…ials

WorkspaceClient doesn't accept credential_strategy directly.
Use Config object as shown in the existing unit tests.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
…t kwarg

The docstring had a typo (credential_strategy vs credentials_strategy).
Fixed both the test and the source docstring to use the correct parameter
name that WorkspaceClient actually accepts.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
The SQL current_user() returns the SP's UUID, not its display_name.
Compare the two whoami() results against each other instead.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
dhruv0811 and others added 2 commits March 2, 2026 15:05
Invoke pre-deployed Model Serving endpoint and Databricks App as two
different SPs, assert each sees their own identity via whoami() tool.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant