feat(e2e): dynamically fetch VM extension version with caching, timeouts, and fallback by surajssd · Pull Request #8064 · Azure/AgentBaker

surajssd · 2026-03-10T21:48:49Z

What this PR does / why we need it:

Replaces the hardcoded VM extension version "1.406" in createVMExtensionLinuxAKSNode with a dynamic Azure API lookup that fetches the latest version at runtime, cached to avoid redundant API calls across parallel tests.

Dynamic version discovery

Add CachedGetLatestVMExtensionImageVersion in e2e/cache.go with a GetLatestExtensionVersionRequest struct key supporting Location, ExtType, and Publisher
Update createVMExtensionLinuxAKSNode in e2e/test_helpers.go to accept a context.Context, resolve location from config.Config.DefaultLocation, and call the cached version lookup
Update all 7 callers in e2e/scenario_gpu_managed_experience_test.go and e2e/scenario_test.go to pass t.Context()

Robustness

Add a 30s context.WithTimeout in createVMExtensionLinuxAKSNode to prevent ListVersions from blocking indefinitely and consuming the global 90-minute test timeout
Fall back to the known-good hardcoded version "1.406" with a warning log when the Azure API call fails (timeout, throttling, network error), instead of failing the test
Fix nil-pointer panic in toolkit.Logf: add missing return after the nil guard's log.Printf fallback, matching the sibling Log function's pattern — without this, any toolkit.Logf call from a context without testing.TB (e.g. t.Context()) dereferences a nil t

Refactoring

Rename VMExtenstionVersion → vmExtensionVersion (unexported, fixing the typo)
Replace sort.Slice + last-element with slices.MaxFunc for O(n) version lookup
Rename less → cmp using cmp.Compare from stdlib, returning -1/0/1
Add nil-name guard in parseVersion as an early return, and on the slices.MaxFunc result in GetLatestVMExtensionImageVersion
Extract testable getLatestVMExtensionImageVersion function behind a vmExtensionImageVersionLister interface
Fix %w → %v in fmt.Sprintf panic call in e2e/config/config.go (%w is only valid in fmt.Errorf)

Tests

e2e/config/azure_vmext_test.go — 26 table-driven unit tests:
- Test_parseVersion: 10 cases (three-part, two-part, single-part, nil name, non-numeric, partially numeric, empty string, extra parts, large numbers, leading zeros)
- Test_vmExtensionVersion_cmp: 9 cases (equal, major/minor/patch comparisons, zero values)
- Test_getLatestVMExtensionImageVersion: 7 cases (multiple versions, single version, two-part versions, API errors, empty list, nil names, mixed valid/malformed)
e2e/cache_test.go — 5 unit tests for cachedFunc: consistent results, warm-vs-cold performance, per-key isolation, error caching, struct key support
e2e/scenario_gpu_managed_experience_test.go — Test_CreateVMExtensionLinuxAKSNode_Timing integration test with testing.Short() skip guard and deterministic cache consistency assertions

Testing Done

cd e2e
go test -run Test_CreateVMExtensionLinuxAKSNode_Timing -v -count=1
go test -run "Test_cachedFunc*" -v

cd config
go test -run Test_parseVersion -v
go test -run Test_vmExtensionVersion_cmp -v
go test -run Test_getLatestVMExtensionImageVersion -v

Copilot

Pull request overview

This PR updates the e2e VMSS model generation to dynamically resolve the latest Compute.AKS.Linux.AKSNode VM extension version via the Azure API and memoize the result to avoid repeated lookups.

Changes:

Add an e2e cache key + cached wrapper for GetLatestVMExtensionImageVersion.
Update createVMExtensionLinuxAKSNode to accept a context.Context and use the cached Azure lookup instead of a hardcoded version.
Optimize GetLatestVMExtensionImageVersion to request only the top result using Top + Orderby.

Reviewed changes

Copilot reviewed 5 out of 5 changed files in this pull request and generated 5 comments.

Show a summary per file

File	Description
e2e/test_helpers.go	Switch AKSNode extension creation to use cached latest-version lookup; update one call site for new signature.
e2e/scenario_test.go	Update scenario to pass a context into `createVMExtensionLinuxAKSNode`.
e2e/scenario_gpu_managed_experience_test.go	Update multiple GPU scenarios to pass a context into `createVMExtensionLinuxAKSNode`.
e2e/config/azure.go	Change extension version lookup to a server-side “top 1, order by name desc” query.
e2e/cache.go	Add cached wrapper + cache-key struct for VM extension latest-version lookups.

