fiberkit/query_test.go at main · vo0ov/fiberkit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package fiberkit

import (
	"encoding/json"
	"net/http/httptest"
	"testing"

	"github.com/gofiber/fiber/v3"
)

type listTasksQuery struct {
	Page int `query:"page" validate:"required"`
}

func TestQueryParsesValues(t *testing.T) {
	app := fiber.New()
	app.Get("/tasks", Query(func(ctx fiber.Ctx, query listTasksQuery) error {
		return ctx.JSON(query)
	}))

	req := httptest.NewRequest(fiber.MethodGet, "/tasks?page=3", nil)
	resp, err := app.Test(req)
	if err != nil {
		t.Fatalf("app.Test() error = %v", err)
	}

	if resp.StatusCode != fiber.StatusOK {
		t.Fatalf("status = %d, want %d", resp.StatusCode, fiber.StatusOK)
	}

	var got listTasksQuery
	if err := json.NewDecoder(resp.Body).Decode(&got); err != nil {
		t.Fatalf("decode response: %v", err)
	}

	if got.Page != 3 {
		t.Fatalf("page = %d, want %d", got.Page, 3)
	}
}

func TestQueryValidatesInput(t *testing.T) {
	app := fiber.New()
	app.Get("/tasks", Query(func(ctx fiber.Ctx, query listTasksQuery) error {
		return ctx.JSON(query)
	}))

	req := httptest.NewRequest(fiber.MethodGet, "/tasks", nil)
	resp, err := app.Test(req)
	if err != nil {
		t.Fatalf("app.Test() error = %v", err)
	}

	if resp.StatusCode != fiber.StatusBadRequest {
		t.Fatalf("status = %d, want %d", resp.StatusCode, fiber.StatusBadRequest)
	}

	var got map[string]any
	if err := json.NewDecoder(resp.Body).Decode(&got); err != nil {
		t.Fatalf("decode response: %v", err)
	}

	if got["error"] != "validation failed" {
		t.Fatalf("error = %v, want %q", got["error"], "validation failed")
	}

	details, ok := got["details"].(map[string]any)
	if !ok {
		t.Fatalf("details type = %T, want map[string]any", got["details"])
	}

	if details["Page"] != "required" {
		t.Fatalf("details[Page] = %v, want %q", details["Page"], "required")
	}
}

func TestQueryRejectsInvalidQueryInput(t *testing.T) {
	app := fiber.New()
	app.Get("/tasks", Query(func(ctx fiber.Ctx, query listTasksQuery) error {
		return ctx.JSON(query)
	}))

	req := httptest.NewRequest(fiber.MethodGet, "/tasks?page=invalid", nil)
	resp, err := app.Test(req)
	if err != nil {
		t.Fatalf("app.Test() error = %v", err)
	}

	if resp.StatusCode != fiber.StatusBadRequest {
		t.Fatalf("status = %d, want %d", resp.StatusCode, fiber.StatusBadRequest)
	}

	var got map[string]any
	if err := json.NewDecoder(resp.Body).Decode(&got); err != nil {
		t.Fatalf("decode response: %v", err)
	}

	if got["error"] != "invalid query" {
		t.Fatalf("error = %v, want %q", got["error"], "invalid query")
	}
}