github.com/cilium/cilium@v1.16.2/pkg/bgpv1/mock/mocks.go (about)

     1  // SPDX-License-Identifier: Apache-2.0
     2  // Copyright Authors of Cilium
     3  
     4  package mock
     5  
     6  import (
     7  	"context"
     8  
     9  	v1 "k8s.io/api/core/v1"
    10  	k8sLabels "k8s.io/apimachinery/pkg/labels"
    11  	v1listers "k8s.io/client-go/listers/core/v1"
    12  
    13  	"github.com/cilium/cilium/api/v1/models"
    14  	restapi "github.com/cilium/cilium/api/v1/server/restapi/bgp"
    15  	"github.com/cilium/cilium/pkg/bgpv1/agent"
    16  	v2 "github.com/cilium/cilium/pkg/k8s/apis/cilium.io/v2"
    17  	"github.com/cilium/cilium/pkg/k8s/apis/cilium.io/v2alpha1"
    18  )
    19  
    20  var _ v1listers.NodeLister = (*MockNodeLister)(nil)
    21  
    22  type MockNodeLister struct {
    23  	List_ func(selector k8sLabels.Selector) (ret []*v1.Node, err error)
    24  	Get_  func(name string) (*v1.Node, error)
    25  	v1listers.NodeListerExpansion
    26  }
    27  
    28  func (m *MockNodeLister) List(selector k8sLabels.Selector) (ret []*v1.Node, err error) {
    29  	return m.List_(selector)
    30  }
    31  
    32  func (m *MockNodeLister) Get(name string) (*v1.Node, error) {
    33  	return m.Get_(name)
    34  }
    35  
    36  var _ agent.BGPRouterManager = (*MockBGPRouterManager)(nil)
    37  
    38  type MockBGPRouterManager struct {
    39  	ConfigurePeers_     func(ctx context.Context, policy *v2alpha1.CiliumBGPPeeringPolicy, ciliumNode *v2.CiliumNode) error
    40  	ReconcileInstances_ func(ctx context.Context, bgpnc *v2alpha1.CiliumBGPNodeConfig, ciliumNode *v2.CiliumNode) error
    41  	GetPeers_           func(ctx context.Context) ([]*models.BgpPeer, error)
    42  	GetRoutes_          func(ctx context.Context, params restapi.GetBgpRoutesParams) ([]*models.BgpRoute, error)
    43  	GetRoutePolicies_   func(ctx context.Context, params restapi.GetBgpRoutePoliciesParams) ([]*models.BgpRoutePolicy, error)
    44  	Stop_               func()
    45  }
    46  
    47  func (m *MockBGPRouterManager) ConfigurePeers(ctx context.Context, policy *v2alpha1.CiliumBGPPeeringPolicy, ciliumNode *v2.CiliumNode) error {
    48  	return m.ConfigurePeers_(ctx, policy, ciliumNode)
    49  }
    50  
    51  func (m *MockBGPRouterManager) ReconcileInstances(ctx context.Context, bgpnc *v2alpha1.CiliumBGPNodeConfig, ciliumNode *v2.CiliumNode) error {
    52  	return m.ReconcileInstances_(ctx, bgpnc, ciliumNode)
    53  }
    54  
    55  func (m *MockBGPRouterManager) GetPeers(ctx context.Context) ([]*models.BgpPeer, error) {
    56  	return m.GetPeers_(ctx)
    57  }
    58  
    59  func (m *MockBGPRouterManager) GetRoutes(ctx context.Context, params restapi.GetBgpRoutesParams) ([]*models.BgpRoute, error) {
    60  	return m.GetRoutes_(ctx, params)
    61  }
    62  
    63  func (m *MockBGPRouterManager) GetRoutePolicies(ctx context.Context, params restapi.GetBgpRoutePoliciesParams) ([]*models.BgpRoutePolicy, error) {
    64  	return m.GetRoutePolicies_(ctx, params)
    65  }
    66  
    67  func (m *MockBGPRouterManager) Stop() {
    68  	m.Stop_()
    69  }