file_db_test.go

package ip2location

import (
	"testing"
)

var DBpath = "IP-COUNTRY-SAMPLE.BIN"

func TestNewLocation(t *testing.T) {
	t.Run("Read from file", func(t *testing.T) {
		db, err := NewFileDB(DBpath, false)
		if err != nil {
			t.Error(err)
		}

		r, err := db.Query("8.8.8.8", All)
		if err != nil {
			t.Error(err)
		}

		if r.CountryLong != "UNITED STATES" {
			t.Error("County name is not equal. Expected UNITED STATES, got", r.CountryLong)
		}

		err = db.Close()
		if err != nil {
			t.Error(err)
		}

	})

	t.Run("Read from memory", func(t *testing.T) {
		db, err := NewFileDB(DBpath, true)
		if err != nil {
			t.Error(err)
		}

		r, err := db.Query("8.8.8.8", CountryLong)
		if err != nil {
			t.Error(err)
		}

		if r.CountryLong != "UNITED STATES" {
			t.Error("County name is not equal. Expected UNITED STATES, got", r.CountryLong)
		}

		err = db.Close()
		if err != nil {
			t.Error(err)
		}
	})

	t.Run("Wrong path", func(t *testing.T) {
		_, err := NewFileDB("", false)
		if err == nil {
			t.Error(err)
		}
	})

	t.Run("Dir", func(t *testing.T) {
		_, err := NewFileDB(".", false)
		if err == nil {
			t.Error(err)
		}
	})
}