mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-24 01:30:25 +08:00
31 lines
578 B
Go
31 lines
578 B
Go
|
package stat
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"gopkg.in/h2non/gock.v1"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestRemoteWriter(t *testing.T) {
|
||
|
|
defer gock.Off()
|
||
|
|
|
||
|
|
gock.New("http://foo.com").Reply(200).BodyString("foo")
|
||
|
|
writer := NewRemoteWriter("http://foo.com")
|
||
|
|
err := writer.Write(&StatReport{
|
||
|
|
Name: "bar",
|
||
|
|
})
|
||
|
|
assert.Nil(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRemoteWriterFail(t *testing.T) {
|
||
|
|
defer gock.Off()
|
||
|
|
|
||
|
|
gock.New("http://foo.com").Reply(503).BodyString("foo")
|
||
|
|
writer := NewRemoteWriter("http://foo.com")
|
||
|
|
err := writer.Write(&StatReport{
|
||
|
|
Name: "bar",
|
||
|
|
})
|
||
|
|
assert.NotNil(t, err)
|
||
|
|
}
|