// wiredtiger_util_test.cpp
/**
* Copyright (C) 2014 MongoDB Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License, version 3,
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see .
*
* As a special exception, the copyright holders give permission to link the
* code of portions of this program with the OpenSSL library under certain
* conditions as described in each individual source file and distribute
* linked combinations including the program with the OpenSSL library. You
* must comply with the GNU Affero General Public License in all respects for
* all of the code used other than as permitted herein. If you modify file(s)
* with this exception, you may extend this exception to your version of the
* file(s), but you are not obligated to do so. If you do not wish to do so,
* delete this exception statement from your version. If you delete this
* exception statement from all source files in the program, then also delete
* it in the license file.
*/
#include "mongo/platform/basic.h"
#include
#include
#include "mongo/base/string_data.h"
#include "mongo/db/operation_context_noop.h"
#include "mongo/db/storage/wiredtiger/wiredtiger_recovery_unit.h"
#include "mongo/db/storage/wiredtiger/wiredtiger_session_cache.h"
#include "mongo/db/storage/wiredtiger/wiredtiger_util.h"
#include "mongo/unittest/temp_dir.h"
#include "mongo/unittest/unittest.h"
namespace mongo {
using std::string;
using std::stringstream;
class WiredTigerConnection {
public:
WiredTigerConnection(StringData dbpath, StringData extraStrings) : _conn(NULL) {
std::stringstream ss;
ss << "create,";
ss << extraStrings;
string config = ss.str();
int ret = wiredtiger_open(dbpath.toString().c_str(), NULL, config.c_str(), &_conn);
ASSERT_OK(wtRCToStatus(ret));
ASSERT(_conn);
}
~WiredTigerConnection() {
_conn->close(_conn, NULL);
}
WT_CONNECTION* getConnection() const {
return _conn;
}
private:
WT_CONNECTION* _conn;
};
class WiredTigerUtilHarnessHelper {
public:
WiredTigerUtilHarnessHelper(StringData extraStrings)
: _dbpath("wt_test"),
_connection(_dbpath.path(), extraStrings),
_sessionCache(_connection.getConnection()) {}
WiredTigerSessionCache* getSessionCache() {
return &_sessionCache;
}
OperationContext* newOperationContext() {
return new OperationContextNoop(new WiredTigerRecoveryUnit(getSessionCache()));
}
private:
unittest::TempDir _dbpath;
WiredTigerConnection _connection;
WiredTigerSessionCache _sessionCache;
};
class WiredTigerUtilMetadataTest : public mongo::unittest::Test {
public:
virtual void setUp() {
_harnessHelper.reset(new WiredTigerUtilHarnessHelper(""));
_opCtx.reset(_harnessHelper->newOperationContext());
}
virtual void tearDown() {
_opCtx.reset(NULL);
_harnessHelper.reset(NULL);
}
protected:
const char* getURI() const {
return "table:mytable";
}
OperationContext* getOperationContext() const {
ASSERT(_opCtx.get());
return _opCtx.get();
}
void createSession(const char* config) {
WT_SESSION* wtSession =
WiredTigerRecoveryUnit::get(_opCtx.get())->getSession(_opCtx.get())->getSession();
ASSERT_OK(wtRCToStatus(wtSession->create(wtSession, getURI(), config)));
}
private:
std::unique_ptr _harnessHelper;
std::unique_ptr _opCtx;
};
TEST_F(WiredTigerUtilMetadataTest, GetConfigurationStringInvalidURI) {
StatusWith result = WiredTigerUtil::getMetadata(getOperationContext(), getURI());
ASSERT_NOT_OK(result.getStatus());
ASSERT_EQUALS(ErrorCodes::NoSuchKey, result.getStatus().code());
}
TEST_F(WiredTigerUtilMetadataTest, GetConfigurationStringNull) {
const char* config = NULL;
createSession(config);
StatusWith result = WiredTigerUtil::getMetadata(getOperationContext(), getURI());
ASSERT_OK(result.getStatus());
ASSERT_FALSE(result.getValue().empty());
}
TEST_F(WiredTigerUtilMetadataTest, GetConfigurationStringSimple) {
const char* config = "app_metadata=(abc=123)";
createSession(config);
StatusWith result = WiredTigerUtil::getMetadata(getOperationContext(), getURI());
ASSERT_OK(result.getStatus());
ASSERT_STRING_CONTAINS(result.getValue(), config);
}
TEST_F(WiredTigerUtilMetadataTest, GetApplicationMetadataInvalidURI) {
StatusWith result =
WiredTigerUtil::getApplicationMetadata(getOperationContext(), getURI());
ASSERT_NOT_OK(result.getStatus());
ASSERT_EQUALS(ErrorCodes::NoSuchKey, result.getStatus().code());
}
TEST_F(WiredTigerUtilMetadataTest, GetApplicationMetadataNull) {
const char* config = NULL;
createSession(config);
StatusWith result =
WiredTigerUtil::getApplicationMetadata(getOperationContext(), getURI());
ASSERT_OK(result.getStatus());
ASSERT_TRUE(result.getValue().isEmpty());
}
TEST_F(WiredTigerUtilMetadataTest, GetApplicationMetadataString) {
const char* config = "app_metadata=\"abc\"";
createSession(config);
StatusWith result =
WiredTigerUtil::getApplicationMetadata(getOperationContext(), getURI());
ASSERT_NOT_OK(result.getStatus());
ASSERT_EQUALS(ErrorCodes::FailedToParse, result.getStatus().code());
}
TEST_F(WiredTigerUtilMetadataTest, GetApplicationMetadataDuplicateKeys) {
const char* config = "app_metadata=(abc=123,abc=456)";
createSession(config);
StatusWith result =
WiredTigerUtil::getApplicationMetadata(getOperationContext(), getURI());
ASSERT_NOT_OK(result.getStatus());
ASSERT_EQUALS(ErrorCodes::DuplicateKey, result.getStatus().code());
}
TEST_F(WiredTigerUtilMetadataTest, GetApplicationMetadataTypes) {
const char* config =
"app_metadata=(stringkey=\"abc\",boolkey1=true,boolkey2=false,"
"idkey=def,numkey=123,"
"structkey=(k1=v2,k2=v2))";
createSession(config);
StatusWith result =
WiredTigerUtil::getApplicationMetadata(getOperationContext(), getURI());
ASSERT_OK(result.getStatus());
const BSONObj& obj = result.getValue();
BSONElement stringElement = obj.getField("stringkey");
ASSERT_EQUALS(mongo::String, stringElement.type());
ASSERT_EQUALS("abc", stringElement.String());
BSONElement boolElement1 = obj.getField("boolkey1");
ASSERT_TRUE(boolElement1.isBoolean());
ASSERT_TRUE(boolElement1.boolean());
BSONElement boolElement2 = obj.getField("boolkey2");
ASSERT_TRUE(boolElement2.isBoolean());
ASSERT_FALSE(boolElement2.boolean());
BSONElement identifierElement = obj.getField("idkey");
ASSERT_EQUALS(mongo::String, identifierElement.type());
ASSERT_EQUALS("def", identifierElement.String());
BSONElement numberElement = obj.getField("numkey");
ASSERT_TRUE(numberElement.isNumber());
ASSERT_EQUALS(123, numberElement.numberInt());
BSONElement structElement = obj.getField("structkey");
ASSERT_EQUALS(mongo::String, structElement.type());
ASSERT_EQUALS("(k1=v2,k2=v2)", structElement.String());
}
TEST_F(WiredTigerUtilMetadataTest, CheckApplicationMetadataFormatVersionMissingKey) {
createSession("app_metadata=(abc=123)");
ASSERT_OK(WiredTigerUtil::checkApplicationMetadataFormatVersion(
getOperationContext(), getURI(), 1, 1));
ASSERT_NOT_OK(WiredTigerUtil::checkApplicationMetadataFormatVersion(
getOperationContext(), getURI(), 2, 2));
}
TEST_F(WiredTigerUtilMetadataTest, CheckApplicationMetadataFormatVersionString) {
createSession("app_metadata=(formatVersion=\"bar\")");
ASSERT_NOT_OK(WiredTigerUtil::checkApplicationMetadataFormatVersion(
getOperationContext(), getURI(), 1, 1));
}
TEST_F(WiredTigerUtilMetadataTest, CheckApplicationMetadataFormatVersionNumber) {
createSession("app_metadata=(formatVersion=2)");
ASSERT_EQUALS(
WiredTigerUtil::checkApplicationMetadataFormatVersion(getOperationContext(), getURI(), 2, 3)
.getValue(),
2);
ASSERT_NOT_OK(WiredTigerUtil::checkApplicationMetadataFormatVersion(
getOperationContext(), getURI(), 1, 1));
ASSERT_NOT_OK(WiredTigerUtil::checkApplicationMetadataFormatVersion(
getOperationContext(), getURI(), 3, 3));
}
TEST_F(WiredTigerUtilMetadataTest, CheckApplicationMetadataFormatInvalidURI) {
createSession("\"");
Status result =
WiredTigerUtil::checkApplicationMetadataFormatVersion(getOperationContext(), getURI(), 0, 3)
.getStatus();
ASSERT_NOT_OK(result);
ASSERT_EQUALS(ErrorCodes::FailedToParse, result.code());
}
TEST(WiredTigerUtilTest, GetStatisticsValueMissingTable) {
WiredTigerUtilHarnessHelper harnessHelper("statistics=(all)");
WiredTigerRecoveryUnit recoveryUnit(harnessHelper.getSessionCache());
WiredTigerSession* session = recoveryUnit.getSession(NULL);
StatusWith result =
WiredTigerUtil::getStatisticsValue(session->getSession(),
"statistics:table:no_such_table",
"statistics=(fast)",
WT_STAT_DSRC_BLOCK_SIZE);
ASSERT_NOT_OK(result.getStatus());
ASSERT_EQUALS(ErrorCodes::CursorNotFound, result.getStatus().code());
}
TEST(WiredTigerUtilTest, GetStatisticsValueStatisticsDisabled) {
WiredTigerUtilHarnessHelper harnessHelper("statistics=(none)");
WiredTigerRecoveryUnit recoveryUnit(harnessHelper.getSessionCache());
WiredTigerSession* session = recoveryUnit.getSession(NULL);
WT_SESSION* wtSession = session->getSession();
ASSERT_OK(wtRCToStatus(wtSession->create(wtSession, "table:mytable", NULL)));
StatusWith result = WiredTigerUtil::getStatisticsValue(session->getSession(),
"statistics:table:mytable",
"statistics=(fast)",
WT_STAT_DSRC_BLOCK_SIZE);
ASSERT_NOT_OK(result.getStatus());
ASSERT_EQUALS(ErrorCodes::CursorNotFound, result.getStatus().code());
}
TEST(WiredTigerUtilTest, GetStatisticsValueInvalidKey) {
WiredTigerUtilHarnessHelper harnessHelper("statistics=(all)");
WiredTigerRecoveryUnit recoveryUnit(harnessHelper.getSessionCache());
WiredTigerSession* session = recoveryUnit.getSession(NULL);
WT_SESSION* wtSession = session->getSession();
ASSERT_OK(wtRCToStatus(wtSession->create(wtSession, "table:mytable", NULL)));
// Use connection statistics key which does not apply to a table.
StatusWith result = WiredTigerUtil::getStatisticsValue(session->getSession(),
"statistics:table:mytable",
"statistics=(fast)",
WT_STAT_CONN_SESSION_OPEN);
ASSERT_NOT_OK(result.getStatus());
ASSERT_EQUALS(ErrorCodes::NoSuchKey, result.getStatus().code());
}
TEST(WiredTigerUtilTest, GetStatisticsValueValidKey) {
WiredTigerUtilHarnessHelper harnessHelper("statistics=(all)");
WiredTigerRecoveryUnit recoveryUnit(harnessHelper.getSessionCache());
WiredTigerSession* session = recoveryUnit.getSession(NULL);
WT_SESSION* wtSession = session->getSession();
ASSERT_OK(wtRCToStatus(wtSession->create(wtSession, "table:mytable", NULL)));
// Use connection statistics key which does not apply to a table.
StatusWith result = WiredTigerUtil::getStatisticsValue(session->getSession(),
"statistics:table:mytable",
"statistics=(fast)",
WT_STAT_DSRC_LSM_CHUNK_COUNT);
ASSERT_OK(result.getStatus());
// Expect statistics value to be zero for a LSM key on a Btree.
ASSERT_EQUALS(0U, result.getValue());
}
TEST(WiredTigerUtilTest, GetStatisticsValueAsUInt8) {
WiredTigerUtilHarnessHelper harnessHelper("statistics=(all)");
WiredTigerRecoveryUnit recoveryUnit(harnessHelper.getSessionCache());
WiredTigerSession* session = recoveryUnit.getSession(NULL);
WT_SESSION* wtSession = session->getSession();
ASSERT_OK(wtRCToStatus(wtSession->create(wtSession, "table:mytable", NULL)));
// Use data source statistics that has a value > 256 on an empty table.
StatusWith resultUInt64 =
WiredTigerUtil::getStatisticsValue(session->getSession(),
"statistics:table:mytable",
"statistics=(fast)",
WT_STAT_DSRC_ALLOCATION_SIZE);
ASSERT_OK(resultUInt64.getStatus());
ASSERT_GREATER_THAN(resultUInt64.getValue(),
static_cast(std::numeric_limits::max()));
// Ensure that statistics value retrieved as an 8-bit unsigned value
// is capped at maximum value for that type.
StatusWith resultUInt8 =
WiredTigerUtil::getStatisticsValueAs(session->getSession(),
"statistics:table:mytable",
"statistics=(fast)",
WT_STAT_DSRC_ALLOCATION_SIZE);
ASSERT_OK(resultUInt8.getStatus());
ASSERT_EQUALS(std::numeric_limits::max(), resultUInt8.getValue());
// Read statistics value as signed 16-bit value with alternative maximum value to
// std::numeric_limits.
StatusWith resultInt16 =
WiredTigerUtil::getStatisticsValueAs(session->getSession(),
"statistics:table:mytable",
"statistics=(fast)",
WT_STAT_DSRC_ALLOCATION_SIZE,
static_cast(100));
ASSERT_OK(resultInt16.getStatus());
ASSERT_EQUALS(static_cast(100), resultInt16.getValue());
}
} // namespace mongo