feat: test list augmented response (#2732)

add augmented list
kubeshop · Jun 22, 2023 · 924b9e5 · 924b9e5
1 parent 7972d5a
commit 924b9e5
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 10 deletions.
diff --git a/server/app/app.go b/server/app/app.go
@@ -441,6 +441,9 @@ func registerDataStoreResource(repository *datastoreresource.Repository, router
 func registerTestResource(repository test.Repository, router *mux.Router, provisioner *provisioning.Provisioner, tracer trace.Tracer) {
 	operations := []resourcemanager.Operation{
 		resourcemanager.OperationList,
+		// TODO: replace it with the option `resourcemanager.CanBeAugmented()`
+		// once we have the `resourcemanager.OperationGet` operation.
+		resourcemanager.OperationListAugmented,
 	}
 
 	manager := resourcemanager.New[test.Test](

diff --git a/server/test/test_repository.go b/server/test/test_repository.go
@@ -13,6 +13,7 @@ import (
 
 type Repository interface {
 	List(_ context.Context, take, skip int, query, sortBy, sortDirection string) ([]Test, error)
+	ListAugmented(_ context.Context, take, skip int, query, sortBy, sortDirection string) ([]Test, error)
 	Count(context.Context, string) (int, error)
 	SortingFields() []string
 	Provision(context.Context, Test) error
@@ -81,6 +82,27 @@ const (
 )
 
 func (r *repository) List(ctx context.Context, take, skip int, query, sortBy, sortDirection string) ([]Test, error) {
+	tests, err := r.list(ctx, take, skip, query, sortBy, sortDirection)
+	if err != nil {
+		return []Test{}, err
+	}
+
+	for i, test := range tests {
+		test.CreatedAt = nil
+		test.Summary = nil
+		test.Version = nil
+		tests[i] = test
+	}
+
+	return tests, err
+
+}
+
+func (r *repository) ListAugmented(ctx context.Context, take, skip int, query, sortBy, sortDirection string) ([]Test, error) {
+	return r.list(ctx, take, skip, query, sortBy, sortDirection)
+}
+
+func (r *repository) list(ctx context.Context, take, skip int, query, sortBy, sortDirection string) ([]Test, error) {
 	sql := getTestSQL + testMaxVersionQuery
 	params := []any{take, skip}
 
@@ -155,7 +177,14 @@ func (r *repository) readRows(ctx context.Context, rows *sql.Rows) ([]Test, erro
 }
 
 func (r *repository) readRow(ctx context.Context, row scanner) (Test, error) {
-	test := Test{}
+	version := 0
+	createdAt := time.Now()
+
+	test := Test{
+		CreatedAt: &createdAt,
+		Version:   &version,
+		Summary:   &Summary{},
+	}
 
 	var (
 		jsonServiceUnderTest,

diff --git a/server/test/tests.go b/server/test/tests.go
@@ -20,15 +20,15 @@ const (
 type (
 	// this struct yaml/json encoding is handled at ./test_json.go for custom encodings
 	Test struct {
-		ID          id.ID                        `json:"id"`
-		CreatedAt   time.Time                    `json:"createdAt"`
-		Name        string                       `json:"name"`
-		Description string                       `json:"description"`
-		Version     int                          `json:"version"`
-		Trigger     trigger.Trigger              `json:"trigger"`
-		Specs       Specs                        `json:"specs"`
-		Outputs     maps.Ordered[string, Output] `json:"outputs"`
-		Summary     Summary                      `json:"summary"`
+		ID          id.ID                        `json:"id,omitempty"`
+		CreatedAt   *time.Time                   `json:"createdAt,omitempty"`
+		Name        string                       `json:"name,omitempty"`
+		Description string                       `json:"description,omitempty"`
+		Version     *int                         `json:"version,omitempty"`
+		Trigger     trigger.Trigger              `json:"trigger,omitempty"`
+		Specs       Specs                        `json:"specs,omitempty"`
+		Outputs     maps.Ordered[string, Output] `json:"outputs,omitempty"`
+		Summary     *Summary                     `json:"summary,omitempty"`
 	}
 
 	Specs []TestSpec