add tests

zc2638 · Dec 22, 2023 · 66839db · 66839db
1 parent c099653
commit 66839db
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/endpoint/endpoint_test.go b/endpoint/endpoint_test.go
@@ -206,6 +206,27 @@ type Model struct {
 	String string `json:"s"`
 }
 
+func TestBodyR(t *testing.T) {
+	expected := swag.Parameter{
+		In:       "body",
+		Name:     "body",
+		Required: true,
+		Schema: &swag.Schema{
+			Ref:       "#/definitions/endpoint.Model",
+			Prototype: reflect.TypeOf(Model{}),
+		},
+	}
+
+	e := New(
+		"get", "/",
+		Summary("get thing"),
+		BodyR(Model{}),
+	)
+
+	assert.Equal(t, 1, len(e.Parameters))
+	assert.Equal(t, expected, e.Parameters[0])
+}
+
 func TestBody(t *testing.T) {
 	expected := swag.Parameter{
 		In:          "body",