package model

import (
	"testing"

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

// Test Values for NewJID from RFC7622
// https://tools.ietf.org/html/rfc7622
func TestNewJID(t *testing.T) {
	assert := assert.New(t)

	checkList := map[string]*JID{
		"juliet@example.com": &JID{
			Local:  "juliet",
			Domain: "example.com",
		},
		"juliet@example.com/foo": &JID{
			Local:    "juliet",
			Domain:   "example.com",
			Resource: "foo",
		},
		"juliet@example.com/foo bar": &JID{
			Local:    "juliet",
			Domain:   "example.com",
			Resource: "foo bar",
		},
		"juliet@example.com/foo@bar": &JID{
			Local:    "juliet",
			Domain:   "example.com",
			Resource: "foo@bar",
		},
		"foo\\20bar@example.com": &JID{
			Local:  "foo\\20bar",
			Domain: "example.com",
		},
		"fussball@example.com": &JID{
			Local:  "fussball",
			Domain: "example.com",
		},
		"fußball@example.com": &JID{
			Local:  "fußball",
			Domain: "example.com",
		},
		"π@example.com": &JID{
			Local:  "π",
			Domain: "example.com",
		},
		"Σ@example.com/foo": &JID{
			Local:    "Σ",
			Domain:   "example.com",
			Resource: "foo",
		},
		"σ@example.com/foo": &JID{
			Local:    "σ",
			Domain:   "example.com",
			Resource: "foo",
		},
		"ς@example.com/foo": &JID{
			Local:    "ς",
			Domain:   "example.com",
			Resource: "foo",
		},
		"king@example.com/♚": &JID{
			Local:    "king",
			Domain:   "example.com",
			Resource: "♚",
		},
		"example.com": &JID{
			Domain: "example.com",
		},
		"example.com/foobar": &JID{
			Domain:   "example.com",
			Resource: "foobar",
		},
		"a.example.com/b@example.net": &JID{
			Domain:   "a.example.com",
			Resource: "b@example.net",
		},
		"\"juliet\"@example.com":  nil,
		"foo bar@example.com":     nil,
		"juliet@example.com/ foo": nil,
		"@example.com/":           nil,
		// "henryⅣ@example.com": nil, -- ignore for easier implementation
		// "♚@example.com":      nil,
		"juliet@": nil,
		"/foobar": nil,
	}

	for jidString, jidValid := range checkList {
		jid := NewJID(jidString)

		if jidValid != nil {
			assert.NotNil(jid, "this should be a valid JID:"+jidString)
			if jid == nil {
				continue
			}

			assert.Equal(jidValid.Local, jid.Local, "the local part was not right detectet:"+jidString)
			assert.Equal(jidValid.Domain, jid.Domain, "the domain part was not right detectet:"+jidString)
			assert.Equal(jidValid.Resource, jid.Resource, "the resource part was not right detectet:"+jidString)
			assert.Equal(jidValid.Full(), jidString, "the function full of jid did not work")
		} else {
			assert.Nil(jid, "this should not be a valid JID:"+jidString)
		}

	}
}

func TestJIDBare(t *testing.T) {
	assert := assert.New(t)

	checkList := map[string]*JID{
		"aaa@example.com": &JID{
			Local:  "aaa",
			Domain: "example.com",
		},
		"aab@example.com": &JID{
			Local:    "aab",
			Domain:   "example.com",
			Resource: "foo",
		},
		"example.com": &JID{
			Domain:   "example.com",
			Resource: "foo",
		},
	}
	for jidValid, jid := range checkList {
		jidBase := jid.Bare()
		assert.Equal(jidValid, jidBase)

	}
}

func TestMarshal(t *testing.T) {
	assert := assert.New(t)

	jid := &JID{}
	err := jid.UnmarshalText([]byte("juliet@example.com/foo"))

	assert.NoError(err)
	assert.Equal(jid.Local, "juliet")
	assert.Equal(jid.Domain, "example.com")
	assert.Equal(jid.Resource, "foo")

	err = jid.UnmarshalText([]byte("juliet@example.com/ foo"))

	assert.Error(err)

	jid = &JID{
		Local:    "romeo",
		Domain:   "example.com",
		Resource: "bar",
	}
	jidString, err := jid.MarshalText()
	assert.NoError(err)
	assert.Equal("romeo@example.com/bar", string(jidString))
}