vsphere-influxdb-go/vendor/github.com/influxdata/influxdb/stress/v2/stress_client/commune_test.go

package stressClient

import (
	"testing"
)

func TestCommunePoint(t *testing.T) {
	comm := newCommune(5)
	pt := "write,tag=tagVal fooField=5 1460912595"
	comm.ch <- pt
	point := comm.point("s")
	if string(point.Name()) != "write" {
		t.Errorf("expected: write\ngot: %v", string(point.Name()))
	}
	if point.Tags().GetString("tag") != "tagVal" {
		t.Errorf("expected: tagVal\ngot: %v", point.Tags().GetString("tag"))
	}
	fields, err := point.Fields()
	if err != nil {
		t.Fatal(err)
	}
	if int(fields["fooField"].(float64)) != 5 {
		t.Errorf("expected: 5\ngot: %v\n", fields["fooField"])
	}
	// Make sure commune returns the prev point
	comm.ch <- ""
	point = comm.point("s")
	if string(point.Name()) != "write" {
		t.Errorf("expected: write\ngot: %v", string(point.Name()))
	}
	if point.Tags().GetString("tag") != "tagVal" {
		t.Errorf("expected: tagVal\ngot: %v", point.Tags().GetString("tag"))
	}
	if int(fields["fooField"].(float64)) != 5 {
		t.Errorf("expected: 5\ngot: %v\n", fields["fooField"])
	}
}

func TestSetCommune(t *testing.T) {
	sf, _, _ := NewTestStressTest()
	ch := sf.SetCommune("foo_name")
	ch <- "write,tag=tagVal fooField=5 1460912595"
	pt := sf.GetPoint("foo_name", "s")
	if string(pt.Name()) != "write" {
		t.Errorf("expected: write\ngot: %v", string(pt.Name()))
	}
	if pt.Tags().GetString("tag") != "tagVal" {
		t.Errorf("expected: tagVal\ngot: %v", pt.Tags().GetString("tag"))
	}
	fields, err := pt.Fields()
	if err != nil {
		t.Fatal(err)
	}
	if int(fields["fooField"].(float64)) != 5 {
		t.Errorf("expected: 5\ngot: %v\n", fields["fooField"])
	}
}