Add ShowSpaces for SessionPool

label.go

@@ -84,7 +84,10 @@ func (edge LabelSchema) BuildDropEdgeQL() string {
 	return q
 }
 
-// For internal use
 type LabelName struct {
 	Name string `nebula:"Name"`
 }
+
+type SpaceName struct {
+	Name string `nebula:"Name"`
+}

session_pool.go

@@ -287,6 +287,18 @@ func (pool *SessionPool) ExecuteAndCheck(q string) (*ResultSet, error) {
 	return rs, nil
 }
 
+func (pool *SessionPool) ShowSpaces() ([]SpaceName, error) {
+	rs, err := pool.ExecuteAndCheck("SHOW SPACES;")
+	if err != nil {
+		return nil, err
+	}
+
+	var names []SpaceName
+	rs.Scan(&names)
+
+	return names, nil
+}
+
 func (pool *SessionPool) ShowTags() ([]LabelName, error) {
 	rs, err := pool.ExecuteAndCheck("SHOW TAGS;")
 	if err != nil {

session_pool_test.go

@@ -385,6 +385,17 @@ func TestSessionPoolApplySchema(t *testing.T) {
 	}
 	defer sessionPool.Close()
 
+	spaces, err := sessionPool.ShowSpaces()
+	if err != nil {
+		t.Fatal(err)
+	}
+	assert.LessOrEqual(t, 1, len(spaces), "should have at least 1 space")
+	var spaceNames []string
+	for _, space := range spaces {
+		spaceNames = append(spaceNames, space.Name)
+	}
+	assert.Contains(t, spaceNames, "test_space_schema", "should have test_space_schema")
+
 	tagSchema := LabelSchema{
 		Name: "account",
 		Fields: []LabelFieldSchema{
@@ -423,7 +434,7 @@ func TestSessionPoolApplySchema(t *testing.T) {
 	assert.Equal(t, "email", labels[1].Field, "field name should be email")
 	assert.Equal(t, "string", labels[1].Type, "field type should be string")
 	assert.Equal(t, "phone", labels[2].Field, "field name should be phone")
-	assert.Equal(t, "int64", labels[2].Type, "field type should be string")
+	assert.Equal(t, "int64", labels[2].Type, "field type should be int64")
 
 	edgeSchema := LabelSchema{
 		Name: "account_email",