mirror of
https://github.com/zitadel/zitadel.git
synced 2025-08-12 02:47:33 +00:00
34 lines
734 B
Go
34 lines
734 B
Go
package mock
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"go.uber.org/mock/gomock"
|
|
)
|
|
|
|
func NewIDGenerator(t *testing.T) *MockGenerator {
|
|
m := NewMockGenerator(gomock.NewController(t))
|
|
m.EXPECT().Next().Return("1", nil)
|
|
return m
|
|
}
|
|
|
|
func NewIDGeneratorExpectIDs(t *testing.T, ids ...string) *MockGenerator {
|
|
m := NewMockGenerator(gomock.NewController(t))
|
|
for _, id := range ids {
|
|
m.EXPECT().Next().Return(id, nil)
|
|
}
|
|
return m
|
|
}
|
|
|
|
func ExpectID(t *testing.T, id string) *MockGenerator {
|
|
m := NewMockGenerator(gomock.NewController(t))
|
|
m.EXPECT().Next().Return(id, nil)
|
|
return m
|
|
}
|
|
|
|
func NewIDGeneratorExpectError(t *testing.T, err error) *MockGenerator {
|
|
m := NewMockGenerator(gomock.NewController(t))
|
|
m.EXPECT().Next().Return("", err)
|
|
return m
|
|
}
|