e2e/scenario_gpu_managed_experience_test.go

e2e/test_helpers.go

e2e/cache.go

e2e/test_helpers.go

e2e/scenario_test.go

Copilot

Pull request overview

Copilot reviewed 6 out of 6 changed files in this pull request and generated 1 comment.

e2e/scenario_gpu_managed_experience_test.go

Copilot

Pull request overview

Copilot reviewed 8 out of 8 changed files in this pull request and generated 2 comments.

e2e/config/azure.go

e2e/scenario_gpu_managed_experience_test.go

Replace hardcoded VM extension version "1.406" with a cached Azure API call that fetches the latest version at runtime. Add a caching layer via CachedGetLatestVMExtensionImageVersion to avoid redundant API calls across tests. Update all callers to pass context for proper propagation. Signed-off-by: Suraj Deshmukh <suraj.deshmukh@microsoft.com>

…ookup Replace `sort.Slice` + last-element pattern with `slices.MaxFunc` in `GetLatestVMExtensionImageVersion` for `O(n)` instead of `O(n log n)`. Rename the less method to `cmp` using `cmp.Compare` from stdlib, and add a nil name guard before accessing the result. Move the nil name check in `parseVersion` to an early return before splitting. Signed-off-by: Suraj Deshmukh <suraj.deshmukh@microsoft.com>

Replace the observation-only Test_CreateVMExtensionLinuxAKSNode_Timing from the GPU scenario file with proper unit tests in cache_test.go that assert caching correctness, warm call performance, per-key isolation, error caching, and struct key support. Signed-off-by: Suraj Deshmukh <suraj.deshmukh@microsoft.com>

Prevent the ListVersions Azure API call from blocking indefinitely when the API hangs, which would cause the global 90-minute test timeout to panic and kill all parallel tests. Signed-off-by: Suraj Deshmukh <suraj.deshmukh@microsoft.com>

…ilure Instead of failing the test when the Azure API call to discover the latest VM extension version fails, fall back to a known-good hardcoded version (1.406) and log a warning. This prevents transient API issues like network errors or throttling from causing test failures. Signed-off-by: Suraj Deshmukh <suraj.deshmukh@microsoft.com>

…ison Extract GetLatestVMExtensionImageVersion logic into a testable unexported function accepting a vmExtensionImageVersionLister interface. Add 26 table-driven tests covering parseVersion, vmExtensionVersion.cmp, and getLatestVMExtensionImageVersion including edge cases for nil names, malformed versions, API errors, and empty responses. Signed-off-by: Suraj Deshmukh <suraj.deshmukh@microsoft.com>

fmt.Sprintf does not support the error-wrapping directive %w, which caused a build failure. The %w verb is only valid in fmt.Errorf. Signed-off-by: Suraj Deshmukh <suraj.deshmukh@microsoft.com>

Replace fragile wall-clock timing assertions with deterministic checks for the createVMExtensionLinuxAKSNode cache test. The previous 10s/1s thresholds tested CI/network performance rather than caching logic. - Add testing.Short() skip guard for integration test - Add nil/empty checks on returned extensions and properties - Replace assert with require for fail-fast behavior - Keep cache consistency check and duration logging for diagnostics - Remove unused assert import Signed-off-by: Suraj Deshmukh <suraj.deshmukh@microsoft.com>

The nil guard in Logf fell through to t.Helper() and t.Logf() when the context had no testing.TB, causing a nil-pointer dereference panic. Add the missing return to match the sibling Log function's pattern. Signed-off-by: Suraj Deshmukh <suraj.deshmukh@microsoft.com>

surajssd · 2026-03-11T19:51:31Z

➜  cd e2e
➜  go test -run Test_CreateVMExtensionLinuxAKSNode_Timing -v -count=1
...
=== RUN   Test_CreateVMExtensionLinuxAKSNode_Timing
    scenario_gpu_managed_experience_test.go:619: First call duration: 6.699633333s
    scenario_gpu_managed_experience_test.go:627: Second call duration: 17.875µs
--- PASS: Test_CreateVMExtensionLinuxAKSNode_Timing (6.70s)
PASS
ok  	github.com/Azure/agentbaker/e2e	7.547s

