diff --git a/services/backup/src/BackupServiceImpl.h b/services/backup/src/BackupServiceImpl.h --- a/services/backup/src/BackupServiceImpl.h +++ b/services/backup/src/BackupServiceImpl.h @@ -31,6 +31,10 @@ grpc::ServerWriteReactor *PullBackup( grpc::CallbackServerContext *context, const backup::PullBackupRequest *request) override; + + grpc::ServerReadReactor *AddAttachment( + grpc::CallbackServerContext *context, + google::protobuf::Empty *response) override; }; } // namespace network diff --git a/services/backup/src/BackupServiceImpl.cpp b/services/backup/src/BackupServiceImpl.cpp --- a/services/backup/src/BackupServiceImpl.cpp +++ b/services/backup/src/BackupServiceImpl.cpp @@ -1,5 +1,6 @@ #include "BackupServiceImpl.h" +#include "AddAttachmentReactor.h" #include "CreateNewBackupReactor.h" #include "PullBackupReactor.h" #include "RecoverBackupKeyReactor.h" @@ -47,5 +48,12 @@ return reactor; } +grpc::ServerReadReactor * +BackupServiceImpl::AddAttachment( + grpc::CallbackServerContext *context, + google::protobuf::Empty *response) { + return new reactor::AddAttachmentReactor(response); +} + } // namespace network } // namespace comm diff --git a/services/backup/test/DatabaseManagerTest.cpp b/services/backup/test/DatabaseManagerTest.cpp --- a/services/backup/test/DatabaseManagerTest.cpp +++ b/services/backup/test/DatabaseManagerTest.cpp @@ -71,11 +71,15 @@ for (const std::string &logID : logIDs1) { DatabaseManager::getInstance().putLogItem( generateLogItem(backupID1, logID)); + EXPECT_NE( + DatabaseManager::getInstance().findLogItem(backupID1, logID), nullptr); } std::vector logIDs2 = {"log021", "log022"}; for (const std::string &logID : logIDs2) { DatabaseManager::getInstance().putLogItem( generateLogItem(backupID2, logID)); + EXPECT_NE( + DatabaseManager::getInstance().findLogItem(backupID2, logID), nullptr); } std::vector> items1 = @@ -90,6 +94,9 @@ for (size_t i = 0; i < items1.size(); ++i) { EXPECT_EQ(logIDs1.at(i), items1.at(i)->getLogID()); DatabaseManager::getInstance().removeLogItem(items1.at(i)); + EXPECT_EQ( + DatabaseManager::getInstance().findLogItem(backupID1, logIDs1.at(i)), + nullptr); } EXPECT_EQ( DatabaseManager::getInstance().findLogItemsForBackup(backupID1).size(), @@ -98,6 +105,9 @@ for (size_t i = 0; i < items2.size(); ++i) { EXPECT_EQ(logIDs2.at(i), items2.at(i)->getLogID()); DatabaseManager::getInstance().removeLogItem(items2.at(i)); + EXPECT_EQ( + DatabaseManager::getInstance().findLogItem(backupID2, logIDs2.at(i)), + nullptr); } EXPECT_EQ( DatabaseManager::getInstance().findLogItemsForBackup(backupID2).size(),