package osm

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestOSM_Category(t *testing.T) {
	t.Run("hill", func(t *testing.T) {

		l := &Location{LocCategory: "natural", LocName: "Nice title", LocType: "hill", LocDisplayName: "dipslay name"}
		assert.Equal(t, "hill", l.Category())
	})

	t.Run("water", func(t *testing.T) {

		l := &Location{LocCategory: "", LocName: "Nice title", LocType: "water", LocDisplayName: "dipslay name"}
		assert.Equal(t, "water", l.Category())
	})

	t.Run("shop", func(t *testing.T) {

		l := &Location{LocCategory: "shop", LocName: "Nice title", LocType: "", LocDisplayName: "dipslay name"}
		assert.Equal(t, "shop", l.Category())
	})

	t.Run("no label found", func(t *testing.T) {

		l := &Location{LocCategory: "xxx", LocName: "Nice title", LocType: "", LocDisplayName: "dipslay name"}
		assert.Equal(t, "", l.Category())
	})
}