mirror of https://github.com/knative/client.git
121 lines
4.3 KiB
Go
121 lines
4.3 KiB
Go
// Copyright © 2019 The Knative Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package v1alpha1
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"knative.dev/eventing/pkg/apis/eventing/v1alpha1"
|
|
|
|
"knative.dev/client/pkg/util/mock"
|
|
)
|
|
|
|
// MockKnEventingClient is a combine of test object and recorder
|
|
type MockKnEventingClient struct {
|
|
t *testing.T
|
|
recorder *EventingRecorder
|
|
namespace string
|
|
}
|
|
|
|
// NewMockKnEventingClient returns a new mock instance which you need to record for
|
|
func NewMockKnEventingClient(t *testing.T, ns ...string) *MockKnEventingClient {
|
|
namespace := "default"
|
|
if len(ns) > 0 {
|
|
namespace = ns[0]
|
|
}
|
|
return &MockKnEventingClient{
|
|
t: t,
|
|
recorder: &EventingRecorder{mock.NewRecorder(t, namespace)},
|
|
}
|
|
}
|
|
|
|
// Ensure that the interface is implemented
|
|
var _ KnEventingClient = &MockKnEventingClient{}
|
|
|
|
// EventingRecorder is recorder for eventing objects
|
|
type EventingRecorder struct {
|
|
r *mock.Recorder
|
|
}
|
|
|
|
// Recorder returns the recorder for registering API calls
|
|
func (c *MockKnEventingClient) Recorder() *EventingRecorder {
|
|
return c.recorder
|
|
}
|
|
|
|
// Namespace of this client
|
|
func (c *MockKnEventingClient) Namespace() string {
|
|
return c.recorder.r.Namespace()
|
|
}
|
|
|
|
// CreateTrigger records a call for CreateCronJobSource with the expected error
|
|
func (sr *EventingRecorder) CreateTrigger(trigger interface{}, err error) {
|
|
sr.r.Add("CreateTrigger", []interface{}{trigger}, []interface{}{err})
|
|
}
|
|
|
|
// CreateTrigger performs a previously recorded action
|
|
func (c *MockKnEventingClient) CreateTrigger(trigger *v1alpha1.Trigger) error {
|
|
call := c.recorder.r.VerifyCall("CreateTrigger", trigger)
|
|
return mock.ErrorOrNil(call.Result[0])
|
|
}
|
|
|
|
// GetTrigger records a call for GetTrigger with the expected object or error. Either trigger or err should be nil
|
|
func (sr *EventingRecorder) GetTrigger(name interface{}, trigger *v1alpha1.Trigger, err error) {
|
|
sr.r.Add("GetTrigger", []interface{}{name}, []interface{}{trigger, err})
|
|
}
|
|
|
|
// GetTrigger performs a previously recorded action
|
|
func (c *MockKnEventingClient) GetTrigger(name string) (*v1alpha1.Trigger, error) {
|
|
call := c.recorder.r.VerifyCall("GetTrigger", name)
|
|
return call.Result[0].(*v1alpha1.Trigger), mock.ErrorOrNil(call.Result[1])
|
|
}
|
|
|
|
// DeleteTrigger records a call for DeleteTrigger with the expected error (nil if none)
|
|
func (sr *EventingRecorder) DeleteTrigger(name interface{}, err error) {
|
|
sr.r.Add("DeleteTrigger", []interface{}{name}, []interface{}{err})
|
|
}
|
|
|
|
// DeleteTrigger performs a previously recorded action, failing if non has been registered
|
|
func (c *MockKnEventingClient) DeleteTrigger(name string) error {
|
|
call := c.recorder.r.VerifyCall("DeleteTrigger", name)
|
|
return mock.ErrorOrNil(call.Result[0])
|
|
}
|
|
|
|
// ListTriggers records a call for ListTriggers with the expected result and error (nil if none)
|
|
func (sr *EventingRecorder) ListTriggers(triggerList *v1alpha1.TriggerList, err error) {
|
|
sr.r.Add("ListTriggers", nil, []interface{}{triggerList, err})
|
|
}
|
|
|
|
// ListTriggers performs a previously recorded action
|
|
func (c *MockKnEventingClient) ListTriggers() (*v1alpha1.TriggerList, error) {
|
|
call := c.recorder.r.VerifyCall("ListTriggers")
|
|
return call.Result[0].(*v1alpha1.TriggerList), mock.ErrorOrNil(call.Result[1])
|
|
}
|
|
|
|
// UpdateTrigger records a call for ListTriggers with the expected result and error (nil if none)
|
|
func (sr *EventingRecorder) UpdateTrigger(trigger interface{}, err error) {
|
|
sr.r.Add("UpdateTrigger", []interface{}{trigger}, []interface{}{err})
|
|
}
|
|
|
|
// UpdateTrigger performs a previously recorded action
|
|
func (c *MockKnEventingClient) UpdateTrigger(trigger *v1alpha1.Trigger) error {
|
|
call := c.recorder.r.VerifyCall("UpdateTrigger")
|
|
return mock.ErrorOrNil(call.Result[0])
|
|
}
|
|
|
|
// Validate validates whether every recorded action has been called
|
|
func (sr *EventingRecorder) Validate() {
|
|
sr.r.CheckThatAllRecordedMethodsHaveBeenCalled()
|
|
}
|