btree_test_help.h

Go to the documentation of this file.

#pragma once

#include <string>

#include "mongo/db/json.h"
#include "mongo/db/storage/mmap_v1/btree/btree_logic.h"
#include "mongo/db/storage/mmap_v1/heap_record_store_btree.h"
#include "mongo/db/storage/mmap_v1/record_store_v1_test_help.h"

namespace mongo {

std::string bigNumString(long long n, int len);

BSONObj simpleKey(char c, int n = 1);

class TestHeadManager : public HeadManager {
public:
    virtual const RecordId getHead(OperationContext* opCtx) const {
        return _head;
    }

    virtual void setHead(OperationContext* opCtx, const RecordId newHead) {
        _head = newHead;
    }

private:
    RecordId _head;
};


template <class OnDiskFormat>
struct BtreeLogicTestHelper {
    BtreeLogicTestHelper(const BSONObj& order);

    // Everything needed for a fully-functional Btree logic
    TestHeadManager headManager;
    HeapRecordStoreBtree recordStore;
    SavedCursorRegistry cursorRegistry;
    BtreeLogic<OnDiskFormat> btree;
    DiskLoc dummyDiskLoc;
};


template <class OnDiskFormat>
class ArtificialTreeBuilder {
public:
    typedef typename BtreeLogic<OnDiskFormat>::BucketType BucketType;
    typedef typename BtreeLogic<OnDiskFormat>::KeyDataOwnedType KeyDataOwnedType;
    typedef typename BtreeLogic<OnDiskFormat>::KeyHeaderType KeyHeaderType;

    typedef typename OnDiskFormat::FixedWidthKeyType FixedWidthKeyType;

    ArtificialTreeBuilder(OperationContext* opCtx, BtreeLogicTestHelper<OnDiskFormat>* helper)
        : _opCtx(opCtx), _helper(helper) {}

    void makeTree(const std::string& spec);

    void checkStructure(const std::string& spec) const;

    void push(const DiskLoc bucketLoc, const BSONObj& key, const DiskLoc child);

    int fillBucketToExactSize(const DiskLoc bucketLoc, int targetSize, char startKey);

private:
    DiskLoc makeTree(const BSONObj& spec);

    void checkStructure(const BSONObj& spec, const DiskLoc node) const;

    bool isPresent(const BSONObj& key, int direction) const;

    static std::string expectedKey(const char* spec);

    OperationContext* _opCtx;
    BtreeLogicTestHelper<OnDiskFormat>* _helper;
};

}  // namespace mongo