func TestBroker_WillRunQueries(t *testing.T) {
// TODO fix the raciness in this test
t.Skip()
// this handler should just work
testHandler := &BrokerTestHandler{}
server := httptest.NewServer(testHandler)
defer server.Close()
// this will timeout on the trigger request
timeoutHandler := &BrokerTestHandler{wait: 1100 * time.Millisecond}
timeoutServer := httptest.NewServer(timeoutHandler)
defer timeoutServer.Close()
// this will return a 500
badHandler := &BrokerTestHandler{sendError: true}
badServer := httptest.NewServer(badHandler)
defer badServer.Close()
b := influxdb.NewBroker()
// set the trigger times and failure sleeps for the test
b.TriggerInterval = 2 * time.Millisecond
b.TriggerTimeout = 100 * time.Millisecond
b.TriggerFailurePause = 2 * time.Millisecond
f := tempfile()
defer os.Remove(f)
if err := b.Open(f, &url.URL{Host: "127.0.0.1:8080"}); err != nil {
t.Fatalf("error opening broker: %s", err)
}
if err := b.Initialize(); err != nil {
t.Fatalf("error initializing broker: %s", err)
}
defer b.Close()
// set the data nodes (replicas) so all the failure cases get hit first
if err := b.Broker.CreateReplica(1, &url.URL{Host: "127.0.0.1:8090"}); err != nil {
t.Fatalf("couldn't create replica %s", err.Error())
}
b.Broker.CreateReplica(2, &url.URL{Host: timeoutServer.URL[7:]})
b.Broker.CreateReplica(3, &url.URL{Host: badServer.URL[7:]})
b.Broker.CreateReplica(4, &url.URL{Host: server.URL[7:]})
b.RunContinuousQueryLoop()
// every failure and success case should be hit in this time frame
time.Sleep(1400 * time.Millisecond)
if timeoutHandler.requestCount != 1 {
t.Fatal("broker should have only sent 1 request to the server that times out.")
}
if badHandler.requestCount != 1 {
t.Fatal("broker should have only sent 1 request to the bad server. i.e. it didn't keep the state to make request to the good server")
}
if testHandler.requestCount < 1 || testHandler.processRequestCount < 1 {
t.Fatal("broker failed to send multiple continuous query requests to the data node")
}
}
// creates and initializes a broker.
func openBroker(path string, u *url.URL, initializing bool, joinURLs []*url.URL, w io.Writer) *influxdb.Broker {
// Create broker.
b := influxdb.NewBroker()
b.SetLogOutput(w)
if err := b.Open(path, u); err != nil {
log.Fatalf("failed to open broker: %s", err)
}
// If this is a new broker then we can initialize two ways:
// 1) Start a brand new cluster.
// 2) Join an existing cluster.
if initializing {
if len(joinURLs) == 0 {
initializeBroker(b)
} else {
joinBroker(b, joinURLs)
}
}
return b
}