mirror of https://github.com/ergochat/ergo.git
134 lines
2.6 KiB
Go
134 lines
2.6 KiB
Go
// Copyright (c) 2019 Shivaram Lingamneni
|
|
// released under the MIT license
|
|
|
|
package irc
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/ergochat/ergo/irc/languages"
|
|
"github.com/ergochat/ergo/irc/utils"
|
|
)
|
|
|
|
func TestGenerateBatchID(t *testing.T) {
|
|
var session Session
|
|
s := make(utils.HashSet[string])
|
|
|
|
count := 100000
|
|
for i := 0; i < count; i++ {
|
|
s.Add(session.generateBatchID())
|
|
}
|
|
|
|
if len(s) != count {
|
|
t.Error("duplicate batch ID detected")
|
|
}
|
|
}
|
|
|
|
func BenchmarkGenerateBatchID(b *testing.B) {
|
|
var session Session
|
|
for i := 0; i < b.N; i++ {
|
|
session.generateBatchID()
|
|
}
|
|
}
|
|
|
|
func BenchmarkNames(b *testing.B) {
|
|
channelSize := 1024
|
|
server := &Server{
|
|
name: "ergo.test",
|
|
}
|
|
lm, err := languages.NewManager(false, "", "")
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
server.config.Store(&Config{
|
|
languageManager: lm,
|
|
})
|
|
for i := 0; i < b.N; i++ {
|
|
channel := &Channel{
|
|
name: "#test",
|
|
nameCasefolded: "#test",
|
|
server: server,
|
|
members: make(MemberSet),
|
|
}
|
|
for j := 0; j < channelSize; j++ {
|
|
nick := fmt.Sprintf("client_%d", j)
|
|
client := &Client{
|
|
server: server,
|
|
nick: nick,
|
|
nickCasefolded: nick,
|
|
}
|
|
channel.members.Add(client)
|
|
channel.regenerateMembersCache()
|
|
session := &Session{
|
|
client: client,
|
|
}
|
|
rb := NewResponseBuffer(session)
|
|
channel.Names(client, rb)
|
|
if len(rb.messages) < 2 {
|
|
b.Fatalf("not enough messages: %d", len(rb.messages))
|
|
}
|
|
// to inspect the messages: line, _ := rb.messages[0].Line()
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestUserMasks(t *testing.T) {
|
|
var um UserMaskSet
|
|
|
|
if um.Match("horse_!user@tor-network.onion") {
|
|
t.Error("bad match")
|
|
}
|
|
|
|
um.Add("_!*@*", "x", "x")
|
|
if !um.Match("_!user@tor-network.onion") {
|
|
t.Error("failure to match")
|
|
}
|
|
if um.Match("horse_!user@tor-network.onion") {
|
|
t.Error("bad match")
|
|
}
|
|
|
|
um.Add("beer*!*@*", "x", "x")
|
|
if !um.Match("beergarden!user@tor-network.onion") {
|
|
t.Error("failure to match")
|
|
}
|
|
if um.Match("horse_!user@tor-network.onion") {
|
|
t.Error("bad match")
|
|
}
|
|
|
|
um.Add("horse*!user@*", "x", "x")
|
|
if !um.Match("horse_!user@tor-network.onion") {
|
|
t.Error("failure to match")
|
|
}
|
|
}
|
|
|
|
func TestWhoFields(t *testing.T) {
|
|
var w whoxFields
|
|
|
|
if w.Has('a') {
|
|
t.Error("zero value of whoxFields must be empty")
|
|
}
|
|
w = w.Add('a')
|
|
if !w.Has('a') {
|
|
t.Error("failed to set and get")
|
|
}
|
|
if w.Has('A') {
|
|
t.Error("false positive")
|
|
}
|
|
if w.Has('o') {
|
|
t.Error("false positive")
|
|
}
|
|
w = w.Add('🐬')
|
|
if w.Has('🐬') {
|
|
t.Error("should not be able to set invalid who field")
|
|
}
|
|
w = w.Add('o')
|
|
if !w.Has('o') {
|
|
t.Error("failed to set and get")
|
|
}
|
|
w = w.Add('z')
|
|
if !w.Has('z') {
|
|
t.Error("failed to set and get")
|
|
}
|
|
}
|