From 44730d4e705b7274ed22dde74b076a4f3d9b2d60 Mon Sep 17 00:00:00 2001 From: yunlzheng Date: Fri, 6 Mar 2020 23:12:07 +0800 Subject: [PATCH] improve: generate mock for shadow and kubernetes, add test case for connect. --- pkg/mockd/mock/connect_mock.go | 62 +++++++++++++++++ pkg/mockd/mock/kubernetes_mock.go | 109 ++++++++++++++++++++++++++++++ 2 files changed, 171 insertions(+) create mode 100644 pkg/mockd/mock/connect_mock.go create mode 100644 pkg/mockd/mock/kubernetes_mock.go diff --git a/pkg/mockd/mock/connect_mock.go b/pkg/mockd/mock/connect_mock.go new file mode 100644 index 0000000..766f6ce --- /dev/null +++ b/pkg/mockd/mock/connect_mock.go @@ -0,0 +1,62 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: ../kt/connect/types.go + +// Package mock is a generated GoMock package. +package mock + +import ( + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// MockShadowInterface is a mock of ShadowInterface interface +type MockShadowInterface struct { + ctrl *gomock.Controller + recorder *MockShadowInterfaceMockRecorder +} + +// MockShadowInterfaceMockRecorder is the mock recorder for MockShadowInterface +type MockShadowInterfaceMockRecorder struct { + mock *MockShadowInterface +} + +// NewMockShadowInterface creates a new mock instance +func NewMockShadowInterface(ctrl *gomock.Controller) *MockShadowInterface { + mock := &MockShadowInterface{ctrl: ctrl} + mock.recorder = &MockShadowInterfaceMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use +func (m *MockShadowInterface) EXPECT() *MockShadowInterfaceMockRecorder { + return m.recorder +} + +// Inbound mocks base method +func (m *MockShadowInterface) Inbound(exposePort, podName, remoteIP string) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Inbound", exposePort, podName, remoteIP) + ret0, _ := ret[0].(error) + return ret0 +} + +// Inbound indicates an expected call of Inbound +func (mr *MockShadowInterfaceMockRecorder) Inbound(exposePort, podName, remoteIP interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Inbound", reflect.TypeOf((*MockShadowInterface)(nil).Inbound), exposePort, podName, remoteIP) +} + +// Outbound mocks base method +func (m *MockShadowInterface) Outbound(name, podIP string, cidrs []string) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Outbound", name, podIP, cidrs) + ret0, _ := ret[0].(error) + return ret0 +} + +// Outbound indicates an expected call of Outbound +func (mr *MockShadowInterfaceMockRecorder) Outbound(name, podIP, cidrs interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Outbound", reflect.TypeOf((*MockShadowInterface)(nil).Outbound), name, podIP, cidrs) +} diff --git a/pkg/mockd/mock/kubernetes_mock.go b/pkg/mockd/mock/kubernetes_mock.go new file mode 100644 index 0000000..4206dd0 --- /dev/null +++ b/pkg/mockd/mock/kubernetes_mock.go @@ -0,0 +1,109 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: ../kt/cluster/types.go + +// Package mock is a generated GoMock package. +package mock + +import ( + reflect "reflect" + + gomock "github.com/golang/mock/gomock" + v1 "k8s.io/api/apps/v1" +) + +// MockKubernetesInterface is a mock of KubernetesInterface interface +type MockKubernetesInterface struct { + ctrl *gomock.Controller + recorder *MockKubernetesInterfaceMockRecorder +} + +// MockKubernetesInterfaceMockRecorder is the mock recorder for MockKubernetesInterface +type MockKubernetesInterfaceMockRecorder struct { + mock *MockKubernetesInterface +} + +// NewMockKubernetesInterface creates a new mock instance +func NewMockKubernetesInterface(ctrl *gomock.Controller) *MockKubernetesInterface { + mock := &MockKubernetesInterface{ctrl: ctrl} + mock.recorder = &MockKubernetesInterfaceMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use +func (m *MockKubernetesInterface) EXPECT() *MockKubernetesInterfaceMockRecorder { + return m.recorder +} + +// Deployment mocks base method +func (m *MockKubernetesInterface) Deployment(name, namespace string) (*v1.Deployment, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Deployment", name, namespace) + ret0, _ := ret[0].(*v1.Deployment) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Deployment indicates an expected call of Deployment +func (mr *MockKubernetesInterfaceMockRecorder) Deployment(name, namespace interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Deployment", reflect.TypeOf((*MockKubernetesInterface)(nil).Deployment), name, namespace) +} + +// Scale mocks base method +func (m *MockKubernetesInterface) Scale(deployment *v1.Deployment, replicas *int32) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Scale", deployment, replicas) + ret0, _ := ret[0].(error) + return ret0 +} + +// Scale indicates an expected call of Scale +func (mr *MockKubernetesInterfaceMockRecorder) Scale(deployment, replicas interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Scale", reflect.TypeOf((*MockKubernetesInterface)(nil).Scale), deployment, replicas) +} + +// ServiceHosts mocks base method +func (m *MockKubernetesInterface) ServiceHosts(namespace string) map[string]string { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ServiceHosts", namespace) + ret0, _ := ret[0].(map[string]string) + return ret0 +} + +// ServiceHosts indicates an expected call of ServiceHosts +func (mr *MockKubernetesInterfaceMockRecorder) ServiceHosts(namespace interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServiceHosts", reflect.TypeOf((*MockKubernetesInterface)(nil).ServiceHosts), namespace) +} + +// ClusterCrids mocks base method +func (m *MockKubernetesInterface) ClusterCrids(podCIDR string) ([]string, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ClusterCrids", podCIDR) + ret0, _ := ret[0].([]string) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ClusterCrids indicates an expected call of ClusterCrids +func (mr *MockKubernetesInterfaceMockRecorder) ClusterCrids(podCIDR interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClusterCrids", reflect.TypeOf((*MockKubernetesInterface)(nil).ClusterCrids), podCIDR) +} + +// CreateShadow mocks base method +func (m *MockKubernetesInterface) CreateShadow(name, namespace, image string, labels map[string]string) (string, string, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateShadow", name, namespace, image, labels) + ret0, _ := ret[0].(string) + ret1, _ := ret[1].(string) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// CreateShadow indicates an expected call of CreateShadow +func (mr *MockKubernetesInterfaceMockRecorder) CreateShadow(name, namespace, image, labels interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateShadow", reflect.TypeOf((*MockKubernetesInterface)(nil).CreateShadow), name, namespace, image, labels) +}