package profile

import (
	"testing"

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

func TestAddressClaim(t *testing.T) {
	address := Address{
		City:       "city",
		Country:    "country",
		Street:     "15 street st",
		State:      "state",
		PostalCode: "12345",
	}

	addressClaims := address.ToAddressClaim()
	assert.Equal(t, "city", addressClaims.Locality)
	assert.Equal(t, "country", addressClaims.Country)
	assert.Equal(t, "15 street st", addressClaims.Street)
	assert.Equal(t, "state", addressClaims.Region)
	assert.Equal(t, "12345", addressClaims.PostalCode)
	assert.Equal(t, "15 street st\r\ncity\r\nstate\r\n12345\r\ncountry", addressClaims.Formatted)

	m := addressClaims.ToMap()
	assert.Equal(t, "country", m["country"])
	assert.Equal(t, "15 street st", m["street_address"])
	assert.Equal(t, "state", m["region"])
	assert.Equal(t, "city", m["locality"])
	assert.Equal(t, "12345", m["postal_code"])
	assert.Equal(t, "15 street st\r\ncity\r\nstate\r\n12345\r\ncountry", m["formatted"])
}

func TestAddressClaim_PartialData(t *testing.T) {
	address := Address{
		City:    "city",
		Country: "country",
		Street:  "15 street st",
	}

	addressClaims := address.ToAddressClaim()
	assert.Equal(t, "city", addressClaims.Locality)
	assert.Equal(t, "country", addressClaims.Country)
	assert.Equal(t, "15 street st", addressClaims.Street)
	assert.Equal(t, "", addressClaims.Region)
	assert.Equal(t, "", addressClaims.PostalCode)
	assert.Equal(t, "15 street st\r\ncity\r\ncountry", addressClaims.Formatted)

	m := addressClaims.ToMap()
	assert.Equal(t, "15 street st", m["street_address"])
	assert.Equal(t, "city", m["locality"])
	assert.Equal(t, "country", m["country"])
	assert.Equal(t, "15 street st\r\ncity\r\ncountry", m["formatted"])
	assert.Nil(t, m["region"])
	assert.Nil(t, m["postal_code"])
}