// Copyright (c) 2021 OceanBase
// obagent is licensed under Mulan PSL v2.
// You can use this software according to the terms and conditions of the Mulan PSL v2.
// You may obtain a copy of Mulan PSL v2 at:
//
// http://license.coscl.org.cn/MulanPSL2
//
// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
// EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
// MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
// See the Mulan PSL v2 for more details.

package web

import (
	"context"
	"net/http"
	"net/http/httptest"
	"testing"
	"time"

	"github.com/gin-gonic/gin"
	. "github.com/smartystreets/goconvey/convey"
)

func TestCounter(t *testing.T) {
	count := new(Counter)
	Convey("counter after init", t, func() {
		So(count.sessionCount, ShouldEqual, 0)
	})

	count.incr()
	Convey("counter after incr", t, func() {
		So(count.sessionCount, ShouldEqual, 1)
	})

	count.decr()
	Convey("counter after decr", t, func() {
		So(count.sessionCount, ShouldEqual, 0)
	})
}

func TestHttpServer(t *testing.T) {
	server := &HttpServer{
		Counter: new(Counter),
		Router:  gin.Default(),
		Server: &http.Server{
			Addr: ":0",
		},
	}

	w := httptest.NewRecorder()
	server.UseCounter()
	server.Router.GET("/foo", fooHandler)
	end := make(chan bool, 1)
	handler := func(w http.ResponseWriter, r *http.Request) {
		server.Router.ServeHTTP(w, r)
		time.Sleep(time.Second)
	}
	req := httptest.NewRequest(http.MethodGet, "/foo", nil)
	go func() {
		handler(w, req)
		end <- true
	}()

	time.Sleep(10 * time.Millisecond)
	t.Run("handle a 1 second request", func(t *testing.T) {
		Convey("session count should be 1", t, func() {
			So(server.Counter.sessionCount, ShouldEqual, 1)
		})

		err := server.Shutdown(context.Background())
		Convey("server shutdown should fail", t, func() {
			So(err, ShouldNotBeNil)
			So(err.Error(), ShouldContainSubstring, "server shutdown failed")
		})
	})

	<-end
	t.Run("handle request end", func(t *testing.T) {
		Convey("session count should be 0", t, func() {
			So(server.Counter.sessionCount, ShouldEqual, 0)
		})
		err := server.Shutdown(context.Background())
		Convey("server shutdown should success", t, func() {
			So(err, ShouldBeNil)
		})
	})
}

func fooHandler(c *gin.Context) {
	time.Sleep(time.Second)
}