mirror of
https://github.com/hanzoai/psrpc.git
synced 2026-09-30 23:47:22 +00:00
zeekay
24eca416c1
PSRPC for Hanzo LiveKit, forked from livekit/psrpc (Apache-2.0). Two rewrites: github.com/livekit/psrpc -> github.com/hanzoai/psrpc github.com/redis/go-redis/v9 -> github.com/hanzokv/go/v9 psrpc's message bus (bus.go, internal/bus/bus_redis.go, internal/bus/bustest) speaks to a KV over go-redis. After hanzoai/livekit moved to our client, psrpc was the one remaining edge pulling upstream go-redis into that binary — not our code, but our dependency, the same shape redsync had for hanzoai/git. The other livekit/* modules (protocol, mediatransportutil, mageutil) are NOT forked and should not be: none of them touch a KV client, so they are ordinary external dependencies and their names are addresses.
83 lines
1.8 KiB
Go
83 lines
1.8 KiB
Go
// Copyright 2025 LiveKit, Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package bustest
|
|
|
|
import (
|
|
"context"
|
|
"math/rand/v2"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/ory/dockertest/v4"
|
|
|
|
"github.com/hanzoai/psrpc/internal/bus"
|
|
)
|
|
|
|
var (
|
|
baseID = rand.Uint32N(1000)
|
|
servers []serverInfo
|
|
)
|
|
|
|
type serverInfo struct {
|
|
Name string
|
|
Func ServerFunc
|
|
}
|
|
|
|
type ServerFunc func(t testing.TB, pool dockertest.Pool) Server
|
|
|
|
func RegisterServer(name string, fnc ServerFunc) {
|
|
servers = append(servers, serverInfo{
|
|
Name: name,
|
|
Func: fnc,
|
|
})
|
|
}
|
|
|
|
func waitTCPPort(t testing.TB, pool dockertest.Pool, addr string) {
|
|
if err := pool.Retry(context.Background(), 0, func() error {
|
|
conn, err := net.Dial("tcp", addr)
|
|
if err != nil {
|
|
t.Log(err)
|
|
return err
|
|
}
|
|
_ = conn.Close()
|
|
return nil
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func Docker(t testing.TB) dockertest.Pool {
|
|
pool, err := dockertest.NewPool(context.Background(), "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return pool
|
|
}
|
|
|
|
type Server interface {
|
|
Connect(t testing.TB) bus.MessageBus
|
|
}
|
|
|
|
func TestAll(t *testing.T, test func(t *testing.T, bus func(t testing.TB) bus.MessageBus)) {
|
|
pool := Docker(t)
|
|
for _, c := range servers {
|
|
c := c
|
|
t.Run(c.Name, func(t *testing.T) {
|
|
s := c.Func(t, pool)
|
|
test(t, s.Connect)
|
|
})
|
|
}
|
|
}
|