mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-12-11 23:50:19 +00:00
7b06b0f218
Bumps the github-dependencies group with 2 updates: [github.com/onsi/ginkgo/v2](https://github.com/onsi/ginkgo) and [github.com/onsi/gomega](https://github.com/onsi/gomega). Updates `github.com/onsi/ginkgo/v2` from 2.20.2 to 2.21.0 - [Release notes](https://github.com/onsi/ginkgo/releases) - [Changelog](https://github.com/onsi/ginkgo/blob/master/CHANGELOG.md) - [Commits](https://github.com/onsi/ginkgo/compare/v2.20.2...v2.21.0) Updates `github.com/onsi/gomega` from 1.34.2 to 1.35.1 - [Release notes](https://github.com/onsi/gomega/releases) - [Changelog](https://github.com/onsi/gomega/blob/master/CHANGELOG.md) - [Commits](https://github.com/onsi/gomega/compare/v1.34.2...v1.35.1) --- updated-dependencies: - dependency-name: github.com/onsi/ginkgo/v2 dependency-type: direct:production update-type: version-update:semver-minor dependency-group: github-dependencies - dependency-name: github.com/onsi/gomega dependency-type: direct:production update-type: version-update:semver-minor dependency-group: github-dependencies ... Signed-off-by: dependabot[bot] <support@github.com>
138 lines
4.0 KiB
Go
138 lines
4.0 KiB
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/onsi/gomega/types"
|
|
)
|
|
|
|
type Gomega struct {
|
|
Fail types.GomegaFailHandler
|
|
THelper func()
|
|
DurationBundle DurationBundle
|
|
}
|
|
|
|
func NewGomega(bundle DurationBundle) *Gomega {
|
|
return &Gomega{
|
|
Fail: nil,
|
|
THelper: nil,
|
|
DurationBundle: bundle,
|
|
}
|
|
}
|
|
|
|
func (g *Gomega) IsConfigured() bool {
|
|
return g.Fail != nil && g.THelper != nil
|
|
}
|
|
|
|
func (g *Gomega) ConfigureWithFailHandler(fail types.GomegaFailHandler) *Gomega {
|
|
g.Fail = fail
|
|
g.THelper = func() {}
|
|
return g
|
|
}
|
|
|
|
func (g *Gomega) ConfigureWithT(t types.GomegaTestingT) *Gomega {
|
|
g.Fail = func(message string, _ ...int) {
|
|
t.Helper()
|
|
t.Fatalf("\n%s", message)
|
|
}
|
|
g.THelper = t.Helper
|
|
return g
|
|
}
|
|
|
|
func (g *Gomega) Ω(actual interface{}, extra ...interface{}) types.Assertion {
|
|
return g.ExpectWithOffset(0, actual, extra...)
|
|
}
|
|
|
|
func (g *Gomega) Expect(actual interface{}, extra ...interface{}) types.Assertion {
|
|
return g.ExpectWithOffset(0, actual, extra...)
|
|
}
|
|
|
|
func (g *Gomega) ExpectWithOffset(offset int, actual interface{}, extra ...interface{}) types.Assertion {
|
|
return NewAssertion(actual, g, offset, extra...)
|
|
}
|
|
|
|
func (g *Gomega) Eventually(actualOrCtx interface{}, args ...interface{}) types.AsyncAssertion {
|
|
return g.makeAsyncAssertion(AsyncAssertionTypeEventually, 0, actualOrCtx, args...)
|
|
}
|
|
|
|
func (g *Gomega) EventuallyWithOffset(offset int, actualOrCtx interface{}, args ...interface{}) types.AsyncAssertion {
|
|
return g.makeAsyncAssertion(AsyncAssertionTypeEventually, offset, actualOrCtx, args...)
|
|
}
|
|
|
|
func (g *Gomega) Consistently(actualOrCtx interface{}, args ...interface{}) types.AsyncAssertion {
|
|
return g.makeAsyncAssertion(AsyncAssertionTypeConsistently, 0, actualOrCtx, args...)
|
|
}
|
|
|
|
func (g *Gomega) ConsistentlyWithOffset(offset int, actualOrCtx interface{}, args ...interface{}) types.AsyncAssertion {
|
|
return g.makeAsyncAssertion(AsyncAssertionTypeConsistently, offset, actualOrCtx, args...)
|
|
}
|
|
|
|
func (g *Gomega) makeAsyncAssertion(asyncAssertionType AsyncAssertionType, offset int, actualOrCtx interface{}, args ...interface{}) types.AsyncAssertion {
|
|
baseOffset := 3
|
|
timeoutInterval := -time.Duration(1)
|
|
pollingInterval := -time.Duration(1)
|
|
intervals := []interface{}{}
|
|
var ctx context.Context
|
|
|
|
actual := actualOrCtx
|
|
startingIndex := 0
|
|
if _, isCtx := actualOrCtx.(context.Context); isCtx && len(args) > 0 {
|
|
// the first argument is a context, we should accept it as the context _only if_ it is **not** the only argumnent **and** the second argument is not a parseable duration
|
|
// this is due to an unfortunate ambiguity in early version of Gomega in which multi-type durations are allowed after the actual
|
|
if _, err := toDuration(args[0]); err != nil {
|
|
ctx = actualOrCtx.(context.Context)
|
|
actual = args[0]
|
|
startingIndex = 1
|
|
}
|
|
}
|
|
|
|
for _, arg := range args[startingIndex:] {
|
|
switch v := arg.(type) {
|
|
case context.Context:
|
|
ctx = v
|
|
default:
|
|
intervals = append(intervals, arg)
|
|
}
|
|
}
|
|
var err error
|
|
if len(intervals) > 0 {
|
|
timeoutInterval, err = toDuration(intervals[0])
|
|
if err != nil {
|
|
g.Fail(err.Error(), offset+baseOffset)
|
|
}
|
|
}
|
|
if len(intervals) > 1 {
|
|
pollingInterval, err = toDuration(intervals[1])
|
|
if err != nil {
|
|
g.Fail(err.Error(), offset+baseOffset)
|
|
}
|
|
}
|
|
|
|
return NewAsyncAssertion(asyncAssertionType, actual, g, timeoutInterval, pollingInterval, 1, ctx, offset)
|
|
}
|
|
|
|
func (g *Gomega) SetDefaultEventuallyTimeout(t time.Duration) {
|
|
g.DurationBundle.EventuallyTimeout = t
|
|
}
|
|
|
|
func (g *Gomega) SetDefaultEventuallyPollingInterval(t time.Duration) {
|
|
g.DurationBundle.EventuallyPollingInterval = t
|
|
}
|
|
|
|
func (g *Gomega) SetDefaultConsistentlyDuration(t time.Duration) {
|
|
g.DurationBundle.ConsistentlyDuration = t
|
|
}
|
|
|
|
func (g *Gomega) SetDefaultConsistentlyPollingInterval(t time.Duration) {
|
|
g.DurationBundle.ConsistentlyPollingInterval = t
|
|
}
|
|
|
|
func (g *Gomega) EnforceDefaultTimeoutsWhenUsingContexts() {
|
|
g.DurationBundle.EnforceDefaultTimeoutsWhenUsingContexts = true
|
|
}
|
|
|
|
func (g *Gomega) DisableDefaultTimeoutsWhenUsingContext() {
|
|
g.DurationBundle.EnforceDefaultTimeoutsWhenUsingContexts = false
|
|
}
|