mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2025-10-25 04:17:20 +02:00
568 lines
14 KiB
C++
568 lines
14 KiB
C++
//
|
|
// MongoDBTest.cpp
|
|
//
|
|
// Copyright (c) 2004-2006, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#include "Poco/DateTime.h"
|
|
#include "Poco/ObjectPool.h"
|
|
#include "Poco/MongoDB/Array.h"
|
|
#include "Poco/MongoDB/InsertRequest.h"
|
|
#include "Poco/MongoDB/QueryRequest.h"
|
|
#include "Poco/MongoDB/DeleteRequest.h"
|
|
#include "Poco/MongoDB/GetMoreRequest.h"
|
|
#include "Poco/MongoDB/PoolableConnectionFactory.h"
|
|
#include "Poco/MongoDB/Database.h"
|
|
#include "Poco/MongoDB/Connection.h"
|
|
#include "Poco/MongoDB/Cursor.h"
|
|
#include "Poco/MongoDB/ObjectId.h"
|
|
#include "Poco/MongoDB/Binary.h"
|
|
#include "Poco/Net/NetException.h"
|
|
#include "Poco/UUIDGenerator.h"
|
|
#include "MongoDBTest.h"
|
|
#include "CppUnit/TestCaller.h"
|
|
#include "CppUnit/TestSuite.h"
|
|
#include <iostream>
|
|
|
|
|
|
using namespace Poco::MongoDB;
|
|
|
|
|
|
Poco::MongoDB::Connection::Ptr MongoDBTest::_mongo;
|
|
Poco::Int64 MongoDBTest::_wireVersion {0};
|
|
|
|
|
|
MongoDBTest::MongoDBTest(const std::string& name):
|
|
CppUnit::TestCase("MongoDB")
|
|
{
|
|
}
|
|
|
|
|
|
MongoDBTest::~MongoDBTest()
|
|
{
|
|
}
|
|
|
|
|
|
void MongoDBTest::setUp()
|
|
{
|
|
}
|
|
|
|
|
|
void MongoDBTest::tearDown()
|
|
{
|
|
}
|
|
|
|
|
|
void MongoDBTest::testInsertRequest()
|
|
{
|
|
Poco::MongoDB::Document::Ptr player = new Poco::MongoDB::Document();
|
|
player->add("lastname", std::string("Braem"));
|
|
player->add("firstname", std::string("Franky"));
|
|
|
|
Poco::DateTime birthdate;
|
|
birthdate.assign(1969, 3, 9);
|
|
player->add("birthdate", birthdate.timestamp());
|
|
|
|
player->add("start", 1993);
|
|
player->add("active", false);
|
|
|
|
Poco::DateTime now;
|
|
player->add("lastupdated", now.timestamp());
|
|
|
|
player->add("unknown", NullValue());
|
|
|
|
Poco::MongoDB::InsertRequest request("team.players");
|
|
request.documents().push_back(player);
|
|
_mongo->sendRequest(request);
|
|
}
|
|
|
|
void MongoDBTest::testArray()
|
|
{
|
|
Poco::MongoDB::Array::Ptr arr = new Poco::MongoDB::Array();
|
|
|
|
arr->add(std::string("First"));
|
|
|
|
Poco::DateTime birthdate;
|
|
birthdate.assign(1969, 3, 9);
|
|
arr->add(birthdate.timestamp());
|
|
|
|
arr->add(static_cast<Poco::Int32>(1993));
|
|
arr->add(false);
|
|
|
|
// Document-style interface
|
|
arr->add("4", "12.4E");
|
|
|
|
assertEqual(arr->size(), 5);
|
|
assertTrue(arr->exists("0"));
|
|
assertTrue(arr->exists("1"));
|
|
assertTrue(arr->exists("2"));
|
|
assertTrue(arr->exists("3"));
|
|
assertTrue(arr->exists("4"));
|
|
assertFalse(arr->exists("5"));
|
|
|
|
assertEqual(arr->get<std::string>(0), "First");
|
|
assertEqual(arr->get<Poco::Timestamp>(1).raw(), birthdate.timestamp().raw());
|
|
assertEqual(arr->get<Poco::Int32>(2), 1993);
|
|
assertEqual(arr->get<bool>(3), false);
|
|
assertEqual(arr->get<std::string>(4), "12.4E");
|
|
|
|
// Document-style interface
|
|
assertEqual(arr->get<Poco::Int32>("2"), 1993);
|
|
assertEqual(arr->get<std::string>("4"), "12.4E");
|
|
|
|
}
|
|
|
|
|
|
void MongoDBTest::testQueryRequest()
|
|
{
|
|
Poco::MongoDB::QueryRequest request("team.players");
|
|
request.selector().add("lastname" , std::string("Braem"));
|
|
request.setNumberToReturn(1);
|
|
|
|
Poco::MongoDB::ResponseMessage response;
|
|
|
|
_mongo->sendRequest(request, response);
|
|
|
|
if ( response.documents().size() > 0 )
|
|
{
|
|
Poco::MongoDB::Document::Ptr doc = response.documents()[0];
|
|
|
|
try
|
|
{
|
|
std::string lastname = doc->get<std::string>("lastname");
|
|
assertTrue (lastname.compare("Braem") == 0);
|
|
std::string firstname = doc->get<std::string>("firstname");
|
|
assertTrue (firstname.compare("Franky") == 0);
|
|
Poco::Timestamp birthDateTimestamp = doc->get<Poco::Timestamp>("birthdate");
|
|
Poco::DateTime birthDate(birthDateTimestamp);
|
|
assertTrue (birthDate.year() == 1969 && birthDate.month() == 3 && birthDate.day() == 9);
|
|
Poco::Timestamp lastupdatedTimestamp = doc->get<Poco::Timestamp>("lastupdated");
|
|
assertTrue (doc->isType<NullValue>("unknown"));
|
|
bool active = doc->get<bool>("active");
|
|
assertTrue (!active);
|
|
|
|
std::string id = doc->get("_id")->toString();
|
|
}
|
|
catch(Poco::NotFoundException& nfe)
|
|
{
|
|
fail(nfe.message() + " not found.");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
fail("No document returned");
|
|
}
|
|
}
|
|
|
|
|
|
void MongoDBTest::testDBQueryRequest()
|
|
{
|
|
Database db("team");
|
|
Poco::SharedPtr<Poco::MongoDB::QueryRequest> request = db.createQueryRequest("players");
|
|
request->selector().add("lastname" , std::string("Braem"));
|
|
|
|
Poco::MongoDB::ResponseMessage response;
|
|
_mongo->sendRequest(*request, response);
|
|
|
|
if ( response.documents().size() > 0 )
|
|
{
|
|
Poco::MongoDB::Document::Ptr doc = response.documents()[0];
|
|
|
|
try
|
|
{
|
|
std::string lastname = doc->get<std::string>("lastname");
|
|
assertTrue (lastname.compare("Braem") == 0);
|
|
std::string firstname = doc->get<std::string>("firstname");
|
|
assertTrue (firstname.compare("Franky") == 0);
|
|
Poco::Timestamp birthDateTimestamp = doc->get<Poco::Timestamp>("birthdate");
|
|
Poco::DateTime birthDate(birthDateTimestamp);
|
|
assertTrue (birthDate.year() == 1969 && birthDate.month() == 3 && birthDate.day() == 9);
|
|
Poco::Timestamp lastupdatedTimestamp = doc->get<Poco::Timestamp>("lastupdated");
|
|
assertTrue (doc->isType<NullValue>("unknown"));
|
|
|
|
std::string id = doc->get("_id")->toString();
|
|
}
|
|
catch(Poco::NotFoundException& nfe)
|
|
{
|
|
fail(nfe.message() + " not found.");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
fail("No document returned");
|
|
}
|
|
}
|
|
|
|
|
|
void MongoDBTest::testCountCommand()
|
|
{
|
|
Poco::MongoDB::QueryRequest request("team.$cmd");
|
|
request.setNumberToReturn(1);
|
|
request.selector().add("count", std::string("players"));
|
|
|
|
Poco::MongoDB::ResponseMessage response;
|
|
|
|
_mongo->sendRequest(request, response);
|
|
|
|
if ( response.documents().size() > 0 )
|
|
{
|
|
Poco::MongoDB::Document::Ptr doc = response.documents()[0];
|
|
assertTrue (doc->getInteger("n") == 1);
|
|
}
|
|
else
|
|
{
|
|
fail("Didn't get a response from the count command");
|
|
}
|
|
}
|
|
|
|
|
|
void MongoDBTest::testDBCountCommand()
|
|
{
|
|
Poco::MongoDB::Database db("team");
|
|
Poco::SharedPtr<Poco::MongoDB::QueryRequest> request = db.createCountRequest("players");
|
|
|
|
Poco::MongoDB::ResponseMessage response;
|
|
_mongo->sendRequest(*request, response);
|
|
|
|
if ( response.documents().size() > 0 )
|
|
{
|
|
Poco::MongoDB::Document::Ptr doc = response.documents()[0];
|
|
assertTrue (doc->getInteger("n") == 1);
|
|
}
|
|
else
|
|
{
|
|
fail("Didn't get a response from the count command");
|
|
}
|
|
}
|
|
|
|
|
|
void MongoDBTest::testDBCount2Command()
|
|
{
|
|
Poco::MongoDB::Database db("team");
|
|
Poco::Int64 count = db.count(*_mongo, "players");
|
|
assertTrue (count == 1);
|
|
}
|
|
|
|
|
|
void MongoDBTest::testDeleteRequest()
|
|
{
|
|
Poco::MongoDB::DeleteRequest request("team.players");
|
|
request.selector().add("lastname", std::string("Braem"));
|
|
|
|
_mongo->sendRequest(request);
|
|
}
|
|
|
|
|
|
void MongoDBTest::testCursorRequest()
|
|
{
|
|
Poco::MongoDB::Database db("team");
|
|
|
|
Poco::SharedPtr<Poco::MongoDB::DeleteRequest> deleteRequest = db.createDeleteRequest("numbers");
|
|
_mongo->sendRequest(*deleteRequest);
|
|
|
|
Poco::SharedPtr<Poco::MongoDB::InsertRequest> insertRequest = db.createInsertRequest("numbers");
|
|
for(int i = 0; i < 10000; ++i)
|
|
{
|
|
Document::Ptr doc = new Document();
|
|
doc->add("number", i);
|
|
insertRequest->documents().push_back(doc);
|
|
}
|
|
_mongo->sendRequest(*insertRequest);
|
|
|
|
Poco::Int64 count = db.count(*_mongo, "numbers");
|
|
assertTrue (count == 10000);
|
|
|
|
Poco::MongoDB::Cursor cursor("team", "numbers");
|
|
|
|
int n = 0;
|
|
Poco::MongoDB::ResponseMessage& response = cursor.next(*_mongo);
|
|
while(1)
|
|
{
|
|
n += static_cast<int>(response.documents().size());
|
|
if ( response.cursorID() == 0 )
|
|
break;
|
|
response = cursor.next(*_mongo);
|
|
}
|
|
assertTrue (n == 10000);
|
|
|
|
Poco::MongoDB::QueryRequest drop("team.$cmd");
|
|
drop.setNumberToReturn(1);
|
|
drop.selector().add("drop", std::string("numbers"));
|
|
|
|
Poco::MongoDB::ResponseMessage responseDrop;
|
|
_mongo->sendRequest(drop, responseDrop);
|
|
}
|
|
|
|
void MongoDBTest::testBuildInfo()
|
|
{
|
|
// build info can be issued on "config" system database
|
|
|
|
Poco::MongoDB::Database db("config");
|
|
try
|
|
{
|
|
Poco::MongoDB::Document::Ptr doc = db.queryBuildInfo(*_mongo);
|
|
std::cout << doc->toString(2);
|
|
}
|
|
catch(Poco::NotImplementedException& nie)
|
|
{
|
|
std::cout << nie.message() << std::endl;
|
|
}
|
|
}
|
|
|
|
void MongoDBTest::testHello()
|
|
{
|
|
Poco::MongoDB::Database db("config");
|
|
try
|
|
{
|
|
Poco::MongoDB::Document::Ptr doc = db.queryServerHello(*_mongo);
|
|
std::cout << doc->toString(2);
|
|
}
|
|
catch(Poco::NotImplementedException& nie)
|
|
{
|
|
std::cout << nie.message() << std::endl;
|
|
}
|
|
}
|
|
|
|
|
|
void MongoDBTest::testConnectionPool()
|
|
{
|
|
#if POCO_OS == POCO_OS_ANDROID
|
|
std::string host = "10.0.2.2";
|
|
#else
|
|
std::string host = "127.0.0.1";
|
|
#endif
|
|
|
|
Poco::Net::SocketAddress sa(host, 27017);
|
|
Poco::PoolableObjectFactory<Poco::MongoDB::Connection, Poco::MongoDB::Connection::Ptr> factory(sa);
|
|
Poco::ObjectPool<Poco::MongoDB::Connection, Poco::MongoDB::Connection::Ptr> pool(factory, 10, 15);
|
|
|
|
Poco::MongoDB::PooledConnection pooledConnection(pool);
|
|
|
|
Poco::MongoDB::QueryRequest request("team.$cmd");
|
|
request.setNumberToReturn(1);
|
|
request.selector().add("count", std::string("players"));
|
|
|
|
Poco::MongoDB::ResponseMessage response;
|
|
((Connection::Ptr) pooledConnection)->sendRequest(request, response);
|
|
|
|
if ( response.documents().size() > 0 )
|
|
{
|
|
Poco::MongoDB::Document::Ptr doc = response.documents()[0];
|
|
assertTrue (doc->getInteger("n") == 1);
|
|
}
|
|
else
|
|
{
|
|
fail("Didn't get a response from the count command");
|
|
}
|
|
}
|
|
|
|
|
|
void MongoDBTest::testObjectID()
|
|
{
|
|
ObjectId oid("536aeebba081de6815000002");
|
|
std::string str2 = oid.toString();
|
|
assertTrue (str2 == "536aeebba081de6815000002");
|
|
}
|
|
|
|
|
|
void MongoDBTest::testCommand() {
|
|
Poco::MongoDB::Database db("team");
|
|
Poco::SharedPtr<Poco::MongoDB::QueryRequest> command = db.createCommand();
|
|
command->selector().add("create", "fixCol")
|
|
.add("capped", true)
|
|
.add("max", 1024*1024)
|
|
.add("size", 1024);
|
|
|
|
Poco::MongoDB::ResponseMessage response;
|
|
_mongo->sendRequest(*command, response);
|
|
if ( response.documents().size() > 0 )
|
|
{
|
|
Poco::MongoDB::Document::Ptr doc = response.documents()[0];
|
|
}
|
|
else
|
|
{
|
|
Poco::MongoDB::Document::Ptr lastError = db.getLastErrorDoc(*_mongo);
|
|
fail(lastError->toString(2));
|
|
}
|
|
}
|
|
|
|
|
|
void MongoDBTest::testUUID()
|
|
{
|
|
Poco::MongoDB::Document::Ptr club = new Poco::MongoDB::Document();
|
|
club->add("name", std::string("Barcelona"));
|
|
|
|
Poco::UUIDGenerator generator;
|
|
Poco::UUID uuid = generator.create();
|
|
Poco::MongoDB::Binary::Ptr uuidBinary = new Poco::MongoDB::Binary(uuid);
|
|
club->add("uuid", uuidBinary);
|
|
|
|
Poco::MongoDB::InsertRequest request("team.club");
|
|
request.documents().push_back(club);
|
|
|
|
_mongo->sendRequest(request);
|
|
|
|
Poco::MongoDB::QueryRequest queryReq("team.club");
|
|
queryReq.selector().add("name" , std::string("Barcelona"));
|
|
|
|
Poco::MongoDB::ResponseMessage response;
|
|
_mongo->sendRequest(queryReq, response);
|
|
|
|
if ( response.documents().size() > 0 )
|
|
{
|
|
Poco::MongoDB::Document::Ptr doc = response.documents()[0];
|
|
|
|
try
|
|
{
|
|
std::string name = doc->get<std::string>("name");
|
|
assertTrue (name.compare("Barcelona") == 0);
|
|
|
|
Poco::MongoDB::Binary::Ptr uuidBinary = doc->get<Binary::Ptr>("uuid");
|
|
assertTrue (uuid == uuidBinary->uuid());
|
|
}
|
|
catch(Poco::NotFoundException& nfe)
|
|
{
|
|
fail(nfe.message() + " not found.");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
fail("No document returned");
|
|
}
|
|
|
|
Poco::MongoDB::DeleteRequest delRequest("team.club");
|
|
delRequest.selector().add("name", std::string("Barcelona"));
|
|
_mongo->sendRequest(delRequest);
|
|
}
|
|
|
|
|
|
void MongoDBTest::testConnectURI()
|
|
{
|
|
Poco::MongoDB::Connection conn;
|
|
Poco::MongoDB::Connection::SocketFactory sf;
|
|
|
|
#if POCO_OS == POCO_OS_ANDROID
|
|
std::string host = "10.0.2.2";
|
|
#else
|
|
std::string host = "127.0.0.1";
|
|
#endif
|
|
|
|
conn.connect("mongodb://" + host, sf);
|
|
conn.disconnect();
|
|
|
|
try
|
|
{
|
|
conn.connect("http://" + host, sf);
|
|
fail("invalid URI scheme - must throw");
|
|
}
|
|
catch (Poco::UnknownURISchemeException&)
|
|
{
|
|
}
|
|
|
|
try
|
|
{
|
|
conn.connect("mongodb://" + host + "?ssl=true", sf);
|
|
fail("SSL not supported, must throw");
|
|
}
|
|
catch (Poco::NotImplementedException&)
|
|
{
|
|
}
|
|
|
|
conn.connect("mongodb://" + host + "/admin?ssl=false&connectTimeoutMS=10000&socketTimeoutMS=10000", sf);
|
|
conn.disconnect();
|
|
|
|
try
|
|
{
|
|
conn.connect("mongodb://" + host + "/admin?connectTimeoutMS=foo", sf);
|
|
fail("invalid parameter - must throw");
|
|
}
|
|
catch (Poco::Exception&)
|
|
{
|
|
}
|
|
|
|
#ifdef MONGODB_TEST_AUTH
|
|
conn.connect("mongodb://admin:admin@127.0.0.1/admin", sf);
|
|
conn.disconnect();
|
|
#endif
|
|
}
|
|
|
|
|
|
CppUnit::Test* MongoDBTest::suite()
|
|
{
|
|
#if POCO_OS == POCO_OS_ANDROID
|
|
std::string host = "10.0.2.2";
|
|
#else
|
|
std::string host = "127.0.0.1";
|
|
#endif
|
|
try
|
|
{
|
|
_wireVersion = 0;
|
|
_mongo = new Poco::MongoDB::Connection(host, 27017);
|
|
std::cout << "Connected to [" << host << ":27017]" << std::endl;
|
|
|
|
Poco::MongoDB::Database db("config");
|
|
Poco::MongoDB::Document::Ptr doc = db.queryServerHello(*_mongo);
|
|
_wireVersion = doc->getInteger("maxWireVersion");
|
|
std::cout << "MongoDB wire version: " << _wireVersion << std::endl;
|
|
}
|
|
catch (Poco::Net::ConnectionRefusedException& e)
|
|
{
|
|
std::cout << "Couldn't connect to " << e.message() << ". " << std::endl;
|
|
return 0;
|
|
}
|
|
CppUnit::TestSuite* pSuite = new CppUnit::TestSuite("MongoDBTest");
|
|
|
|
CppUnit_addTest(pSuite, MongoDBTest, testObjectID);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testArray);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testConnectURI);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testHello);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testBuildInfo);
|
|
|
|
if (_wireVersion < Poco::MongoDB::Database::VER_51)
|
|
{
|
|
// Database supports old wire protocol
|
|
CppUnit_addTest(pSuite, MongoDBTest, testInsertRequest);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testQueryRequest);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testDBQueryRequest);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testCountCommand);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testDBCountCommand);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testDBCount2Command);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testConnectionPool);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testDeleteRequest);
|
|
|
|
CppUnit_addTest(pSuite, MongoDBTest, testCursorRequest);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testCommand);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testUUID);
|
|
}
|
|
|
|
if (_wireVersion >= Poco::MongoDB::Database::VER_36)
|
|
{
|
|
// Database supports OP_MSG wire protocol
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdWriteRead);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdHello);
|
|
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdInsert);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdFind);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdCount);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdConnectionPool);
|
|
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdDelete);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdUnaknowledgedInsert);
|
|
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdCursor);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdCursorAggregate);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdKillCursor);
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdCursorEmptyFirstBatch);
|
|
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdUUID);
|
|
|
|
CppUnit_addTest(pSuite, MongoDBTest, testOpCmdDropDatabase);
|
|
}
|
|
|
|
return pSuite;
|
|
}
|