Merge branch 'cli-full-refresh' into 'dle-4-0'

feat(cli): add full-refresh command

See merge request postgres-ai/database-lab!1024

Author: Bogdan Tsechoev

engine/cmd/cli/commands/instance/actions.go

@@ -66,3 +66,20 @@ func health(cliCtx *cli.Context) error {
 
 	return err
 }
+
+// refresh runs a request to initiate a full refresh.
+func refresh(cliCtx *cli.Context) error {
+	client, err := commands.ClientByCLIContext(cliCtx)
+	if err != nil {
+		return err
+	}
+
+	response, err := client.FullRefresh(cliCtx.Context)
+	if err != nil {
+		return err
+	}
+
+	_, err = fmt.Fprintln(cliCtx.App.Writer, response.Message)
+
+	return err
+}

engine/cmd/cli/commands/instance/command_list.go

@@ -25,6 +25,11 @@ func CommandList() []*cli.Command {
 					Usage:  "display instance's version",
 					Action: health,
 				},
+				{
+					Name:   "full-refresh",
+					Usage:  "initiate full refresh",
+					Action: refresh,
+				},
 			},
 		},
 	}

engine/pkg/client/dblabapi/status.go

@@ -72,3 +72,27 @@ func (c *Client) Health(ctx context.Context) (*models.Engine, error) {
 
 	return &engine, nil
 }
+
+// FullRefresh triggers a full refresh of the dataset.
+func (c *Client) FullRefresh(ctx context.Context) (*models.Response, error) {
+	u := c.URL("/full-refresh")
+
+	request, err := http.NewRequest(http.MethodPost, u.String(), nil)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to make a request")
+	}
+
+	response, err := c.Do(ctx, request)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to get response")
+	}
+
+	defer func() { _ = response.Body.Close() }()
+
+	var result models.Response
+	if err := json.NewDecoder(response.Body).Decode(&result); err != nil {
+		return nil, errors.Wrap(err, "failed to get response")
+	}
+
+	return &result, nil
+}

engine/pkg/client/dblabapi/status_test.go

@@ -111,3 +111,58 @@ func TestClientStatusWithFailedRequest(t *testing.T) {
 	require.EqualError(t, err, "failed to get response: EOF")
 	require.Nil(t, status)
 }
+
+func TestClientFullRefresh(t *testing.T) {
+	expectedResponse := &models.Response{
+		Status:  "OK",
+		Message: "Full refresh started",
+	}
+
+	mockClient := NewTestClient(func(req *http.Request) *http.Response {
+		assert.Equal(t, req.URL.String(), "https://example.com/full-refresh")
+		assert.Equal(t, req.Method, http.MethodPost)
+
+		body, err := json.Marshal(expectedResponse)
+		require.NoError(t, err)
+
+		return &http.Response{
+			StatusCode: http.StatusOK,
+			Body:       io.NopCloser(bytes.NewBuffer(body)),
+			Header:     make(http.Header),
+		}
+	})
+
+	c, err := NewClient(Options{
+		Host:              "https://example.com/",
+		VerificationToken: "testVerify",
+	})
+	require.NoError(t, err)
+
+	c.client = mockClient
+
+	resp, err := c.FullRefresh(context.Background())
+	require.NoError(t, err)
+	assert.EqualValues(t, expectedResponse, resp)
+}
+
+func TestClientFullRefreshWithFailedDecode(t *testing.T) {
+	mockClient := NewTestClient(func(req *http.Request) *http.Response {
+		return &http.Response{
+			StatusCode: http.StatusOK,
+			Body:       io.NopCloser(bytes.NewBuffer([]byte{})),
+			Header:     make(http.Header),
+		}
+	})
+
+	c, err := NewClient(Options{
+		Host:              "https://example.com/",
+		VerificationToken: "testVerify",
+	})
+	require.NoError(t, err)
+
+	c.client = mockClient
+
+	resp, err := c.FullRefresh(context.Background())
+	require.EqualError(t, err, "failed to get response: EOF")
+	require.Nil(t, resp)
+}