128 lines
2.7 KiB
Go
128 lines
2.7 KiB
Go
package binary
|
|
|
|
import (
|
|
"github.com/omniscale/imposm3/element"
|
|
"testing"
|
|
)
|
|
|
|
func compareRefs(a []int64, b []int64) bool {
|
|
if len(a) != len(b) {
|
|
return false
|
|
}
|
|
for i, v := range a {
|
|
if v != b[i] {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func TestMarshalNode(t *testing.T) {
|
|
node := &element.Node{}
|
|
node.Id = 12345
|
|
node.Tags = make(element.Tags)
|
|
node.Tags["name"] = "test"
|
|
node.Tags["place"] = "city"
|
|
|
|
data, _ := MarshalNode(node)
|
|
node, _ = UnmarshalNode(data)
|
|
|
|
if node.Tags["name"] != "test" {
|
|
t.Error("name tag does not match")
|
|
}
|
|
if node.Tags["place"] != "city" {
|
|
t.Error("place tag does not match")
|
|
}
|
|
|
|
if len(node.Tags) != 2 {
|
|
t.Error("tags len does not match")
|
|
}
|
|
|
|
}
|
|
|
|
func TestMarshalWay(t *testing.T) {
|
|
way := &element.Way{}
|
|
way.Id = 12345
|
|
way.Tags = make(element.Tags)
|
|
way.Tags["name"] = "test"
|
|
way.Tags["highway"] = "trunk"
|
|
way.Refs = append(way.Refs, 1, 2, 3, 4)
|
|
|
|
data, _ := MarshalWay(way)
|
|
way, _ = UnmarshalWay(data)
|
|
|
|
if way.Tags["name"] != "test" {
|
|
t.Error("name tag does not match")
|
|
}
|
|
if way.Tags["highway"] != "trunk" {
|
|
t.Error("highway tag does not match")
|
|
}
|
|
|
|
if len(way.Tags) != 2 {
|
|
t.Error("tags len does not match")
|
|
}
|
|
|
|
if !compareRefs(way.Refs, []int64{1, 2, 3, 4}) {
|
|
t.Error("nodes do not match")
|
|
}
|
|
|
|
}
|
|
|
|
func TestMarshalRelation(t *testing.T) {
|
|
rel := &element.Relation{}
|
|
rel.Id = 12345
|
|
rel.Tags = make(element.Tags)
|
|
rel.Tags["name"] = "test"
|
|
rel.Tags["landusage"] = "forest"
|
|
rel.Members = append(rel.Members, element.Member{123, element.WAY, "outer", nil})
|
|
rel.Members = append(rel.Members, element.Member{124, element.WAY, "inner", nil})
|
|
|
|
data, _ := MarshalRelation(rel)
|
|
rel, _ = UnmarshalRelation(data)
|
|
|
|
if rel.Tags["name"] != "test" {
|
|
t.Error("name tag does not match")
|
|
}
|
|
if rel.Tags["landusage"] != "forest" {
|
|
t.Error("landusage tag does not match")
|
|
}
|
|
|
|
if len(rel.Tags) != 2 {
|
|
t.Error("tags len does not match")
|
|
}
|
|
|
|
if len(rel.Members) != 2 {
|
|
t.Error("members len does not match")
|
|
}
|
|
|
|
if rel.Members[0].Id != 123 || rel.Members[0].Type != element.WAY || rel.Members[0].Role != "outer" {
|
|
t.Error("members do not match", rel.Members[0])
|
|
}
|
|
|
|
if rel.Members[1].Id != 124 || rel.Members[1].Type != element.WAY || rel.Members[1].Role != "inner" {
|
|
t.Error("members do not match", rel.Members[1])
|
|
}
|
|
}
|
|
|
|
func TestDeltaPack(t *testing.T) {
|
|
ids := []int64{1000, 999, 1001, -8, 1234}
|
|
deltaPack(ids)
|
|
|
|
for i, id := range []int64{1000, -1, 2, -1009, 1242} {
|
|
if ids[i] != id {
|
|
t.Fatal(ids[i], id, ids)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDeltaUnpack(t *testing.T) {
|
|
ids := []int64{1000, -1, 2, -1009, 1242}
|
|
deltaUnpack(ids)
|
|
|
|
for i, id := range []int64{1000, 999, 1001, -8, 1234} {
|
|
if ids[i] != id {
|
|
t.Fatal(ids[i], id, ids)
|
|
}
|
|
}
|
|
}
|