Copilot AI review requested due to automatic review settings March 10, 2026 21:48

surajssd requested review from AbelHu, Devinwong, awesomenix, calvin197, cameronmeissner, djsly, ganeshkumarashok, juan-lee, junjiezhang1997, lilypan26, mxj220, pdamianov-dev, phealy, r2k1, sulixu, timmy-wright and zachary-bailey as code owners March 10, 2026 21:48

surajssd temporarily deployed to test March 10, 2026 21:48 — with GitHub Actions Inactive

Copilot started reviewing on behalf of surajssd March 10, 2026 21:49 View session

Copilot AI reviewed Mar 10, 2026

View reviewed changes

e2e/scenario_gpu_managed_experience_test.go Outdated Show resolved Hide resolved

e2e/test_helpers.go Show resolved Hide resolved

e2e/cache.go Show resolved Hide resolved

e2e/test_helpers.go Outdated Show resolved Hide resolved

e2e/scenario_test.go Outdated Show resolved Hide resolved

surajssd force-pushed the suraj/optimize-extensio-version-lookup branch from 8982d50 to a9e9293 Compare March 10, 2026 22:29

surajssd temporarily deployed to test March 10, 2026 22:29 — with GitHub Actions Inactive

surajssd force-pushed the suraj/optimize-extensio-version-lookup branch from a9e9293 to 7400daf Compare March 11, 2026 05:22

Copilot AI review requested due to automatic review settings March 11, 2026 05:22

surajssd temporarily deployed to test March 11, 2026 05:22 — with GitHub Actions Inactive

Copilot started reviewing on behalf of surajssd March 11, 2026 05:23 View session

Copilot AI reviewed Mar 11, 2026

View reviewed changes

e2e/scenario_gpu_managed_experience_test.go Outdated Show resolved Hide resolved

surajssd temporarily deployed to test March 11, 2026 05:33 — with GitHub Actions Inactive

surajssd changed the title ~~test(e2e): add timing test for createVMExtensionLinuxAKSNode~~ feat(e2e): dynamically fetch VM extension version with caching, timeouts, and fallback Mar 11, 2026

Copilot AI review requested due to automatic review settings March 11, 2026 05:43

surajssd temporarily deployed to test March 11, 2026 05:43 — with GitHub Actions Inactive

Copilot started reviewing on behalf of surajssd March 11, 2026 05:44 View session

Copilot AI reviewed Mar 11, 2026

View reviewed changes

e2e/config/azure.go Show resolved Hide resolved

e2e/scenario_gpu_managed_experience_test.go Outdated Show resolved Hide resolved

surajssd force-pushed the suraj/optimize-extensio-version-lookup branch from b7537c1 to fb410cb Compare March 11, 2026 18:49

surajssd temporarily deployed to test March 11, 2026 18:49 — with GitHub Actions Inactive

surajssd added 9 commits March 11, 2026 11:50

fix(e2e): use %v instead of %w in fmt.Sprintf for panic message

28a981a

fmt.Sprintf does not support the error-wrapping directive %w, which caused a build failure. The %w verb is only valid in fmt.Errorf. Signed-off-by: Suraj Deshmukh <suraj.deshmukh@microsoft.com>

surajssd force-pushed the suraj/optimize-extensio-version-lookup branch from fb410cb to 7872d35 Compare March 11, 2026 18:50

surajssd temporarily deployed to test March 11, 2026 18:50 — with GitHub Actions Inactive

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

feat(e2e): dynamically fetch VM extension version with caching, timeouts, and fallback#8064

feat(e2e): dynamically fetch VM extension version with caching, timeouts, and fallback#8064
surajssd wants to merge 9 commits intomainfrom
suraj/optimize-extensio-version-lookup

surajssd commented Mar 10, 2026 •

edited

Loading

Uh oh!

Copilot AI left a comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Uh oh!

Uh oh!

Copilot AI left a comment

Uh oh!

Uh oh!

Uh oh!

surajssd commented Mar 11, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

2 participants

Conversation

surajssd commented Mar 10, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Dynamic version discovery

Robustness

Refactoring

Tests

Testing Done

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Reviewed changes

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

surajssd commented Mar 11, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

2 participants

surajssd commented Mar 10, 2026 •

edited

Loading