|
| 1 | +package tests |
| 2 | + |
| 3 | +import ( |
| 4 | + "context" |
| 5 | + "testing" |
| 6 | + |
| 7 | + "github.com/arangodb/go-driver" |
| 8 | + api "github.com/arangodb/kube-arangodb/pkg/apis/deployment/v1alpha" |
| 9 | + "github.com/arangodb/kube-arangodb/pkg/client" |
| 10 | + "github.com/arangodb/kube-arangodb/pkg/util" |
| 11 | + "github.com/arangodb/kube-arangodb/pkg/util/arangod" |
| 12 | + "github.com/dchest/uniuri" |
| 13 | +) |
| 14 | + |
| 15 | +func TestIsVersionSet(t *testing.T) { |
| 16 | + longOrSkip(t) |
| 17 | + c := client.MustNewInCluster() |
| 18 | + kubecli := mustNewKubeClient(t) |
| 19 | + ns := getNamespace(t) |
| 20 | + |
| 21 | + expectedVersion := driver.Version("3.3.17") |
| 22 | + // Prepare deployment config |
| 23 | + depl := newDeployment("test-auth-sng-def-" + uniuri.NewLen(4)) |
| 24 | + depl.Spec.Mode = api.NewMode(api.DeploymentModeSingle) |
| 25 | + depl.Spec.SetDefaults(depl.GetName()) |
| 26 | + depl.Spec.Image = util.NewString("arangodb/arangodb:" + string(expectedVersion)) |
| 27 | + // Create deployment |
| 28 | + apiObject, err := c.DatabaseV1alpha().ArangoDeployments(ns).Create(depl) |
| 29 | + if err != nil { |
| 30 | + t.Fatalf("Create deployment failed: %v", err) |
| 31 | + } |
| 32 | + defer deferedCleanupDeployment(c, depl.GetName(), ns) |
| 33 | + // Wait for deployment to be ready |
| 34 | + depl, err = waitUntilDeployment(c, depl.GetName(), ns, deploymentIsReady()) |
| 35 | + if err != nil { |
| 36 | + t.Fatalf("Deployment not running in time: %v", err) |
| 37 | + } |
| 38 | + |
| 39 | + single := depl.Status.Members.Single |
| 40 | + if single == nil || len(single) == 0 { |
| 41 | + t.Fatalf("single member is empty") |
| 42 | + } |
| 43 | + |
| 44 | + if single[0].ArangoVersion.CompareTo(expectedVersion) != 0 { |
| 45 | + t.Fatalf("version %s has not been set for the single member status", expectedVersion) |
| 46 | + } |
| 47 | + |
| 48 | + // Create a database client |
| 49 | + ctx := arangod.WithRequireAuthentication(context.Background()) |
| 50 | + client := mustNewArangodDatabaseClient(ctx, kubecli, apiObject, t, nil) |
| 51 | + |
| 52 | + // Wait for single server available with a valid database version |
| 53 | + err = waitUntilVersionUp(client, func(version driver.VersionInfo) error { |
| 54 | + if version.Version.CompareTo(expectedVersion) != 0 { |
| 55 | + t.Fatalf("database version %s is not equal expected version %s", version.Version, version) |
| 56 | + } |
| 57 | + return nil |
| 58 | + }) |
| 59 | + if err != nil { |
| 60 | + t.Fatalf("Single server not running returning version in time: %v", err) |
| 61 | + } |
| 62 | + |
| 63 | + //Cleanup |
| 64 | + removeDeployment(c, depl.GetName(), ns) |
| 65 | +} |
0 commit comments