anyproto/any-sync
1
0
Fork 0
mirror of https://github.com/anyproto/any-sync.git synced 2025-06-08 14:07:02 +09:00
Code Activity

Request manager and context tests

Browse source
This commit is contained in:
mcrakhman 2024-08-05 18:43:51 +02:00
parent f4beb91316
commit 120561a7f6
No known key found for this signature in database
GPG key ID: DED12CFEF5B8396B
2 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
commonspace/requestmanager/requestmanager_test.go
View file

@ -66,12 +66,14 @@ func TestRequestManager_SyncRequest(t *testing.T) {
defer fx.stop()
peerId := "PeerId"
peerCtx := peer.CtxWithPeerId(ctx, peerId)
peerMock := mock_peer.NewMockPeer(fx.ctrl)
conn := &drpcconn.Conn{}
msg := &spacesyncproto.ObjectSyncMessage{}
resp := &spacesyncproto.ObjectSyncMessage{}
fx.peerPoolMock.EXPECT().Get(ctx, peerId).Return(peerMock, nil)
fx.clientMock.EXPECT().ObjectSync(ctx, msg).Return(resp, nil)
peerMock.EXPECT().Context().Return(peerCtx).AnyTimes()
peerMock.EXPECT().DoDrpc(ctx, gomock.Any()).DoAndReturn(func(ctx context.Context, drpcHandler func(conn drpc.Conn) error) {
drpcHandler(conn)
}).Return(nil)
@ -86,12 +88,14 @@ func TestRequestManager_SyncRequest(t *testing.T) {
ctx = fx.requestManager.ctx
peerId := "PeerId"
peerCtx := peer.CtxWithPeerId(ctx, peerId)
peerMock := mock_peer.NewMockPeer(fx.ctrl)
conn := &drpcconn.Conn{}
msg := &spacesyncproto.ObjectSyncMessage{}
resp := &spacesyncproto.ObjectSyncMessage{}
fx.peerPoolMock.EXPECT().Get(ctx, peerId).Return(peerMock, nil)
fx.clientMock.EXPECT().ObjectSync(ctx, msg).Return(resp, nil)
peerMock.EXPECT().Context().Return(peerCtx).AnyTimes()
peerMock.EXPECT().DoDrpc(ctx, gomock.Any()).DoAndReturn(func(ctx context.Context, drpcHandler func(conn drpc.Conn) error) {
drpcHandler(conn)
}).Return(nil)

14
net/peer/context_test.go Normal file
View file

@ -0,0 +1,14 @@
package peer
import (
"testing"
"github.com/stretchr/testify/require"
)
func TestCtxProtoVersion(t *testing.T) {
ctx := CtxWithProtoVersion(ctx, 1)
ver, err := CtxProtoVersion(ctx)
require.NoError(t, err)
require.Equal(t, uint32(1), ver)
}