openapi3_test.go

package httpstub

import (
	"fmt"
	"net/http"
	"strings"
	"testing"

	"github.com/golang/mock/gomock"
	mock_httpstub "github.com/k1LoW/httpstub/mock"
)

func TestOpenAPI3(t *testing.T) {
	tests := []struct {
		name    string
		req     *http.Request
		wantErr bool
	}{
		{"valid req/res", newRequest(t, http.MethodPost, "/api/v1/users", `{"username": "alice", "password": "passw0rd"}`), false},
		{"invalid route", newRequest(t, http.MethodPost, "/api/v1/invalid/route", `{"username": "alice", "password": "passw0rd"}`), true},
		{"invalid req", newRequest(t, http.MethodPost, "/api/v1/users", `{"invalid": "alice", "req": "passw0rd"}`), true},
		{"invalid res", newRequest(t, http.MethodGet, "/api/v1/users", ``), true},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			ctrl := gomock.NewController(t)
			defer ctrl.Finish()
			mockTB := mock_httpstub.NewMockTB(ctrl)
			mockTB.EXPECT().Helper().AnyTimes()
			if tt.wantErr {
				mockTB.EXPECT().Errorf(gomock.Any(), gomock.Any())
			}
			rt := NewRouter(mockTB, OpenApi3("testdata/openapi3.yml"))
			rt.Method(http.MethodPost).Path("/api/v1/users").Header("Content-Type", "application/json").ResponseString(http.StatusCreated, `{"name":"alice"}`)
			// invalid response
			rt.Method(http.MethodGet).Path("/api/v1/users").Header("Content-Type", "application/json").ResponseString(http.StatusBadRequest, `{"invalid":"data"}`)
			ts := rt.Server()
			t.Cleanup(func() {
				ts.Close()
			})
			tc := ts.Client()
			if _, err := tc.Do(tt.req); err != nil {
				t.Error(err)
			}
		})
	}
}

func TestSkipCircularReferenceCheck(t *testing.T) {
	tests := []struct {
		skipCircularReferenceCheck bool
		wantErr                    bool
	}{
		{false, true},
		{true, false},
	}
	for _, tt := range tests {
		t.Run(fmt.Sprintf("skipCircularReferenceCheck=%v", tt.skipCircularReferenceCheck), func(t *testing.T) {
			ctrl := gomock.NewController(t)
			defer ctrl.Finish()
			mockTB := mock_httpstub.NewMockTB(ctrl)
			mockTB.EXPECT().Helper().AnyTimes()
			if tt.wantErr {
				mockTB.EXPECT().Fatal(gomock.Any())
			}
			rt := NewRouter(mockTB, OpenApi3("testdata/openapi3-circular-references.yml"), SkipCircularReferenceCheck(tt.skipCircularReferenceCheck), SkipValidateResponse(true))
			// invalid response
			rt.Method(http.MethodGet).Path("/api/hello").Header("Content-Type", "application/json").ResponseString(http.StatusOK, `{"rows":[]}`)
			ts := rt.Server()
			t.Cleanup(func() {
				ts.Close()
			})
			tc := ts.Client()
			req := newRequest(t, http.MethodGet, "/api/hello", "")
			if _, err := tc.Do(req); err != nil {
				t.Error(err)
			}
		})
	}
}

func newRequest(t *testing.T, method string, path string, body string) *http.Request {
	t.Helper()
	req, err := http.NewRequest(method, path, strings.NewReader(body))
	if err != nil {
		t.Fatal(err)
	}
	req.Header.Add("Content-Type", "application/json")
	return req
}