42 lines
849 B
Go
42 lines
849 B
Go
|
package enterprise_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/influxdata/enterprise-client/v2"
|
||
|
|
"github.com/influxdata/telegraf/services/enterprise"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_RegistersWithEnterprise(t *testing.T) {
|
||
|
|
success := make(chan struct{})
|
||
|
|
mockEnterprise := http.NewServeMux()
|
||
|
|
mockEnterprise.HandleFunc("/api/v2/products", func(rw http.ResponseWriter, r *http.Request) {
|
||
|
|
if r.Method == "POST" {
|
||
|
|
close(success)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
srv := httptest.NewServer(mockEnterprise)
|
||
|
|
defer srv.Close()
|
||
|
|
|
||
|
|
c := enterprise.Config{
|
||
|
|
Hosts: []*client.Host{
|
||
|
|
&client.Host{URL: srv.URL},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
e := enterprise.NewEnterprise(c)
|
||
|
|
e.Open()
|
||
|
|
|
||
|
|
timeout := time.After(50 * time.Millisecond)
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case <-success:
|
||
|
|
return
|
||
|
|
case <-timeout:
|
||
|
|
t.Fatal("Expected to receive call to Enterprise API, but received none")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|