mirror of
https://github.com/mpolden/echoip.git
synced 2025-06-28 13:47:50 +02:00
Rename package
This commit is contained in:
@ -6,7 +6,7 @@ import (
|
||||
"html/template"
|
||||
|
||||
"github.com/mpolden/ipd/iputil"
|
||||
"github.com/mpolden/ipd/iputil/db"
|
||||
"github.com/mpolden/ipd/iputil/database"
|
||||
"github.com/mpolden/ipd/useragent"
|
||||
"github.com/sirupsen/logrus"
|
||||
|
||||
@ -31,7 +31,7 @@ type Server struct {
|
||||
IPHeader string
|
||||
lookupAddr LookupAddr
|
||||
lookupPort LookupPort
|
||||
db db.Database
|
||||
db database.Client
|
||||
log *logrus.Logger
|
||||
}
|
||||
|
||||
@ -50,7 +50,7 @@ type PortResponse struct {
|
||||
Reachable bool `json:"reachable"`
|
||||
}
|
||||
|
||||
func New(db db.Database, lookupAddr LookupAddr, lookupPort LookupPort, log *logrus.Logger) *Server {
|
||||
func New(db database.Client, lookupAddr LookupAddr, lookupPort LookupPort, log *logrus.Logger) *Server {
|
||||
return &Server{lookupAddr: lookupAddr, lookupPort: lookupPort, db: db, log: log}
|
||||
}
|
||||
|
||||
|
@ -8,7 +8,7 @@ import (
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/mpolden/ipd/iputil/db"
|
||||
"github.com/mpolden/ipd/iputil/database"
|
||||
)
|
||||
|
||||
func lookupAddr(net.IP) ([]string, error) { return []string{"localhost"}, nil }
|
||||
@ -16,8 +16,8 @@ func lookupPort(net.IP, uint64) error { return nil }
|
||||
|
||||
type testDb struct{}
|
||||
|
||||
func (t *testDb) Country(net.IP) (db.Country, error) {
|
||||
return db.Country{Name: "Elbonia", ISO: "EB"}, nil
|
||||
func (t *testDb) Country(net.IP) (database.Country, error) {
|
||||
return database.Country{Name: "Elbonia", ISO: "EB"}, nil
|
||||
}
|
||||
|
||||
func (t *testDb) City(net.IP) (string, error) { return "Bornyasherk", nil }
|
||||
@ -87,7 +87,7 @@ func TestDisabledHandlers(t *testing.T) {
|
||||
server := testServer()
|
||||
server.lookupPort = nil
|
||||
server.lookupAddr = nil
|
||||
server.db = db.Empty()
|
||||
server.db = database.Empty()
|
||||
s := httptest.NewServer(server.Handler())
|
||||
|
||||
var tests = []struct {
|
||||
|
Reference in New Issue
Block a user