mirror of
https://github.com/openziti/ziti.git
synced 2026-10-08 13:51:22 +00:00
Paul Lorenz
85 lines
2.3 KiB
Go
85 lines
2.3 KiB
Go
package tests
|
|
|
|
import (
|
|
rest_identity "github.com/openziti/edge-api/rest_management_api_client/identity"
|
|
"github.com/openziti/edge-api/rest_model"
|
|
edge_apis "github.com/openziti/sdk-golang/edge-apis"
|
|
"github.com/openziti/sdk-golang/ziti"
|
|
"github.com/openziti/ziti/controller"
|
|
"net/url"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func Test_Identity_HasErConnection(t *testing.T) {
|
|
ctx := NewTestContext(t)
|
|
defer ctx.Teardown()
|
|
ctx.StartServer()
|
|
ctx.RequireAdminManagementApiLogin()
|
|
|
|
ctx.CreateEnrollAndStartEdgeRouter()
|
|
|
|
service := ctx.AdminManagementSession.RequireNewServiceAccessibleToAll("smartrouting")
|
|
|
|
sdkIdentity, context := ctx.AdminManagementSession.RequireCreateSdkContext()
|
|
defer context.Close()
|
|
|
|
listener, err := context.Listen(service.Name)
|
|
ctx.Req.NoError(err)
|
|
|
|
isRunning := true
|
|
|
|
defer func() {
|
|
isRunning = false
|
|
ctx.Req.NoError(listener.Close())
|
|
}()
|
|
managementStr := "https://" + ctx.ApiHost + controller.ManagementRestApiBaseUrlV1
|
|
managementUrl, err := url.Parse(managementStr)
|
|
ctx.Req.NoError(err)
|
|
|
|
creds := edge_apis.NewUpdbCredentials(ctx.AdminAuthenticator.Username, ctx.AdminAuthenticator.Password)
|
|
|
|
caPool, err := ziti.GetControllerWellKnownCaPool("https://" + ctx.ApiHost)
|
|
ctx.Req.NoError(err)
|
|
|
|
managementClient := edge_apis.NewManagementApiClient([]*url.URL{managementUrl}, caPool, func(strings chan string) {
|
|
strings <- "123"
|
|
})
|
|
|
|
curSession, err := managementClient.Authenticate(creds, nil)
|
|
ctx.Req.NoError(err)
|
|
ctx.Req.NotNil(curSession)
|
|
|
|
result := make(chan *rest_model.IdentityDetail)
|
|
detailIdentityParams := rest_identity.NewDetailIdentityParams()
|
|
detailIdentityParams.ID = sdkIdentity.Id
|
|
|
|
//HasEdgeRouterConnection can take up to the minimum heartbeat interval (default 60s, configured in tests for 10s)
|
|
//Check every 1s for an update
|
|
go func() {
|
|
for isRunning {
|
|
resp, err := managementClient.API.Identity.DetailIdentity(detailIdentityParams, nil)
|
|
|
|
ctx.Req.NoError(err)
|
|
ctx.NotNil(resp)
|
|
|
|
if *resp.Payload.Data.HasAPISession && *resp.Payload.Data.HasEdgeRouterConnection {
|
|
result <- resp.Payload.Data
|
|
return
|
|
}
|
|
|
|
time.Sleep(1 * time.Second)
|
|
}
|
|
}()
|
|
|
|
//Should receive a valid result no later than ~10s later based on the heartbeat interval.
|
|
select {
|
|
case id := <-result:
|
|
ctx.Req.True(*id.HasAPISession)
|
|
ctx.Req.True(*id.HasEdgeRouterConnection)
|
|
case <-time.After(15 * time.Second):
|
|
ctx.Fail("timed out")
|
|
}
|
|
|
|
}
|