github.com/lingyao2333/mo-zero@v1.4.1/rest/internal/starter_test.go (about)

     1  package internal
     2  
     3  import (
     4  	"net/http"
     5  	"net/http/httptest"
     6  	"strconv"
     7  	"strings"
     8  	"testing"
     9  
    10  	"github.com/stretchr/testify/assert"
    11  )
    12  
    13  func TestStartHttp(t *testing.T) {
    14  	svr := httptest.NewUnstartedServer(http.NotFoundHandler())
    15  	fields := strings.Split(svr.Listener.Addr().String(), ":")
    16  	port, err := strconv.Atoi(fields[1])
    17  	assert.Nil(t, err)
    18  	err = StartHttp(fields[0], port, http.NotFoundHandler(), func(svr *http.Server) {
    19  		svr.IdleTimeout = 0
    20  	})
    21  	assert.NotNil(t, err)
    22  }
    23  
    24  func TestStartHttps(t *testing.T) {
    25  	svr := httptest.NewTLSServer(http.NotFoundHandler())
    26  	fields := strings.Split(svr.Listener.Addr().String(), ":")
    27  	port, err := strconv.Atoi(fields[1])
    28  	assert.Nil(t, err)
    29  	err = StartHttps(fields[0], port, "", "", http.NotFoundHandler(), func(svr *http.Server) {
    30  		svr.IdleTimeout = 0
    31  	})
    32  	assert.NotNil(t, err)
    33  }