56 lines
1.6 KiB
Go
56 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/tinode/chat/server/store/types"
|
|
)
|
|
|
|
func TestHubKeepsSameTopicNameSeparateByTenant(t *testing.T) {
|
|
h := &Hub{topics: &sync.Map{}}
|
|
aKey := types.TopicKey{TenantID: 1, Topic: "sys"}
|
|
bKey := types.TopicKey{TenantID: 2, Topic: "sys"}
|
|
a := &Topic{tenantID: 1, name: "sys"}
|
|
b := &Topic{tenantID: 2, name: "sys"}
|
|
|
|
h.topicPut(aKey, a)
|
|
h.topicPut(bKey, b)
|
|
|
|
if got := h.topicGet(aKey); got != a {
|
|
t.Fatalf("tenant 1 returned wrong topic: %p", got)
|
|
}
|
|
if got := h.topicGet(bKey); got != b {
|
|
t.Fatalf("tenant 2 returned wrong topic: %p", got)
|
|
}
|
|
if h.numTopics != 2 {
|
|
t.Fatalf("expected 2 topics, got %d", h.numTopics)
|
|
}
|
|
}
|
|
|
|
func TestServerMessageKeepsTenantAndRejectsCrossTenantDelivery(t *testing.T) {
|
|
original := &ServerComMessage{TenantID: 2, RcptTo: "sys"}
|
|
if copied := original.copy(); copied.TenantID != original.TenantID {
|
|
t.Fatalf("copy lost tenant: got %d, want %d", copied.TenantID, original.TenantID)
|
|
}
|
|
|
|
sess := &Session{tenantID: 1, send: make(chan any, 1)}
|
|
if sess.queueOut(original) {
|
|
t.Fatal("cross-tenant message was accepted")
|
|
}
|
|
select {
|
|
case <-sess.send:
|
|
t.Fatal("cross-tenant message reached the session queue")
|
|
default:
|
|
}
|
|
}
|
|
|
|
func TestTopicAssignsTenantToServerMessage(t *testing.T) {
|
|
topic := &Topic{tenantID: 7, cat: types.TopicCatGrp, xoriginal: "grpTest"}
|
|
msg := &ServerComMessage{Data: &MsgServerData{Topic: "grpTest"}}
|
|
topic.prepareBroadcastableMessage(msg, types.ZeroUid, false)
|
|
if msg.TenantID != topic.tenantID {
|
|
t.Fatalf("message tenant = %d, want %d", msg.TenantID, topic.tenantID)
|
|
}
|
|
}
|