Support partial JIDs in Bare/Full methods

This commit is contained in:
bodqhrohro 2019-11-06 01:27:37 +02:00
parent 47976624c9
commit 9abd5be0aa
2 changed files with 42 additions and 17 deletions

14
jid.go
View file

@ -51,11 +51,21 @@ func NewJid(sjid string) (*Jid, error) {
} }
func (j *Jid) Full() string { func (j *Jid) Full() string {
return j.Node + "@" + j.Domain + "/" + j.Resource if j.Resource == "" {
return j.Bare()
} else if j.Node == "" {
return j.Node + "/" + j.Resource
} else {
return j.Node + "@" + j.Domain + "/" + j.Resource
}
} }
func (j *Jid) Bare() string { func (j *Jid) Bare() string {
return j.Node + "@" + j.Domain if j.Node == "" {
return j.Domain
} else {
return j.Node + "@" + j.Domain
}
} }
// ============================================================================ // ============================================================================

View file

@ -61,26 +61,41 @@ func TestIncorrectJids(t *testing.T) {
} }
func TestFull(t *testing.T) { func TestFull(t *testing.T) {
jid := "test@domain.com/my resource" fullJids := []string{
parsedJid, err := NewJid(jid) "test@domain.com/my resource",
if err != nil { "test@domain.com",
t.Errorf("could not parse jid: %v", err) "domain.com",
} }
fullJid := parsedJid.Full() for _, sjid := range fullJids {
if fullJid != jid { parsedJid, err := NewJid(sjid)
t.Errorf("incorrect full jid: %s", fullJid) if err != nil {
t.Errorf("could not parse jid: %v", err)
}
fullJid := parsedJid.Full()
if fullJid != sjid {
t.Errorf("incorrect full jid: %s", fullJid)
}
} }
} }
func TestBare(t *testing.T) { func TestBare(t *testing.T) {
jid := "test@domain.com" tests := []struct {
fullJid := jid + "/my resource" jidstr string
parsedJid, err := NewJid(fullJid) expected string
if err != nil { }{
t.Errorf("could not parse jid: %v", err) {jidstr: "test@domain.com", expected: "test@domain.com"},
{jidstr: "test@domain.com/resource", expected: "test@domain.com"},
{jidstr: "domain.com", expected: "domain.com"},
} }
bareJid := parsedJid.Bare()
if bareJid != jid { for _, tt := range tests {
t.Errorf("incorrect bare jid: %s", bareJid) parsedJid, err := NewJid(tt.jidstr)
if err != nil {
t.Errorf("could not parse jid: %v", err)
}
bareJid := parsedJid.Bare()
if bareJid != tt.expected {
t.Errorf("incorrect bare jid: %s", bareJid)
}
} }
} }