diff --git a/services/tunnelbroker/test/DatabaseManagerTest.cpp b/services/tunnelbroker/test/DatabaseManagerTest.cpp index 2a41552a3..12dbb7ed7 100644 --- a/services/tunnelbroker/test/DatabaseManagerTest.cpp +++ b/services/tunnelbroker/test/DatabaseManagerTest.cpp @@ -1,24 +1,98 @@ #include "DatabaseManager.h" #include "ConfigManager.h" #include "Constants.h" #include "Tools.h" #include #include #include #include using namespace comm::network; class DatabaseManagerTest : public testing::Test { protected: virtual void SetUp() { config::ConfigManager::getInstance().load(); Aws::InitAPI({}); } virtual void TearDown() { Aws::ShutdownAPI({}); } }; + +TEST_F(DatabaseManagerTest, PutAndFoundMessageItemsStaticDataIsSame) { + const database::MessageItem item( + "bc0c1aa2-bf09-11ec-9d64-0242ac120002", + "mobile:EMQNoQ7b2ueEmQ4QsevRWlXxFCNt055y20T1PHdoYAQRt0S6TLzZWNM6XSvdWqxm", + "web:JouLWf84zqRIsjBdHLOcHS9M4eSCz7VF84wT1uOD83u1qxDAqmqI4swmxNINjuhd", + "lYlNcO6RR4i9UW3G1DGjdJTRRGbqtPya2aj94ZRjIGZWoHwT5MB9ciAgnQf2VafYb9Tl" + "8SZkX37tg4yZ9pOb4lqslY4g4h58OmWjumghVRvrPUZDalUuK8OLs1Qoengpu9wccxAk" + "Bti2leDTNeiJDy36NnwS9aCIUc0ozsMvXfX1gWdBdmKbiRG1LvpNd6S7BNGG7Zly5zYj" + "xz7s6ZUSDoFfZe3eJWQ15ngYhgMw1TsfbECnMVQTYvY6OyqWPBQi5wiftFcluoxor8G5" + "RJ1NEDQq2q2FRfWjNHLhky92C2C7Nnfe4oVzSinfC1319uUkNLpSzI4MvEMi6g5Ukbl7" + "iGhpnX7Hp4xpBL3h2IkvGviDRQ98UvW0ugwUuPxm1NOQpjLG5dPoqQ0jrMst0Bl5rgPw" + "ajjNGsUWmp9r0ST0wRQXrQcY30PoSoqKSlCEgFMLzHWLrPQ86QFyCICismGSe7iBIqdD" + "6d37StvXBzfJoZVU79UeOF2bFvb3DNoArEOe", + "7s6ZUSDoFfZe3eJWQ15ngYhgMw1TsfbECnMVQTYvY6OyqWPBQi5wiftFcluoxor8", + static_cast(std::time(0)) + 600); + EXPECT_EQ( + database::DatabaseManager::getInstance().isTableAvailable( + item.getTableName()), + true); + database::DatabaseManager::getInstance().putMessageItem(item); + std::shared_ptr foundItem = + database::DatabaseManager::getInstance().findMessageItem( + item.getMessageID()); + EXPECT_NE(foundItem, nullptr); + EXPECT_EQ(item.getFromDeviceID(), foundItem->getFromDeviceID()); + EXPECT_EQ(item.getToDeviceID(), foundItem->getToDeviceID()); + EXPECT_EQ(item.getPayload(), foundItem->getPayload()); + EXPECT_EQ(item.getBlobHashes(), foundItem->getBlobHashes()); + EXPECT_EQ(item.getExpire(), foundItem->getExpire()); + database::DatabaseManager::getInstance().removeMessageItem( + item.getMessageID()); +} + +TEST_F(DatabaseManagerTest, PutAndFoundMessageItemsGeneratedDataIsSame) { + const database::MessageItem item( + tools::generateUUID(), + "mobile:" + tools::generateRandomString(DEVICEID_CHAR_LENGTH), + "web:" + tools::generateRandomString(DEVICEID_CHAR_LENGTH), + tools::generateRandomString(256), + tools::generateRandomString(256), + static_cast(std::time(0)) + 600); + EXPECT_EQ( + database::DatabaseManager::getInstance().isTableAvailable( + item.getTableName()), + true); + database::DatabaseManager::getInstance().putMessageItem(item); + std::shared_ptr foundItem = + database::DatabaseManager::getInstance().findMessageItem( + item.getMessageID()); + EXPECT_NE(foundItem, nullptr); + EXPECT_EQ(item.getFromDeviceID(), foundItem->getFromDeviceID()) + << "Generated FromDeviceID \"" << item.getFromDeviceID() + << "\" differs from what is found in the database " + << foundItem->getFromDeviceID(); + EXPECT_EQ(item.getToDeviceID(), foundItem->getToDeviceID()) + << "Generated ToDeviceID \"" << item.getToDeviceID() + << "\" differs from what is found in the database " + << foundItem->getToDeviceID(); + EXPECT_EQ(item.getPayload(), foundItem->getPayload()) + << "Generated Payload \"" << item.getPayload() + << "\" differs from what is found in the database " + << foundItem->getPayload(); + EXPECT_EQ(item.getBlobHashes(), foundItem->getBlobHashes()) + << "Generated BlobHashes \"" << item.getBlobHashes() + << "\" differs from what is found in the database " + << foundItem->getBlobHashes(); + EXPECT_EQ(item.getExpire(), foundItem->getExpire()) + << "Generated Expire time \"" << item.getExpire() + << "\" differs from what is found in the database " + << foundItem->getExpire(); + database::DatabaseManager::getInstance().removeMessageItem( + item.getMessageID()); +}