Function TestHandlerDownsideLabels

engine/msg_handler_test.go:195–317

(t *testing.T)

}

func TestHandlerDownsideLabels(t *testing.T) {
	e, ami, _ := genHandlerDownsideEngine(t)
	h := &handlerDownside{engineImpl: e}

	// test
	ch, err := e.pb.Subscribe(sync.TopicUpside)
	assert.NoError(t, err)
	assert.NotNil(t, ch)

	handler := &msgLabelUpside{t: t}
	pro := pubsub.NewProcessor(ch, 0, handler)
	pro.Start()

	// msg 8 update node label success
	engMsgWG.Add(1)
	msg8 := &specV1.Message{
		Kind: specV1.MessageCMD,
		Metadata: map[string]string{
			"cmd":     specV1.MessageCommandNodeLabel,
			"subName": "node01",
		},
		Content: specV1.LazyValue{
			Value: map[string]string{
				"beta.kubernetes.io/arch":        "amd64",
				"beta.kubernetes.io/os":          "linux",
				"kubernetes.io/arch":             "amd64",
				"kubernetes.io/hostname":         "docker-desktop",
				"kubernetes.io/os":               "linux",
				"node-role.kubernetes.io/master": "",
				"a":                              "b",
			},
		},
	}
	ami.EXPECT().UpdateNodeLabels("node01", gomock.Any()).Return(nil).Times(1)
	handler.check = func(msg interface{}) {
		m, ok := msg.(*specV1.Message)
		assert.True(t, ok)
		assert.Equal(t, "true", m.Metadata["success"])
		engMsgWG.Done()
	}
	err = h.OnMessage(msg8)
	assert.NoError(t, err)
	engMsgWG.Wait()

	// msg 9 update node label err sub name
	engMsgWG.Add(1)
	msg9 := &specV1.Message{
		Kind: specV1.MessageCMD,
		Metadata: map[string]string{
			"cmd": specV1.MessageCommandNodeLabel,
		},
	}
	handler.check = func(msg interface{}) {
		m, ok := msg.(*specV1.Message)
		assert.True(t, ok)
		assert.Equal(t, "false", m.Metadata["success"])
		assert.Equal(t, ErrSubNodeName, m.Metadata["msg"])
		engMsgWG.Done()