record_store_test_updaterecord.h

Go to the documentation of this file.

// record_store_test_updaterecord.h

#pragma once

#include "mongo/db/record_id.h"
#include "mongo/db/storage/record_data.h"
#include "mongo/db/storage/record_store.h"
#include "mongo/unittest/unittest.h"

namespace mongo {
namespace {

class UpdateNotifierSpy : public UpdateNotifier {
public:
    UpdateNotifierSpy(OperationContext* opCtx, const RecordId& loc, const char* buf, size_t size)
        : _opCtx(opCtx), _loc(loc), _data(buf, size), nInPlaceCalls(0) {}

    ~UpdateNotifierSpy() {}

    Status recordStoreGoingToUpdateInPlace(OperationContext* opCtx, const RecordId& loc) {
        nInPlaceCalls++;
        ASSERT_EQUALS(_opCtx, opCtx);
        ASSERT_EQUALS(_loc, loc);
        return Status::OK();
    }

    int numInPlaceCallbacks() const {
        return nInPlaceCalls;
    }

private:
    OperationContext* _opCtx;
    RecordId _loc;
    std::string _data;

    // To verify the number of callbacks to the notifier.
    int nInPlaceCalls;
};

}  // namespace
}  // namespace mongo