mirror of
https://github.com/shazow/ssh-chat.git
synced 2025-04-13 15:47:17 +03:00
39 lines
521 B
Go
39 lines
521 B
Go
package chat
|
|
|
|
import "testing"
|
|
|
|
func TestSet(t *testing.T) {
|
|
var err error
|
|
s := NewSet()
|
|
u := NewUser(testId("foo"))
|
|
|
|
if s.In(u) {
|
|
t.Errorf("Set should be empty.")
|
|
}
|
|
|
|
err = s.Add(u)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !s.In(u) {
|
|
t.Errorf("Set should contain user.")
|
|
}
|
|
|
|
u2 := NewUser(testId("bar"))
|
|
err = s.Add(u2)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
err = s.Add(u2)
|
|
if err != ErrIdTaken {
|
|
t.Error(err)
|
|
}
|
|
|
|
size := s.Len()
|
|
if size != 2 {
|
|
t.Errorf("Set wrong size: %d (expected %d)", size, 2)
|
|
}
|
|
}
|