Skip to content

feat(flux): enhance TestFlux with proper error and info validation#3181

Closed
Raakshass wants to merge 5 commits intokubeflow:masterfrom
Raakshass:feature/flux-test-improvements-3179
Closed

feat(flux): enhance TestFlux with proper error and info validation#3181
Raakshass wants to merge 5 commits intokubeflow:masterfrom
Raakshass:feature/flux-test-improvements-3179

Conversation

@Raakshass
Copy link

What this PR does / why we need it:

This PR enhances the TestFlux function in pkg/runtime/framework/plugins/flux/flux_test.go to follow the same testing pattern used by the MPI plugin, as discussed in #3064 (comment).

Changes:

  • Add wantMLPolicyError field to verify EnforceMLPolicy() error handling
  • Add wantInfo field to verify info object state after EnforceMLPolicy()
  • Add wantBuildError field to verify Build() error handling
  • Use gocmp.Diff with cmpopts.EquateErrors() for proper error comparison
  • Add info comparison with utiltesting.PodSetEndpointsCmpOpts

This mirrors the testing strategy of the MPI plugin for consistency across plugins.

Note: This PR is based on PR #3064's branch since the Flux plugin doesn't exist in master yet.

Which issue(s) this PR fixes (optional, in Fixes #<issue number>, #<issue number>, ... format, will close the issue(s) when PR gets merged):
Fixes #3179

Checklist:

  • Docs included if any changes are user facing

No docs needed - changes are limited to test file only.

vsoch and others added 5 commits February 3, 2026 18:00
Flux supports the majority of MPI flavors/variants, and can be used
to bootstrap MPI as a plugin. It adds other features for scheduling
and topology that can be used for simulations and ai/ml jobs.
This changeset adds the plugin implementation, including the
plugin module, tests, and an example with a small README to
serve as documentation for the time being.

Signed-off-by: vsoch <vsoch@users.noreply.github.com>
Signed-off-by: vsoch <vsoch@users.noreply.github.com>
Signed-off-by: vsoch <vsoch@users.noreply.github.com>
Signed-off-by: vsoch <vsoch@users.noreply.github.com>
Following the MPI plugin testing pattern, this commit adds:
- wantMLPolicyError field to verify EnforceMLPolicy error handling
- wantInfo field to verify info object after EnforceMLPolicy
- wantBuildError field to verify Build error handling
- gocmp.Diff comparisons with cmpopts.EquateErrors() for proper error comparison

Fixes kubeflow#3179
@google-oss-prow
Copy link

[APPROVALNOTIFIER] This PR is NOT APPROVED

This pull-request has been approved by:
Once this PR has been reviewed and has the lgtm label, please assign astefanutti for approval. For more information see the Kubernetes Code Review Process.

The full list of commands accepted by this bot can be found here.

Details Needs approval from an approver in each of these files:

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

@github-actions
Copy link

github-actions bot commented Feb 5, 2026

🎉 Welcome to the Kubeflow Trainer! 🎉

Thanks for opening your first PR! We're happy to have you as part of our community 🚀

Here's what happens next:

  • If you haven't already, please check out our Contributing Guide for repo-specific guidelines and the Kubeflow Contributor Guide for general community standards.
  • Our team will review your PR soon! cc @kubeflow/kubeflow-trainer-team

Join the community:

Feel free to ask questions in the comments if you need any help or clarification!
Thanks again for contributing to Kubeflow! 🙏

@Raakshass
Copy link
Author

Closing this PR as it was incorrectly based on PR #3064's branch, causing it to include unrelated commits.

I will reopen a clean PR once #3064 is merged. The implementation for #3179 is ready in my branch and I'll rebase onto master when the time comes.

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

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[Testing] FluxPolicy Testing run EnforceMLPolicy and Build

2 participants