diff options
author | buttercat1791 <mjjurkoic@gmail.com> | 2024-05-05 12:32:42 -0500 |
---|---|---|
committer | buttercat1791 <mjjurkoic@gmail.com> | 2024-05-05 12:32:42 -0500 |
commit | 8a170b56b5c53c658af14f82111254e05062a23c (patch) | |
tree | 2b6863603174612d508f55591744e1c1bbc4c7d6 /test/nostr_service_test.cpp | |
parent | ae458b29b7c5f9124e6cc4499bed60c865d7badd (diff) |
Close relays after batch query and update unit test
Diffstat (limited to 'test/nostr_service_test.cpp')
-rw-r--r-- | test/nostr_service_test.cpp | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/test/nostr_service_test.cpp b/test/nostr_service_test.cpp index b2f6876..14eb048 100644 --- a/test/nostr_service_test.cpp +++ b/test/nostr_service_test.cpp @@ -25,6 +25,7 @@ using std::unordered_map; using std::vector; using ::testing::_; using ::testing::Args; +using ::testing::HasSubstr; using ::testing::Invoke; using ::testing::Return; using ::testing::Truly; @@ -711,7 +712,6 @@ TEST_F(NostrServiceTest, PublishEvent_CorrectlyIndicates_EventRejectedBySomeRela ASSERT_EQ(failures[0], defaultTestRelays[1]); }; -// TODO: Add unit tests for queries. TEST_F(NostrServiceTest, QueryRelays_ReturnsEvents_UpToEOSE) { mutex connectionStatusMutex; @@ -753,7 +753,8 @@ TEST_F(NostrServiceTest, QueryRelays_ReturnsEvents_UpToEOSE) signedTestEvents.push_back(signedEvent); } - EXPECT_CALL(*mockClient, send(_, _, _)) + // Expect the query messages. + EXPECT_CALL(*mockClient, send(HasSubstr("REQ"), _, _)) .Times(2) .WillRepeatedly(Invoke([&testEvents, &signer]( string message, @@ -776,6 +777,9 @@ TEST_F(NostrServiceTest, QueryRelays_ReturnsEvents_UpToEOSE) return make_tuple(uri, true); })); + // Expect the close subscription messages after the client receives events. + // TODO: Expect close message. + EXPECT_CALL(*mockClient, send(HasSubstr("CLOSE"), _)).Times(2); auto filters = make_shared<nostr::Filters>(getKind0And1TestFilters()); auto results = nostrService->queryRelays(filters); @@ -886,7 +890,6 @@ TEST_F(NostrServiceTest, QueryRelays_CallsHandler_WithReturnedEvents) [&generatedSubscriptionId, &eoseReceivedPromise, &eoseCount] (const string& subscriptionId) { - std::cout << "EOSE received for subscription ID: " << subscriptionId << std::endl; ASSERT_STREQ(subscriptionId.c_str(), generatedSubscriptionId.c_str()); if (++eoseCount == 2) @@ -900,4 +903,5 @@ TEST_F(NostrServiceTest, QueryRelays_CallsHandler_WithReturnedEvents) }; // TODO: Add unit tests for closing subscriptions. + } // namespace nostr_test |