github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-11 04:52:08 +00:00
Code Issues Projects Releases Wiki Activity

DRA Node E2E: add NodePrepareResourceCalled API

Browse Source
This commit is contained in:
Ed Bartosh 2023-06-08 12:00:06 +03:00
parent 673d0aaa60
commit 5c5f6e8fe2
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
test/e2e/dra/test-driver/app/gomega.go
View File

@ -17,6 +17,8 @@ limitations under the License.
package app package app
import ( import (
"strings"
"github.com/onsi/gomega/gcustom" "github.com/onsi/gomega/gcustom"
) )
@ -30,3 +32,13 @@ var BeRegistered = gcustom.MakeMatcher(func(actualCalls []GRPCCall) (bool, error
} }
return false, nil return false, nil
}).WithMessage("contain successful NotifyRegistrationStatus call") }).WithMessage("contain successful NotifyRegistrationStatus call")
// NodePrepareResouceCalled checks that NodePrepareResource API has been called
var NodePrepareResourceCalled = gcustom.MakeMatcher(func(actualCalls []GRPCCall) (bool, error) {
for _, call := range actualCalls {
if strings.HasSuffix(call.FullMethod, "/NodePrepareResource") && call.Err == nil {
return true, nil
}
}
return false, nil
}).WithMessage("contain NodePrepareResource call")