Test privacy modes
This commit is contained in:
parent
72e7d68176
commit
0fa646ccdf
|
@ -21,76 +21,180 @@ import (
|
|||
|
||||
// this tests the new room.members call
|
||||
func TestRoomMembers(t *testing.T) {
|
||||
testInit(t)
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
r := require.New(t)
|
||||
|
||||
appKey := make([]byte, 32)
|
||||
rand.Read(appKey)
|
||||
|
||||
netOpts := []roomsrv.Option{
|
||||
roomsrv.WithAppKey(appKey),
|
||||
roomsrv.WithContext(ctx),
|
||||
}
|
||||
|
||||
session := makeNamedTestBot(t, "srv", ctx, netOpts)
|
||||
|
||||
aliceKey, err := keys.NewKeyPair(nil)
|
||||
r.NoError(err)
|
||||
|
||||
bobKey, err := keys.NewKeyPair(nil)
|
||||
r.NoError(err)
|
||||
|
||||
bobSession := makeNamedTestBot(t, "bob", ctx, append(netOpts,
|
||||
roomsrv.WithKeyPair(bobKey),
|
||||
))
|
||||
|
||||
_, err = session.srv.Members.Add(ctx, aliceKey.Feed, roomdb.RoleMember)
|
||||
r.NoError(err)
|
||||
|
||||
_, err = session.srv.Members.Add(ctx, bobKey.Feed, roomdb.RoleMember)
|
||||
r.NoError(err)
|
||||
|
||||
// allow bots to dial the remote
|
||||
// side-effect of re-using a room-server as the client
|
||||
_, err = bobSession.srv.Members.Add(ctx, session.srv.Whoami(), roomdb.RoleMember)
|
||||
r.NoError(err)
|
||||
|
||||
err = bobSession.srv.Network.Connect(ctx, session.srv.Network.GetListenAddr())
|
||||
r.NoError(err, "connect A to the Server")
|
||||
|
||||
t.Log("letting handshaking settle..")
|
||||
time.Sleep(1 * time.Second)
|
||||
|
||||
clientForServer, ok := bobSession.srv.Network.GetEndpointFor(session.srv.Whoami())
|
||||
r.True(ok)
|
||||
|
||||
src, err := clientForServer.Source(ctx, muxrpc.TypeString, muxrpc.Method{"room", "members"})
|
||||
r.NoError(err)
|
||||
|
||||
var responses []server.Member
|
||||
for src.Next(ctx) {
|
||||
bytes, err := src.Bytes()
|
||||
r.NoError(err)
|
||||
|
||||
var members []server.Member
|
||||
err = json.Unmarshal(bytes, &members)
|
||||
r.NoError(err)
|
||||
responses = append(responses, members...)
|
||||
}
|
||||
|
||||
r.Equal(
|
||||
[]server.Member{
|
||||
{
|
||||
ID: aliceKey.Feed,
|
||||
},
|
||||
{
|
||||
ID: bobKey.Feed,
|
||||
},
|
||||
testCases := []struct {
|
||||
Name string
|
||||
PrivacyMode roomdb.PrivacyMode
|
||||
ExternalCanConnect bool
|
||||
ExternalCanList bool
|
||||
}{
|
||||
{
|
||||
Name: "open",
|
||||
PrivacyMode: roomdb.ModeOpen,
|
||||
ExternalCanConnect: true,
|
||||
ExternalCanList: true,
|
||||
},
|
||||
responses,
|
||||
)
|
||||
{
|
||||
Name: "community",
|
||||
PrivacyMode: roomdb.ModeCommunity,
|
||||
ExternalCanConnect: true,
|
||||
ExternalCanList: false,
|
||||
},
|
||||
{
|
||||
Name: "restricted",
|
||||
PrivacyMode: roomdb.ModeRestricted,
|
||||
ExternalCanConnect: false,
|
||||
ExternalCanList: false,
|
||||
},
|
||||
}
|
||||
|
||||
for i := range testCases {
|
||||
testCase := testCases[i]
|
||||
|
||||
t.Run(testCase.Name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
r := require.New(t)
|
||||
|
||||
testInit(t)
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
t.Cleanup(cancel)
|
||||
|
||||
appKey := make([]byte, 32)
|
||||
rand.Read(appKey)
|
||||
|
||||
netOpts := []roomsrv.Option{
|
||||
roomsrv.WithAppKey(appKey),
|
||||
roomsrv.WithContext(ctx),
|
||||
}
|
||||
|
||||
session := makeNamedTestBot(t, "srv", ctx, netOpts)
|
||||
|
||||
aliceKey, err := keys.NewKeyPair(nil)
|
||||
r.NoError(err)
|
||||
|
||||
bobKey, err := keys.NewKeyPair(nil)
|
||||
r.NoError(err)
|
||||
|
||||
_, err = session.srv.Members.Add(ctx, aliceKey.Feed, roomdb.RoleMember)
|
||||
r.NoError(err)
|
||||
|
||||
_, err = session.srv.Members.Add(ctx, bobKey.Feed, roomdb.RoleMember)
|
||||
r.NoError(err)
|
||||
|
||||
err = session.srv.Config.SetPrivacyMode(ctx, testCase.PrivacyMode)
|
||||
r.NoError(err)
|
||||
|
||||
t.Run("member", func(t *testing.T) {
|
||||
t.Parallel()
|
||||
r := require.New(t)
|
||||
|
||||
bobSession := makeNamedTestBot(t, "bob", ctx, append(netOpts,
|
||||
roomsrv.WithKeyPair(bobKey),
|
||||
))
|
||||
|
||||
// allow bots to dial the remote
|
||||
// side-effect of re-using a room-server as the client
|
||||
_, err = bobSession.srv.Members.Add(ctx, session.srv.Whoami(), roomdb.RoleMember)
|
||||
r.NoError(err)
|
||||
|
||||
err = bobSession.srv.Network.Connect(ctx, session.srv.Network.GetListenAddr())
|
||||
r.NoError(err, "connect bob to the Server")
|
||||
|
||||
t.Log("letting handshaking settle..")
|
||||
time.Sleep(1 * time.Second)
|
||||
|
||||
clientForServer, ok := bobSession.srv.Network.GetEndpointFor(session.srv.Whoami())
|
||||
r.True(ok)
|
||||
|
||||
src, err := clientForServer.Source(ctx, muxrpc.TypeString, muxrpc.Method{"room", "members"})
|
||||
r.NoError(err)
|
||||
|
||||
var responses []server.Member
|
||||
for src.Next(ctx) {
|
||||
bytes, err := src.Bytes()
|
||||
r.NoError(err)
|
||||
|
||||
var members []server.Member
|
||||
err = json.Unmarshal(bytes, &members)
|
||||
r.NoError(err)
|
||||
responses = append(responses, members...)
|
||||
}
|
||||
|
||||
r.NoError(src.Err())
|
||||
r.Equal(
|
||||
[]server.Member{
|
||||
{
|
||||
ID: aliceKey.Feed,
|
||||
},
|
||||
{
|
||||
ID: bobKey.Feed,
|
||||
},
|
||||
},
|
||||
responses,
|
||||
)
|
||||
})
|
||||
|
||||
t.Run("external", func(t *testing.T) {
|
||||
t.Parallel()
|
||||
r := require.New(t)
|
||||
|
||||
carolKey, err := keys.NewKeyPair(nil)
|
||||
r.NoError(err)
|
||||
|
||||
carolSession := makeNamedTestBot(t, "carol", ctx, append(netOpts,
|
||||
roomsrv.WithKeyPair(carolKey),
|
||||
))
|
||||
|
||||
// allow bots to dial the remote
|
||||
// side-effect of re-using a room-server as the client
|
||||
_, err = carolSession.srv.Members.Add(ctx, session.srv.Whoami(), roomdb.RoleMember)
|
||||
r.NoError(err)
|
||||
|
||||
err = carolSession.srv.Network.Connect(ctx, session.srv.Network.GetListenAddr())
|
||||
r.NoError(err, "connect carol to the Server")
|
||||
|
||||
t.Log("letting handshaking settle..")
|
||||
time.Sleep(1 * time.Second)
|
||||
|
||||
clientForServer, ok := carolSession.srv.Network.GetEndpointFor(session.srv.Whoami())
|
||||
if testCase.ExternalCanConnect {
|
||||
r.True(ok)
|
||||
} else {
|
||||
r.False(ok)
|
||||
return
|
||||
}
|
||||
|
||||
src, err := clientForServer.Source(ctx, muxrpc.TypeString, muxrpc.Method{"room", "members"})
|
||||
r.NoError(err)
|
||||
|
||||
var responses []server.Member
|
||||
for src.Next(ctx) {
|
||||
bytes, err := src.Bytes()
|
||||
r.NoError(err)
|
||||
|
||||
var members []server.Member
|
||||
err = json.Unmarshal(bytes, &members)
|
||||
r.NoError(err)
|
||||
responses = append(responses, members...)
|
||||
}
|
||||
|
||||
if testCase.ExternalCanList {
|
||||
r.NoError(src.Err())
|
||||
r.Equal(
|
||||
[]server.Member{
|
||||
{
|
||||
ID: aliceKey.Feed,
|
||||
},
|
||||
{
|
||||
ID: bobKey.Feed,
|
||||
},
|
||||
},
|
||||
responses,
|
||||
)
|
||||
} else {
|
||||
r.EqualError(src.Err(), "muxrpc CallError: Error - external user are not allowed to list members: roomdb: object not found")
|
||||
}
|
||||
})
|
||||
})
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue