package test

import (
	"database/sql"
	"io/ioutil"
	"math"
	"os"
	"testing"

	"github.com/omniscale/imposm3/geom/geos"
)

func TestRouteRelation_Prepare(t *testing.T) {
	var err error

	ts.dir, err = ioutil.TempDir("", "imposm_test")
	if err != nil {
		t.Fatal(err)
	}
	ts.config = importConfig{
		connection:      "postgis://",
		cacheDir:        ts.dir,
		osmFileName:     "generated/route_relation.pbf",
		mappingFileName: "route_relation_mapping.yml",
	}
	ts.g = geos.NewGeos()

	ts.db, err = sql.Open("postgres", "sslmode=disable")
	if err != nil {
		t.Fatal(err)
	}
	ts.dropSchemas()
}

func TestRouteRelation_Import(t *testing.T) {
	if ts.tableExists(t, dbschemaImport, "osm_routes") != false {
		t.Fatalf("table osm_routes exists in schema %s", dbschemaImport)
	}
	ts.importOsm(t)
	if ts.tableExists(t, dbschemaImport, "osm_routes") != true {
		t.Fatalf("table osm_routes does not exists in schema %s", dbschemaImport)
	}
}

func TestRouteRelation_Deploy(t *testing.T) {
	ts.deployOsm(t)
	if ts.tableExists(t, dbschemaImport, "osm_routes") != false {
		t.Fatalf("table osm_routes exists in schema %s", dbschemaImport)
	}
	if ts.tableExists(t, dbschemaProduction, "osm_routes") != true {
		t.Fatalf("table osm_routes does not exists in schema %s", dbschemaProduction)
	}
}

func TestRouteRelation_RelationData(t *testing.T) {
	// check tags of relation
	r := ts.queryTags(t, "osm_routes", -100901)
	if r.tags["name"] != "Bus 301: A => B" {
		t.Error(r)
	}

	// check tags of master relation
	r = ts.queryTags(t, "osm_routes", -100911)
	if r.tags["name"] != "Bus 301" {
		t.Error(r)
	}
}

func TestRouteRelation_MemberUpdatedByNode1(t *testing.T) {
	// check that member is updated after node was modified
	rows := ts.queryDynamic(t, "osm_route_members", "osm_id = -110901 AND member = 110101")
	if len(rows) != 1 {
		t.Fatal(rows)
	}
	if rows[0]["name"] != "Stop" {
		t.Error(rows[0])
	}
}

func TestRouteRelation_MemberGeomUpdated1(t *testing.T) {
	rows := ts.queryDynamic(t, "osm_route_members", "osm_id = -100902 AND member = 100502")
	if len(rows) != 1 {
		t.Fatal(rows)
	}
	g := ts.g.FromWkt(rows[0]["wkt"])
	if math.Abs(g.Length()-111.32448543701321) > 0.00000001 {
		t.Fatal(g.Length())
	}

	rows = ts.queryDynamic(t, "osm_route_members", "osm_id = -100902 AND member = 100503")
	if len(rows) != 1 {
		t.Fatal(rows)
	}
	if rows[0]["name"] != "" {
		t.Error(rows[0])
	}
}

func TestRouteRelation_NoRouteWithMissingMember(t *testing.T) {
	// current implementation: route members are all or nothing.
	// if one member is missing, no member is imported
	if r := ts.queryDynamic(t, "osm_route_members", "osm_id = -120901 AND member = 120101"); len(r) > 0 {
		t.Error("found member from route with missing members")
	}
}

// #######################################################################

func TestRouteRelation_Update(t *testing.T) {
	ts.updateOsm(t, "./generated/route_relation.osc.gz")
}

// #######################################################################

func TestRouteRelation_MemberGeomUpdated2(t *testing.T) {
	rows := ts.queryDynamic(t, "osm_route_members", "osm_id = -100902 AND member = 100502")
	if len(rows) != 1 {
		t.Fatal(rows)
	}
	g := ts.g.FromWkt(rows[0]["wkt"])
	if math.Abs(g.Length()-184.97560221624542) > 0.00000001 {
		t.Fatal(g.Length())
	}

	// tag from member is updated
	rows = ts.queryDynamic(t, "osm_route_members", "osm_id = -100902 AND member = 100503")
	if len(rows) != 1 {
		t.Fatal(rows)
	}
	if rows[0]["name"] != "new name" {
		t.Error(rows[0])
	}

	// member is removed
	rows = ts.queryDynamic(t, "osm_route_members", "osm_id = -100902 AND member = 100512")
	if len(rows) != 0 {
		t.Fatal(rows)
	}

	// role from member is updated
	rows = ts.queryDynamic(t, "osm_route_members", "osm_id = -100902 AND member = 100102")
	if len(rows) != 1 {
		t.Fatal(rows)
	}
	if rows[0]["role"] != "halt" {
		t.Error(rows[0])
	}

}

func TestRouteRelation_MemberUpdatedByNode2(t *testing.T) {
	// check that member is updated after node was modified
	rows := ts.queryDynamic(t, "osm_route_members", "osm_id = -110901 AND member = 110101")
	if len(rows) != 1 {
		t.Fatal(rows)
	}
	if rows[0]["name"] != "Stop2" {
		t.Error(rows[0])
	}
}

func TestRouteRelation_Cleanup(t *testing.T) {
	ts.dropSchemas()
	if err := os.RemoveAll(ts.dir); err != nil {
		t.Error(err)
	}
}