Function TestAddPodSvcNodeSameIP

pkg/controllers/cache/cache_test.go:184–237

(t *testing.T)

}

func TestAddPodSvcNodeSameIP(t *testing.T) {
	log.SetupZapLogger(log.GetDefaultLogOpts())
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()
	p := pubsub.NewMockPubSubInterface(ctrl)
	var wg sync.WaitGroup
	wg.Add(5) // 2 pod + 2 svc + 1 node publishes
	doFn := func(pubsub.PubSubTopic, interface{}) { wg.Done() }
	p.EXPECT().Publish(common.PubSubPods, gomock.Any()).Times(2).Do(doFn)
	p.EXPECT().Publish(common.PubSubSvc, gomock.Any()).Times(2).Do(doFn)
	p.EXPECT().Publish(common.PubSubNode, gomock.Any()).Times(1).Do(doFn)
	p.EXPECT().Subscribe(common.PubSubAPIServer, gomock.Any()).Times(1)
	c := New(p)
	assert.NotNil(t, c)

	addEndpoints := common.NewRetinaEndpoint("pod1", "ns1", nil)
	addEndpoints.SetLabels(map[string]string{
		"app": "app1",
	})

	addEndpoints.SetIPs(&common.IPAddresses{
		IPv4: net.IPv4(1, 2, 3, 4),
	})

	err := c.UpdateRetinaEndpoint(addEndpoints)
	assert.NoError(t, err)

	addSvc := common.NewRetinaSvc("svc1", "ns1",
		&common.IPAddresses{
			IPv4: net.IPv4(1, 2, 3, 4),
		}, nil, nil)

	err = c.UpdateRetinaSvc(addSvc)
	assert.NoError(t, err)

	obj := c.GetObjByIP("1.2.3.4")
	assert.NotNil(t, obj)
	svc := obj.(*common.RetinaSvc)
	assert.Equal(t, addSvc.Name(), svc.Name())
	assert.Equal(t, addSvc.Namespace(), svc.Namespace())

	addNode := common.NewRetinaNode("node1", net.IPv4(1, 2, 3, 4), "zone-1")

	err = c.UpdateRetinaNode(addNode)
	assert.NoError(t, err)

	obj = c.GetObjByIP("1.2.3.4")
	assert.NotNil(t, obj)
	node := obj.(*common.RetinaNode)
	assert.Equal(t, addNode.Name(), node.Name())
	assert.Equal(t, addNode.IPString(), node.IPString())

	wg.Wait()
}

func TestAddPodSvcNodeSameIPDiffNS(t *testing.T) {
	log.SetupZapLogger(log.GetDefaultLogOpts())