Function TestUpdateNodeProperties

node/node_test.go:357–424

(t *testing.T)

}

func TestUpdateNodeProperties(t *testing.T) {
	f, err := os.CreateTemp("", t.Name())
	assert.NoError(t, err)
	assert.NotNil(t, f)
	fmt.Println("-->tempfile", f.Name())
	defer os.RemoveAll(f.Name())

	s, err := store.NewBoltHold(f.Name())
	assert.NoError(t, err)
	assert.NotNil(t, s)

	ss, err := NewNode(s)
	assert.NoError(t, err)
	assert.NotNil(t, ss)

	router := routing.New()
	router.Put("/node/properties", utils.Wrapper(ss.UpdateNodeProperties))
	go fasthttp.ListenAndServe(":50022", router.HandleRequest)
	time.Sleep(100 * time.Millisecond)

	delta := map[string]interface{}{}
	data, _ := json.Marshal(delta)
	client := &fasthttp.Client{}
	req := fasthttp.AcquireRequest()
	req.SetBody(data)
	resp := fasthttp.AcquireResponse()
	url := fmt.Sprintf("%s%s", "http://127.0.0.1:50022", "/node/properties")
	req.SetRequestURI(url)
	req.Header.SetMethod("PUT")
	err = client.Do(req, resp)
	assert.NoError(t, err)
	assert.Equal(t, resp.StatusCode(), 200)

	report := v1.Report{
		KeyNodeProps: map[string]interface{}{
			"a": "1",
		},
	}
	delta = map[string]interface{}{
		"a": "2",
		"b": "3",
	}
	expect := map[string]interface{}{
		"a": "2",
		"b": "3",
	}
	_, err = ss.Report(report, true)
	assert.NoError(t, err)

	data, _ = json.Marshal(delta)
	req2 := fasthttp.AcquireRequest()
	resp2 := fasthttp.AcquireResponse()
	req2.SetRequestURI(url)
	req2.SetBody(data)
	req2.Header.SetMethod("PUT")
	err = client.Do(req2, resp2)
	assert.NoError(t, err)
	assert.Equal(t, resp2.StatusCode(), 200)