29 lines
529 B
Go
29 lines
529 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestServeHTTP(t *testing.T) {
|
||
|
|
assert := assert.New(t)
|
||
|
|
plugin := Plugin{}
|
||
|
|
w := httptest.NewRecorder()
|
||
|
|
r := httptest.NewRequest(http.MethodGet, "/", nil)
|
||
|
|
|
||
|
|
plugin.ServeHTTP(nil, w, r)
|
||
|
|
|
||
|
|
result := w.Result()
|
||
|
|
assert.NotNil(result)
|
||
|
|
defer result.Body.Close()
|
||
|
|
bodyBytes, err := ioutil.ReadAll(result.Body)
|
||
|
|
assert.Nil(err)
|
||
|
|
bodyString := string(bodyBytes)
|
||
|
|
|
||
|
|
assert.Equal("Hello, world!", bodyString)
|
||
|
|
}
|