package ip

import (
	"github.com/stretchr/testify/assert"
	"log"
	"testing"
)

var d *database

func db() *database {
	if d != nil {
		return d
	}

	d, e := getDB()
	if e != nil {
		log.Fatal(e)
	}
	return d
}

func TestInternal(t *testing.T) {
	loc, e := db().query("192.168.1.1")
	if e != nil {
		t.Fail()
		return
	}
	assert.Equal(t, "局域网", loc.Country)
	assert.Equal(t, "192.168.0.0", loc.BeginIP)
	assert.Equal(t, "192.168.0.0", loc.BeginIP)
}

func TestChina(t *testing.T) {
	loc, e := db().query("47.105.173.69")
	if e != nil {
		t.Fail()
		return
	}
	assert.Equal(t, "山东省青岛市", loc.Country)
	assert.Equal(t, "阿里云", loc.Area)
}

func TestForeign(t *testing.T) {
	loc, e := db().query("8.8.8.8")
	if e != nil {
		t.Fail()
		return
	}
	assert.Equal(t, "美国", loc.Country)
	assert.Equal(t, "加利福尼亚州圣克拉拉县山景市谷歌公司DNS服务器", loc.Area)
}