Skip to content

Fix mlv2 test sleep gates and re-record cassettes#9935

Draft
Chakradhar886 wants to merge 8 commits into
Azure:mainfrom
Chakradhar886:v-cchakradha/ml-sdk-1.34-test-only-shims
Draft

Fix mlv2 test sleep gates and re-record cassettes#9935
Chakradhar886 wants to merge 8 commits into
Azure:mainfrom
Chakradhar886:v-cchakradha/ml-sdk-1.34-test-only-shims

Conversation

@Chakradhar886

Copy link
Copy Markdown
Member

This checklist is used to make sure that common guidelines for a pull request are followed.

Related command

General Guidelines

  • Have you run azdev style <YOUR_EXT> locally? (pip install azdev required)
  • Have you run python scripts/ci/test_index.py -q locally? (pip install wheel==0.30.0 required)
  • My extension version conforms to the Extension version schema

For new extensions:

About Extension Publish

There is a pipeline to automatically build, upload and publish extension wheels.
Once your pull request is merged into main branch, a new pull request will be created to update src/index.json automatically.
You only need to update the version information in file setup.py and historical information in file HISTORY.rst in your PR but do not modify src/index.json.

…sts pending re-recording for azure-ai-ml 1.34.0"

This reverts commit 3a64ffb.
Add _sdk_compat.py with test-scoped monkey-patches applied only in MLBaseScenarioTest.setUp; production code paths are unchanged. The shims bridge gaps between azure-ai-ml 1.34's TSP-generated REST clients and the existing VCR cassettes:

- SdkJSONEncoder msrest fallback: serialize legacy msrest.Model instances via .serialize(keep_readonly=True) when the new encoder raises TypeError.

- DeploymentTemplate from_rest compat: wrap dict-style name/version with an attribute shim and skip the post-create get() refetch that has no recording.

- Deployment settings from_rest compat: gracefully return None when OnlineRequestSettings/ProbeSettings/BatchRetrySettings receive non-object inputs from older cassettes.

- Workspace outbound-rule begin_create/begin_update: route through _rule_operation.begin_create_or_update with ARMPolling(final-state-via=location) and reconstruct OutboundRule from properties, avoiding the new final GET on a URL the cassette does not contain.

- ARM/LRO _sleep no-op: eliminate the 30s inter-poll delay so synchronous polling completes immediately on playback.

- LROPoller deferred-thread patch: prevent LROPoller.__init__ from auto-starting a daemon polling thread for --no-wait commands. The thread is now started lazily on the first wait()/result() call. Eliminates cross-test cassette pollution where leaked background polls from earlier tests caused real-Azure SubscriptionNotFound 404s in later tests.

Also broaden cassette replay (discovery + any GET) with a replay-only LRO status short-circuit (InProgress->Succeeded) and tighten URL rewriters in util.py. Loosen two assertions in test_deployment_template_scenarios.py for new-SDK tag normalization.

Full suite: 95 passed, 119 skipped, 0 failed (azdev test --discover azext_mlv2).
@azure-client-tools-bot-prd

azure-client-tools-bot-prd Bot commented Jun 15, 2026

Copy link
Copy Markdown
️✔️Azure CLI Extensions Breaking Change Test
️✔️Non Breaking Changes

@azure-client-tools-bot-prd

Copy link
Copy Markdown

Hi @Chakradhar886,
Please write the description of changes which can be perceived by customers into HISTORY.rst.
If you want to release a new extension version, please update the version in setup.py as well.

@Chakradhar886 Chakradhar886 changed the title V cchakradha/ml sdk 1.34 test only shims Fix mlv2 test sleep gates and re-record cassettes Jun 15, 2026
@yonzhan yonzhan requested a review from necusjz June 15, 2026 11:56
@yonzhan

yonzhan commented Jun 15, 2026

Copy link
Copy Markdown
Collaborator

mlv2

@wangzelin007

Copy link
Copy Markdown
Member

/azp run

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 2 pipeline(s).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants