Merged in feature/client (pull request #31)

Client Entity * repolevel * servicefunctions * openapiclientget * openapiupdate * client * vendor
2025-01-21 18:24:14 +00:00
parent 4ccb980593
commit 04d8eaf52c
39 changed files with 1532 additions and 432 deletions
@@ -0,0 +1,146 @@
+package queryservice_test
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	queryservice "queryorchestration/api/queryService"
+	"queryorchestration/internal/client"
+	"queryorchestration/internal/database"
+	"queryorchestration/internal/database/repository"
+	"strings"
+	"testing"
+
+	"github.com/go-playground/validator/v10"
+	"github.com/google/uuid"
+	"github.com/labstack/echo/v4"
+	"github.com/pashagolub/pgxmock/v3"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCreateClient(t *testing.T) {
+	pool, err := pgxmock.NewPool()
+	if err != nil {
+		t.Fatalf("failed to open pgxmock database: %v", err)
+	}
+	queries := repository.New(pool)
+	db := &database.Connection{
+		Queries: queries,
+		Pool:    pool,
+	}
+
+	cons := queryservice.NewControllers(validator.New(), &queryservice.Services{
+		Client: client.New(db),
+	})
+
+	body := queryservice.ClientCreate{
+		Name: "example_name",
+	}
+	bodyBytes, err := json.Marshal(body)
+	assert.Nil(t, err)
+
+	e := echo.New()
+	req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(string(bodyBytes)))
+	req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
+	rec := httptest.NewRecorder()
+	ctx := e.NewContext(req, rec)
+
+	id := uuid.New()
+
+	pool.ExpectQuery("-- name: CreateClient :one").WithArgs(body.Name).WillReturnRows(
+		pgxmock.NewRows([]string{"id"}).
+			AddRow(database.MustToDBUUID(id)),
+	)
+
+	err = cons.CreateClient(ctx)
+	assert.Nil(t, err)
+	assert.Equal(t, http.StatusCreated, rec.Code)
+	assert.Equal(t, fmt.Sprintf("{\"id\":\"%s\"}\n", id), rec.Body.String())
+}
+
+func TestGetClient(t *testing.T) {
+	pool, err := pgxmock.NewPool()
+	if err != nil {
+		t.Fatalf("failed to open pgxmock database: %v", err)
+	}
+	queries := repository.New(pool)
+	db := &database.Connection{
+		Queries: queries,
+		Pool:    pool,
+	}
+
+	cons := queryservice.NewControllers(validator.New(), &queryservice.Services{
+		Client: client.New(db),
+	})
+
+	e := echo.New()
+	req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(""))
+	rec := httptest.NewRecorder()
+	ctx := e.NewContext(req, rec)
+
+	id := uuid.New()
+
+	ctx.Set("id", id)
+
+	pool.ExpectQuery("-- name: GetClient :one").WithArgs(database.MustToDBUUID(id)).WillReturnRows(
+		pgxmock.NewRows([]string{"id", "name", "canSync"}).
+			AddRow(database.MustToDBUUID(id), "client_name", true),
+	)
+
+	err = cons.GetClient(ctx, id.String())
+	assert.Nil(t, err)
+	assert.Equal(t, http.StatusOK, rec.Code)
+
+	var res queryservice.JobClient
+	err = json.Unmarshal(rec.Body.Bytes(), &res)
+	assert.Nil(t, err)
+	assert.EqualExportedValues(t, queryservice.JobClient{
+		Id:      id.String(),
+		Name:    "client_name",
+		CanSync: true,
+	}, res)
+}
+
+func TestUpdateClient(t *testing.T) {
+	pool, err := pgxmock.NewPool()
+	if err != nil {
+		t.Fatalf("failed to open pgxmock database: %v", err)
+	}
+	queries := repository.New(pool)
+	db := &database.Connection{
+		Queries: queries,
+		Pool:    pool,
+	}
+
+	cons := queryservice.NewControllers(validator.New(), &queryservice.Services{
+		Client: client.New(db),
+	})
+
+	body := queryservice.QueryUpdate{}
+	bodyBytes, err := json.Marshal(body)
+	assert.Nil(t, err)
+
+	e := echo.New()
+	req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(string(bodyBytes)))
+	req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
+	rec := httptest.NewRecorder()
+	ctx := e.NewContext(req, rec)
+
+	id := uuid.New()
+
+	ctx.Set("id", id)
+
+	pool.ExpectQuery("-- name: GetClient :one").WithArgs(database.MustToDBUUID(id)).WillReturnRows(
+		pgxmock.NewRows([]string{"id", "name", "canSync"}).
+			AddRow(database.MustToDBUUID(id), "client_name", true),
+	)
+
+	pool.ExpectExec("name: UpdateClient :exec").WithArgs("client_name", true, database.MustToDBUUID(id)).
+		WillReturnResult(pgxmock.NewResult("", 1))
+
+	err = cons.UpdateClient(ctx, id.String())
+	assert.Nil(t, err)
+	assert.Equal(t, http.StatusOK, rec.Code)
+	assert.Empty(t, rec.Body.String())
+}