github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-09 20:17:41 +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
import (
"strings"
"github.com/onsi/gomega/gcustom"
)
@ -30,3 +32,13 @@ var BeRegistered = gcustom.MakeMatcher(func(actualCalls []GRPCCall) (bool, error
}
return false, nil
}).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")