From 8310cb3c05550ee941d17a941a6b7acf53bc677f Mon Sep 17 00:00:00 2001 From: mcrakhman Date: Tue, 6 Jun 2023 22:08:06 +0200 Subject: [PATCH] Fix sync protocol integration tests --- .../object/tree/synctree/utils_test.go | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) diff --git a/commonspace/object/tree/synctree/utils_test.go b/commonspace/object/tree/synctree/utils_test.go index 3ac8152a..6aab234b 100644 --- a/commonspace/object/tree/synctree/utils_test.go +++ b/commonspace/object/tree/synctree/utils_test.go @@ -240,9 +240,6 @@ func (h *testSyncHandler) HandleMessage(ctx context.Context, senderId string, re } func (h *testSyncHandler) manager() *requestPeerManager { - if h.SyncHandler != nil { - return h.manager() - } return h.peerManager } @@ -275,10 +272,18 @@ func (h *testSyncHandler) run(ctx context.Context, t *testing.T, wg *sync.WaitGr h.tree().Unlock() continue } - err = h.HandleMessage(ctx, res.senderId, res.msg) - if err != nil { - fmt.Println("error handling message", err.Error()) - continue + if res.description().name == "FullSyncRequest" { + resp, err := h.HandleRequest(ctx, res.senderId, res.msg) + if err != nil { + fmt.Println("error handling request", err.Error()) + continue + } + h.peerManager.SendPeer(ctx, res.senderId, resp) + } else { + err = h.HandleMessage(ctx, res.senderId, res.msg) + if err != nil { + fmt.Println("error handling message", err.Error()) + } } } }()