mirror of https://github.com/knative/client.git
120 lines
4.5 KiB
Go
120 lines
4.5 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 v1alpha2
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"knative.dev/eventing/pkg/apis/sources/v1alpha2"
|
|
|
|
"knative.dev/client/pkg/util/mock"
|
|
)
|
|
|
|
type MockKnPingSourceClient struct {
|
|
t *testing.T
|
|
recorder *PingSourcesRecorder
|
|
namespace string
|
|
}
|
|
|
|
// NewMockKnPingSourceClient returns a new mock instance which you need to record for
|
|
func NewMockKnPingSourceClient(t *testing.T, ns ...string) *MockKnPingSourceClient {
|
|
namespace := "default"
|
|
if len(ns) > 0 {
|
|
namespace = ns[0]
|
|
}
|
|
return &MockKnPingSourceClient{
|
|
t: t,
|
|
recorder: &PingSourcesRecorder{mock.NewRecorder(t, namespace)},
|
|
}
|
|
}
|
|
|
|
// Ensure that the interface is implemented
|
|
var _ KnPingSourcesClient = &MockKnPingSourceClient{}
|
|
|
|
// recorder for service
|
|
type PingSourcesRecorder struct {
|
|
r *mock.Recorder
|
|
}
|
|
|
|
// Recorder returns the recorder for registering API calls
|
|
func (c *MockKnPingSourceClient) Recorder() *PingSourcesRecorder {
|
|
return c.recorder
|
|
}
|
|
|
|
// Namespace of this client
|
|
func (c *MockKnPingSourceClient) Namespace() string {
|
|
return c.recorder.r.Namespace()
|
|
}
|
|
|
|
// CreatePingSource records a call for CreatePingSource with the expected error
|
|
func (sr *PingSourcesRecorder) CreatePingSource(pingSource interface{}, err error) {
|
|
sr.r.Add("CreatePingSource", []interface{}{pingSource}, []interface{}{err})
|
|
}
|
|
|
|
// CreatePingSource performs a previously recorded action, failing if non has been registered
|
|
func (c *MockKnPingSourceClient) CreatePingSource(pingSource *v1alpha2.PingSource) error {
|
|
call := c.recorder.r.VerifyCall("CreatePingSource", pingSource)
|
|
return mock.ErrorOrNil(call.Result[0])
|
|
}
|
|
|
|
// GetPingSource records a call for GetPingSource with the expected object or error. Either pingsource or err should be nil
|
|
func (sr *PingSourcesRecorder) GetPingSource(name interface{}, pingSource *v1alpha2.PingSource, err error) {
|
|
sr.r.Add("GetPingSource", []interface{}{name}, []interface{}{pingSource, err})
|
|
}
|
|
|
|
// GetPingSource performs a previously recorded action, failing if non has been registered
|
|
func (c *MockKnPingSourceClient) GetPingSource(name string) (*v1alpha2.PingSource, error) {
|
|
call := c.recorder.r.VerifyCall("GetPingSource", name)
|
|
return call.Result[0].(*v1alpha2.PingSource), mock.ErrorOrNil(call.Result[1])
|
|
}
|
|
|
|
// UpdatePingSource records a call for UpdatePingSource with the expected error (nil if none)
|
|
func (sr *PingSourcesRecorder) UpdatePingSource(pingSource interface{}, err error) {
|
|
sr.r.Add("UpdatePingSource", []interface{}{pingSource}, []interface{}{err})
|
|
}
|
|
|
|
// UpdatePingSource performs a previously recorded action, failing if non has been registered
|
|
func (c *MockKnPingSourceClient) UpdatePingSource(pingSource *v1alpha2.PingSource) error {
|
|
call := c.recorder.r.VerifyCall("UpdatePingSource", pingSource)
|
|
return mock.ErrorOrNil(call.Result[0])
|
|
}
|
|
|
|
// UpdatePingSource records a call for DeletePingSource with the expected error (nil if none)
|
|
func (sr *PingSourcesRecorder) DeletePingSource(name interface{}, err error) {
|
|
sr.r.Add("DeletePingSource", []interface{}{name}, []interface{}{err})
|
|
}
|
|
|
|
// DeletePingSource performs a previously recorded action, failing if non has been registered
|
|
func (c *MockKnPingSourceClient) DeletePingSource(name string) error {
|
|
call := c.recorder.r.VerifyCall("DeletePingSource", name)
|
|
return mock.ErrorOrNil(call.Result[0])
|
|
}
|
|
|
|
// ListPingSource records a call for ListPingSource with the expected error (nil if none)
|
|
func (sr *PingSourcesRecorder) ListPingSource(pingSourceList *v1alpha2.PingSourceList, err error) {
|
|
sr.r.Add("ListPingSource", []interface{}{}, []interface{}{pingSourceList, err})
|
|
}
|
|
|
|
// ListPingSource performs a previously recorded action, failing if non has been registered
|
|
func (c *MockKnPingSourceClient) ListPingSource() (*v1alpha2.PingSourceList, error) {
|
|
call := c.recorder.r.VerifyCall("ListPingSource")
|
|
return call.Result[0].(*v1alpha2.PingSourceList), mock.ErrorOrNil(call.Result[1])
|
|
}
|
|
|
|
// Validates validates whether every recorded action has been called
|
|
func (sr *PingSourcesRecorder) Validate() {
|
|
sr.r.CheckThatAllRecordedMethodsHaveBeenCalled()
|
|
}
|