From ad04cc9991a64f7b4e8b9806c61e65485d1e32eb Mon Sep 17 00:00:00 2001
From: Yichen Wang <18348405+Aiee@users.noreply.github.com>
Date: Tue, 21 Dec 2021 13:33:40 +0800
Subject: [PATCH] Update docker compose yaml (#172)

* Update docker compose yaml

* Fix ci
---
 client_test.go                            |  3 +++
 nebula-docker-compose/docker-compose.yaml | 15 +++++++++++++++
 2 files changed, 18 insertions(+)

diff --git a/client_test.go b/client_test.go
index 69c3c16d..16849b3a 100644
--- a/client_test.go
+++ b/client_test.go
@@ -1019,6 +1019,9 @@ func TestReconnect(t *testing.T) {
 	for i := 0; i < len(sessionList); i++ {
 		sessionList[i].Release()
 	}
+
+	// Wait for graphd to be up
+	time.Sleep(5 * time.Second)
 }
 
 func TestIpLookup(t *testing.T) {
diff --git a/nebula-docker-compose/docker-compose.yaml b/nebula-docker-compose/docker-compose.yaml
index b1bbfa41..8429d68d 100644
--- a/nebula-docker-compose/docker-compose.yaml
+++ b/nebula-docker-compose/docker-compose.yaml
@@ -410,5 +410,20 @@ services:
     cap_add:
       - SYS_PTRACE
 
+  console:
+    image: vesoft/nebula-console:nightly
+    entrypoint: ""
+    command: 
+      - sh
+      - -c
+      - |
+        sleep 3 &&
+        nebula-console -addr graphd -port 3699 -u root -p nebula -e 'ADD HOSTS "storaged0":44500,"storaged1":44500,"storaged2":44500' &&
+        sleep 36000
+    depends_on:
+      - graphd
+    networks:
+      - nebula-net
+
 networks:
   nebula-net: