diff --git a/internal/entity/location_test.go b/internal/entity/location_test.go new file mode 100644 index 000000000..24c095f0e --- /dev/null +++ b/internal/entity/location_test.go @@ -0,0 +1,13 @@ +package entity + +import ( + "github.com/stretchr/testify/assert" + "testing" +) + +func TestLocation_Label(t *testing.T) { + location := &Location{LocCategory: "restaurant", LocType: "bistro"} + result := location.Label() + + assert.Equal(t, "restaurant", result) +}