mirror of
https://github.com/nezhahq/nezha.git
synced 2025-01-23 13:18:13 -05:00
31 lines
718 B
Go
31 lines
718 B
Go
package monitor
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/naiba/nezha/pkg/utils"
|
|
)
|
|
|
|
func TestGeoIPApi(t *testing.T) {
|
|
for i := 0; i < len(geoIPApiList); i++ {
|
|
resp, err := httpGetWithUA(httpClientV4, geoIPApiList[i])
|
|
assert.Nil(t, err)
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
assert.Nil(t, err)
|
|
resp.Body.Close()
|
|
var ip geoIP
|
|
err = utils.Json.Unmarshal(body, &ip)
|
|
assert.Nil(t, err)
|
|
t.Logf("%s %s %s %s", geoIPApiList[i], ip.CountryCode, utils.IPDesensitize(ip.IP), utils.IPDesensitize(ip.Query))
|
|
assert.True(t, ip.IP != "" || ip.Query != "")
|
|
}
|
|
}
|
|
|
|
func TestFetchGeoIP(t *testing.T) {
|
|
ip := fetchGeoIP(geoIPApiList, false)
|
|
assert.NotEmpty(t, ip.IP)
|
|
}
|