From 65cb2cdf73f99a9dc48979af4fd1b58b902c09cc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kuba=20Podg=C3=B3rski?= <kuba--@users.noreply.github.com>
Date: Thu, 25 Mar 2021 18:00:56 +0100
Subject: [PATCH] Update internal endpoints

---
 client/client.go         | 8 ++++----
 client/client_it_test.go | 6 +++---
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/client/client.go b/client/client.go
index 6b3cd6948..405af53b2 100644
--- a/client/client.go
+++ b/client/client.go
@@ -616,7 +616,7 @@ func (c *Client) fetchFragmentNodes(indexName string, shard uint64) ([]fragmentN
 	if c.manualFragmentNode != nil {
 		return []fragmentNode{*c.manualFragmentNode}, nil
 	}
-	path := fmt.Sprintf("/pb.fragment/nodes?shard=%d&index=%s", shard, indexName)
+	path := fmt.Sprintf("/internal/fragment/nodes?shard=%d&index=%s", shard, indexName)
 	_, body, err := c.HTTPRequest("GET", path, []byte{}, nil)
 	if err != nil {
 		return nil, err
@@ -787,14 +787,14 @@ func (c *Client) readSchema() ([]SchemaIndex, error) {
 }
 
 func (c *Client) shardsMax() (map[string]uint64, error) {
-	_, data, err := c.HTTPRequest("GET", "/pb.shards/max", nil, nil)
+	_, data, err := c.HTTPRequest("GET", "/internal/shards/max", nil, nil)
 	if err != nil {
-		return nil, errors.Wrap(err, "requesting /pb.shards/max")
+		return nil, errors.Wrap(err, "requesting /internal/shards/max")
 	}
 	m := map[string]map[string]uint64{}
 	err = json.Unmarshal(data, &m)
 	if err != nil {
-		return nil, errors.Wrap(err, "unmarshaling /pb.shards/max data")
+		return nil, errors.Wrap(err, "unmarshaling /internal/shards/max data")
 	}
 	return m["standard"], nil
 }
diff --git a/client/client_it_test.go b/client/client_it_test.go
index 08b6ad920..dc743767f 100644
--- a/client/client_it_test.go
+++ b/client/client_it_test.go
@@ -1473,7 +1473,7 @@ func TestExportFieldFailure(t *testing.T) {
 			statusCode:    404,
 			contentLength: -1,
 		},
-		"/pb.shards/max": {
+		"/internal/shards/max": {
 			content:       []byte(`{"standard":{"go-testindex": 0}}`),
 			statusCode:    404,
 			contentLength: -1,
@@ -1493,9 +1493,9 @@ func TestExportFieldFailure(t *testing.T) {
 	if err == nil {
 		t.Fatal("should have failed")
 	}
-	statusItem = paths["/pb.shards/max"]
+	statusItem = paths["/internal/shards/max"]
 	statusItem.statusCode = 200
-	paths["/pb.shards/max"] = statusItem
+	paths["/internal/shards/max"] = statusItem
 	_, err = client.ExportField(testField)
 	if err == nil {
 		t.Fatal("should have failed")