71 lines
1.3 KiB
Go
71 lines
1.3 KiB
Go
package storage
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/khlieng/dispatch/Godeps/_workspace/src/github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func tempdir() string {
|
|
f, _ := ioutil.TempDir("", "")
|
|
return f
|
|
}
|
|
|
|
func TestUser(t *testing.T) {
|
|
defer func() {
|
|
r := recover()
|
|
assert.Nil(t, r)
|
|
}()
|
|
|
|
Initialize(tempdir())
|
|
Open()
|
|
|
|
srv := Server{
|
|
Name: "Freenode",
|
|
Host: "irc.freenode.net",
|
|
Nick: "test",
|
|
}
|
|
chan1 := Channel{
|
|
Server: srv.Host,
|
|
Name: "#test",
|
|
}
|
|
chan2 := Channel{
|
|
Server: srv.Host,
|
|
Name: "#testing",
|
|
}
|
|
|
|
user := NewUser()
|
|
user.AddServer(srv)
|
|
user.AddChannel(chan1)
|
|
user.AddChannel(chan2)
|
|
user.Close()
|
|
|
|
users := LoadUsers()
|
|
assert.Len(t, users, 1)
|
|
|
|
user = users[0]
|
|
assert.Equal(t, uint64(1), user.ID)
|
|
|
|
servers := user.GetServers()
|
|
assert.Len(t, servers, 1)
|
|
assert.Equal(t, srv, servers[0])
|
|
|
|
channels := user.GetChannels()
|
|
assert.Len(t, channels, 2)
|
|
assert.Equal(t, chan1, channels[0])
|
|
assert.Equal(t, chan2, channels[1])
|
|
|
|
user.SetNick("bob", srv.Host)
|
|
assert.Equal(t, "bob", user.GetServers()[0].Nick)
|
|
|
|
user.RemoveChannel(srv.Host, chan1.Name)
|
|
channels = user.GetChannels()
|
|
assert.Len(t, channels, 1)
|
|
assert.Equal(t, chan2, channels[0])
|
|
|
|
user.RemoveServer(srv.Host)
|
|
assert.Len(t, user.GetServers(), 0)
|
|
assert.Len(t, user.GetChannels(), 0)
|
|
}
|