Skip to content

Commit a85036c

Browse files
committed
add CreateArangodClient and version up check
1 parent c9dc159 commit a85036c

File tree

1 file changed

+13
-18
lines changed

1 file changed

+13
-18
lines changed

tests/deployments_test.go

Lines changed: 13 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ import (
3030
driver "github.com/arangodb/go-driver"
3131
api "github.com/arangodb/kube-arangodb/pkg/apis/deployment/v1alpha"
3232
kubeArangoClient "github.com/arangodb/kube-arangodb/pkg/client"
33+
arangod "github.com/arangodb/kube-arangodb/pkg/util/arangod"
3334
)
3435

3536
// TODO - environements (provided from outside)
@@ -113,34 +114,28 @@ func deploymentSubTest(t *testing.T, mode api.DeploymentMode, engine api.Storage
113114
t.Fatalf("Single Server not running in time: %v", err)
114115
}
115116

116-
// FIXME - waitUntilResilientHealth
117117
members := deployment.Status.Members
118-
119118
singles := members.Single
120119
agents := members.Agents
121-
servers := append(append(api.MemberStatusList{}, singles...), agents...)
122120

123121
if len(singles) != 2 || len(agents) != 3 {
124122
t.Fatal("Wrong number of servers: single %v - agents %v", len(singles), len(agents))
125123
}
126124

127-
// FIXMcase s
128-
// - create dbconnection to each of the servers
129-
// - run api version on connection
130-
131-
//t.Fatal("This test is not fully implemented!")
132-
133-
// TEST CODE BELOW
134-
t.Log("##################################")
135-
if len(agents) == 0 {
136-
t.Fatal("This setup requires agents to work!")
125+
for _, agent := range agents {
126+
dbclient, err := arangod.CreateArangodClient(ctx, k8sClient.CoreV1(), deployment, api.ServerGroupAgents, agent.ID)
127+
if err != nil {
128+
t.Fatal("Unable to create connection to: %v", agent.ID)
129+
}
130+
waitUntilVersionUp(dbclient)
137131
}
138-
139-
t.Logf("agents len: %v", len(agents))
140-
for _, server := range servers {
141-
t.Logf("Server: %v ", server)
132+
for _, single := range singles {
133+
dbclient, err := arangod.CreateArangodClient(ctx, k8sClient.CoreV1(), deployment, api.ServerGroupAgents, single.ID)
134+
if err != nil {
135+
t.Fatal("Unable to create connection to: %v", single.ID)
136+
}
137+
waitUntilVersionUp(dbclient)
142138
}
143-
t.Log("##################################")
144139
default:
145140
t.Fatalf("DeploymentMode %v is not supported!", mode)
146141
}

0 commit comments

Comments
 (0)