mirror of
https://github.com/zitadel/zitadel.git
synced 2025-01-10 13:53:39 +00:00
88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
//go:build integration
|
|
|
|
package handlers_test
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io"
|
|
"net"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/zitadel/zitadel/pkg/grpc/management"
|
|
"github.com/zitadel/zitadel/pkg/grpc/system"
|
|
)
|
|
|
|
func TestServer_TelemetryPushMilestones(t *testing.T) {
|
|
primaryDomain, instanceID, systemUserCTX, iamOwnerCtx := Tester.UseIsolatedInstance(CTX)
|
|
bodies := make(chan []byte, 0)
|
|
t.Log("testing against instance with primary domain", primaryDomain)
|
|
mockServer := httptest.NewUnstartedServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
body, err := io.ReadAll(r.Body)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if r.Header.Get("single-value") != "single-value" {
|
|
t.Error("single-value header not set")
|
|
}
|
|
if reflect.DeepEqual(r.Header.Get("multi-value"), "multi-value-1,multi-value-2") {
|
|
t.Error("single-value header not set")
|
|
}
|
|
bodies <- body
|
|
w.WriteHeader(http.StatusOK)
|
|
}))
|
|
listener, err := net.Listen("tcp", "localhost:8081")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
mockServer.Listener = listener
|
|
mockServer.Start()
|
|
t.Cleanup(mockServer.Close)
|
|
awaitMilestone(t, bodies, primaryDomain, "InstanceCreated")
|
|
project, err := MgmtClient.AddProject(iamOwnerCtx, &management.AddProjectRequest{Name: "integration"})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
awaitMilestone(t, bodies, primaryDomain, "ProjectCreated")
|
|
if _, err = MgmtClient.AddOIDCApp(iamOwnerCtx, &management.AddOIDCAppRequest{
|
|
ProjectId: project.GetId(),
|
|
Name: "integration",
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
awaitMilestone(t, bodies, primaryDomain, "ApplicationCreated")
|
|
if _, err = SystemClient.RemoveInstance(systemUserCTX, &system.RemoveInstanceRequest{InstanceId: instanceID}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
awaitMilestone(t, bodies, primaryDomain, "InstanceDeleted")
|
|
}
|
|
|
|
func awaitMilestone(t *testing.T, bodies chan []byte, primaryDomain, expectMilestoneType string) {
|
|
for {
|
|
select {
|
|
case body := <-bodies:
|
|
plain := new(bytes.Buffer)
|
|
if err := json.Indent(plain, body, "", " "); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log("received milestone", plain.String())
|
|
milestone := struct {
|
|
Type string
|
|
PrimaryDomain string
|
|
}{}
|
|
if err := json.Unmarshal(body, &milestone); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if milestone.Type == expectMilestoneType && milestone.PrimaryDomain == primaryDomain {
|
|
return
|
|
}
|
|
case <-time.After(60 * time.Second):
|
|
t.Fatalf("timed out waiting for milestone %s in domain %s", expectMilestoneType, primaryDomain)
|
|
}
|
|
}
|
|
}
|