mirror of
https://github.com/ergochat/ergo.git
synced 2024-12-23 19:22:41 +01:00
d082ec7ab9
* don't send multiline responses to CAP LS 301 This is more or less explicitly prohibited by the spec: https://ircv3.net/specs/extensions/capability-negotiation.html#multiline-replies-to-cap-ls-and-cap-list * switch to whitelist model to be future-proof * bump irctest to include test * add a unit test
110 lines
3.1 KiB
Go
110 lines
3.1 KiB
Go
// Copyright (c) 2017 Daniel Oaks <daniel@danieloaks.net>
|
|
// released under the MIT license
|
|
|
|
package caps
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestSets(t *testing.T) {
|
|
s1 := NewSet()
|
|
|
|
s1.Enable(AccountTag, EchoMessage, UserhostInNames)
|
|
|
|
if !(s1.Has(AccountTag) && s1.Has(EchoMessage) && s1.Has(UserhostInNames)) {
|
|
t.Error("Did not have the tags we expected")
|
|
}
|
|
|
|
if s1.Has(STS) {
|
|
t.Error("Has() returned true when we don't have the given capability")
|
|
}
|
|
|
|
s1.Disable(AccountTag)
|
|
|
|
if s1.Has(AccountTag) {
|
|
t.Error("Disable() did not correctly disable the given capability")
|
|
}
|
|
|
|
enabledCaps := NewSet()
|
|
enabledCaps.Union(s1)
|
|
expectedCaps := NewSet(EchoMessage, UserhostInNames)
|
|
if !reflect.DeepEqual(enabledCaps, expectedCaps) {
|
|
t.Errorf("Enabled and expected capability lists do not match: %v, %v", enabledCaps, expectedCaps)
|
|
}
|
|
|
|
// make sure re-enabling doesn't add to the count or something weird like that
|
|
s1.Enable(EchoMessage)
|
|
|
|
// make sure add and remove work fine
|
|
s1.Add(InviteNotify)
|
|
s1.Remove(EchoMessage)
|
|
|
|
if !s1.Has(InviteNotify) || s1.Has(EchoMessage) {
|
|
t.Error("Add/Remove don't work")
|
|
}
|
|
|
|
// test Strings()
|
|
values := make(Values)
|
|
values[InviteNotify] = "invitemepls"
|
|
|
|
actualCap301ValuesString := s1.Strings(Cap301, values, 0)
|
|
expectedCap301ValuesString := []string{"invite-notify userhost-in-names"}
|
|
if !reflect.DeepEqual(actualCap301ValuesString, expectedCap301ValuesString) {
|
|
t.Errorf("Generated Cap301 values string [%v] did not match expected values string [%v]", actualCap301ValuesString, expectedCap301ValuesString)
|
|
}
|
|
|
|
actualCap302ValuesString := s1.Strings(Cap302, values, 0)
|
|
expectedCap302ValuesString := []string{"invite-notify=invitemepls userhost-in-names"}
|
|
if !reflect.DeepEqual(actualCap302ValuesString, expectedCap302ValuesString) {
|
|
t.Errorf("Generated Cap302 values string [%s] did not match expected values string [%s]", actualCap302ValuesString, expectedCap302ValuesString)
|
|
}
|
|
}
|
|
|
|
func assertEqual(found, expected interface{}) {
|
|
if !reflect.DeepEqual(found, expected) {
|
|
panic(fmt.Sprintf("found %#v, expected %#v", found, expected))
|
|
}
|
|
}
|
|
|
|
func Test301WhitelistNotRespectedFor302(t *testing.T) {
|
|
s1 := NewSet()
|
|
s1.Enable(AccountTag, EchoMessage, StandardReplies)
|
|
assertEqual(s1.Strings(Cap301, nil, 0), []string{"account-tag echo-message"})
|
|
assertEqual(s1.Strings(Cap302, nil, 0), []string{"account-tag echo-message standard-replies"})
|
|
}
|
|
|
|
func TestSubtract(t *testing.T) {
|
|
s1 := NewSet(AccountTag, EchoMessage, UserhostInNames, ServerTime)
|
|
|
|
toRemove := NewSet(UserhostInNames, EchoMessage)
|
|
s1.Subtract(toRemove)
|
|
|
|
if !reflect.DeepEqual(s1, NewSet(AccountTag, ServerTime)) {
|
|
t.Errorf("subtract doesn't work")
|
|
}
|
|
}
|
|
|
|
func BenchmarkSetReads(b *testing.B) {
|
|
set := NewSet(UserhostInNames, EchoMessage)
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
set.Has(UserhostInNames)
|
|
set.Has(LabeledResponse)
|
|
set.Has(EchoMessage)
|
|
set.Has(Nope)
|
|
}
|
|
}
|
|
|
|
func BenchmarkSetWrites(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
set := NewSet(UserhostInNames, EchoMessage)
|
|
set.Add(Nope)
|
|
set.Add(ExtendedJoin)
|
|
set.Remove(UserhostInNames)
|
|
set.Remove(LabeledResponse)
|
|
}
|
|
}
|