1+ package e2e
2+
3+ import (
4+ "fmt"
5+ "io/ioutil"
6+ "net/http"
7+ "path/filepath"
8+ "strings"
9+ "testing"
10+ "time"
11+
12+ "github.com/gruntwork-io/terratest/modules/helm"
13+ "github.com/gruntwork-io/terratest/modules/k8s"
14+ "github.com/gruntwork-io/terratest/modules/random"
15+ digest_auth "github.com/xinsnake/go-http-digest-auth-client"
16+ "github.com/tidwall/gjson"
17+ )
18+
19+ func TestSeparateEDnode (t * testing.T ) {
20+ // Path to the helm chart we will test
21+ helmChartPath , e := filepath .Abs ("../../charts" )
22+ if (e != nil ) {
23+ t .Fatalf (e .Error ())
24+ }
25+ username := "admin"
26+ password := "admin"
27+ var resp * http.Response
28+ var body []byte
29+ var err error
30+
31+ namespaceName := "marklogic-" + strings .ToLower (random .UniqueId ())
32+ kubectlOptions := k8s .NewKubectlOptions ("" , "" , namespaceName )
33+ options := & helm.Options {
34+ KubectlOptions : kubectlOptions ,
35+ SetValues : map [string ]string {
36+ "persistence.enabled" : "false" ,
37+ "replicaCount" : "1" ,
38+ "image.repository" : "marklogic-centos/marklogic-server-centos" ,
39+ "image.tag" : "10-internal" ,
40+ "auth.adminUsername" : username ,
41+ "auth.adminPassword" : password ,
42+ "group.name" : "dnode" ,
43+ "logCollection.enabled" : "false" ,
44+ },
45+ }
46+
47+ t .Logf ("====Creating namespace: " + namespaceName )
48+ k8s .CreateNamespace (t , kubectlOptions , namespaceName )
49+
50+ defer t .Logf ("====Deleting namespace: " + namespaceName )
51+ defer k8s .DeleteNamespace (t , kubectlOptions , namespaceName )
52+
53+ dnodeReleaseName := "test-dnode-group"
54+ t .Logf ("====Installing Helm Chart" + dnodeReleaseName )
55+ helm .Install (t , options , helmChartPath , dnodeReleaseName )
56+
57+ podName := dnodeReleaseName + "-marklogic-0"
58+
59+ // wait until the pod is in Ready status
60+ k8s .WaitUntilPodAvailable (t , kubectlOptions , podName , 10 , 20 * time .Second )
61+
62+ time .Sleep (10 * time .Second )
63+ tunnel := k8s .NewTunnel (
64+ kubectlOptions , k8s .ResourceTypePod , podName , 8002 , 8002 )
65+ defer tunnel .Close ()
66+ tunnel .ForwardPort (t )
67+ hosts_endpoint := fmt .Sprintf ("http://%s/manage/v2/hosts?format=json" , tunnel .Endpoint ())
68+ t .Logf (`Endpoint: %s` , hosts_endpoint )
69+
70+ dr := digest_auth .NewRequest (username , password , "GET" , hosts_endpoint , "" )
71+
72+ if resp , err = dr .Execute (); err != nil {
73+ t .Fatalf (err .Error ())
74+ }
75+ defer resp .Body .Close ()
76+
77+ if body , err = ioutil .ReadAll (resp .Body ); err != nil {
78+ t .Fatalf (err .Error ())
79+ }
80+ t .Logf ("Response:\n " + string (body ))
81+ bootstrapHost := gjson .Get (string (body ), `host-default-list.list-items.list-item.#(roleref="bootstrap").nameref` )
82+ t .Logf (`BootstrapHost: = %s` , bootstrapHost )
83+
84+ // verify bootstrap host exists on the cluster
85+ if bootstrapHost .String () == "" {
86+ t .Errorf ("Bootstrap does not exists on cluster" )
87+ }
88+
89+ enodeOptions := & helm.Options {
90+ KubectlOptions : kubectlOptions ,
91+ SetValues : map [string ]string {
92+ "persistence.enabled" : "false" ,
93+ "replicaCount" : "2" ,
94+ "image.repository" : "marklogic-centos/marklogic-server-centos" ,
95+ "image.tag" : "10-internal" ,
96+ "auth.adminUsername" : username ,
97+ "auth.adminPassword" : password ,
98+ "group.name" : "enode" ,
99+ "bootstrapHostName" : bootstrapHost .String (),
100+ "logCollection.enabled" : "false" ,
101+ },
102+ }
103+ enodeReleaseName := "test-enode-group"
104+ t .Logf ("====Installing Helm Chart " + enodeReleaseName )
105+ helm .Install (t , enodeOptions , helmChartPath , enodeReleaseName )
106+
107+ enodePodName - 0 := enodeReleaseName + "-marklogic-0"
108+
109+ // wait until the first enode pod is in Ready status
110+ k8s .WaitUntilPodAvailable (t , kubectlOptions , enodePodName - 0 , 15 , 20 * time .Second )
111+
112+ group_endpoint := fmt .Sprintf ("http://%s/manage/v2/groups" , tunnel .Endpoint ())
113+ t .Logf (`Endpoint: %s` , group_endpoint )
114+
115+ dr_groups := digest_auth .NewRequest (username , password , "GET" , group_endpoint , "" )
116+
117+ if resp , err = dr_groups .Execute (); err != nil {
118+ t .Fatalf (err .Error ())
119+ }
120+ defer resp .Body .Close ()
121+
122+ if body , err = ioutil .ReadAll (resp .Body ); err != nil {
123+ t .Fatalf (err .Error ())
124+ }
125+ t .Logf ("Response:\n " + string (body ))
126+
127+ // verify groups dnode, enode exists on the cluster
128+ if ! strings .Contains (string (body ), "<nameref>dnode</nameref>" ) && ! strings .Contains (string (body ), "<nameref>enode</nameref>" ) {
129+ t .Errorf ("Groups does not exists on cluster" )
130+ }
131+
132+ enodePodName - 1 := enodeReleaseName + "-marklogic-1"
133+
134+ // wait until the second enode pod is in Ready status
135+ k8s .WaitUntilPodAvailable (t , kubectlOptions , enodePodName - 1 , 15 , 20 * time .Second )
136+
137+ enode_endpoint := fmt .Sprintf ("http://%s/manage/v2/groups/enode?format=json" , tunnel .Endpoint ())
138+ t .Logf (`Endpoint: %s` , enode_endpoint )
139+
140+ dr_enode := digest_auth .NewRequest (username , password , "GET" , enode_endpoint , "" )
141+
142+ if resp , err = dr_enode .Execute (); err != nil {
143+ t .Fatalf (err .Error ())
144+ }
145+ defer resp .Body .Close ()
146+
147+ if body , err = ioutil .ReadAll (resp .Body ); err != nil {
148+ t .Fatalf (err .Error ())
149+ }
150+ t .Logf ("Response:\n " + string (body ))
151+
152+ enode_host_count := gjson .Get (string (body ), `group-default.relations.relation-group.#(typeref="hosts").relation-count.value` )
153+ t .Logf (`enode_host_count: = %s` , enode_host_count )
154+
155+ // verify bootstrap host exists on the cluster
156+ if ! strings .Contains (enode_host_count .String (),"2" ) {
157+ t .Errorf ("enode hosts does not exists on cluster" )
158+ }
159+ }
0 commit comments