mirror of
https://github.com/mongodb/mongo.git
synced 2024-12-01 09:32:32 +01:00
404 lines
13 KiB
C++
404 lines
13 KiB
C++
// pdfiletests.cpp : pdfile unit tests.
|
|
//
|
|
|
|
/**
|
|
* Copyright (C) 2008 10gen 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 <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "pch.h"
|
|
#include "../db/pdfile.h"
|
|
|
|
#include "../db/db.h"
|
|
#include "../db/json.h"
|
|
|
|
#include "dbtests.h"
|
|
|
|
namespace PdfileTests {
|
|
|
|
namespace ScanCapped {
|
|
|
|
class Base {
|
|
public:
|
|
Base() : _context( ns() ) {
|
|
}
|
|
virtual ~Base() {
|
|
if ( !nsd() )
|
|
return;
|
|
string n( ns() );
|
|
dropNS( n );
|
|
}
|
|
void run() {
|
|
stringstream spec;
|
|
spec << "{\"capped\":true,\"size\":2000,\"$nExtents\":" << nExtents() << "}";
|
|
string err;
|
|
ASSERT( userCreateNS( ns(), fromjson( spec.str() ), err, false ) );
|
|
prepare();
|
|
int j = 0;
|
|
for ( boost::shared_ptr<Cursor> i = theDataFileMgr.findAll( ns() );
|
|
i->ok(); i->advance(), ++j )
|
|
ASSERT_EQUALS( j, i->current().firstElement().number() );
|
|
ASSERT_EQUALS( count(), j );
|
|
|
|
j = count() - 1;
|
|
for ( boost::shared_ptr<Cursor> i =
|
|
findTableScan( ns(), fromjson( "{\"$natural\":-1}" ) );
|
|
i->ok(); i->advance(), --j )
|
|
ASSERT_EQUALS( j, i->current().firstElement().number() );
|
|
ASSERT_EQUALS( -1, j );
|
|
}
|
|
protected:
|
|
virtual void prepare() = 0;
|
|
virtual int count() const = 0;
|
|
virtual int nExtents() const {
|
|
return 0;
|
|
}
|
|
// bypass standard alloc/insert routines to use the extent we want.
|
|
static DiskLoc insert( DiskLoc ext, int i ) {
|
|
BSONObjBuilder b;
|
|
b.append( "a", i );
|
|
BSONObj o = b.done();
|
|
int len = o.objsize();
|
|
Extent *e = ext.ext();
|
|
e = getDur().writing(e);
|
|
int ofs;
|
|
if ( e->lastRecord.isNull() )
|
|
ofs = ext.getOfs() + ( e->_extentData - (char *)e );
|
|
else
|
|
ofs = e->lastRecord.getOfs() + e->lastRecord.rec()->lengthWithHeaders;
|
|
DiskLoc dl( ext.a(), ofs );
|
|
Record *r = dl.rec();
|
|
r = (Record*) getDur().writingPtr(r, Record::HeaderSize + len);
|
|
r->lengthWithHeaders = Record::HeaderSize + len;
|
|
r->extentOfs = e->myLoc.getOfs();
|
|
r->nextOfs = DiskLoc::NullOfs;
|
|
r->prevOfs = e->lastRecord.isNull() ? DiskLoc::NullOfs : e->lastRecord.getOfs();
|
|
memcpy( r->data, o.objdata(), len );
|
|
if ( e->firstRecord.isNull() )
|
|
e->firstRecord = dl;
|
|
else
|
|
getDur().writingInt(e->lastRecord.rec()->nextOfs) = ofs;
|
|
e->lastRecord = dl;
|
|
return dl;
|
|
}
|
|
static const char *ns() {
|
|
return "unittests.ScanCapped";
|
|
}
|
|
static NamespaceDetails *nsd() {
|
|
return nsdetails( ns() );
|
|
}
|
|
private:
|
|
dblock lk_;
|
|
Client::Context _context;
|
|
};
|
|
|
|
class Empty : public Base {
|
|
virtual void prepare() {}
|
|
virtual int count() const {
|
|
return 0;
|
|
}
|
|
};
|
|
|
|
class EmptyLooped : public Base {
|
|
virtual void prepare() {
|
|
nsd()->writingWithExtra()->capFirstNewRecord = DiskLoc();
|
|
}
|
|
virtual int count() const {
|
|
return 0;
|
|
}
|
|
};
|
|
|
|
class EmptyMultiExtentLooped : public Base {
|
|
virtual void prepare() {
|
|
nsd()->writingWithExtra()->capFirstNewRecord = DiskLoc();
|
|
}
|
|
virtual int count() const {
|
|
return 0;
|
|
}
|
|
virtual int nExtents() const {
|
|
return 3;
|
|
}
|
|
};
|
|
|
|
class Single : public Base {
|
|
virtual void prepare() {
|
|
nsd()->writingWithExtra()->capFirstNewRecord = insert( nsd()->capExtent, 0 );
|
|
}
|
|
virtual int count() const {
|
|
return 1;
|
|
}
|
|
};
|
|
|
|
class NewCapFirst : public Base {
|
|
virtual void prepare() {
|
|
DiskLoc x = insert( nsd()->capExtent, 0 );
|
|
nsd()->writingWithExtra()->capFirstNewRecord = x;
|
|
insert( nsd()->capExtent, 1 );
|
|
}
|
|
virtual int count() const {
|
|
return 2;
|
|
}
|
|
};
|
|
|
|
class NewCapLast : public Base {
|
|
virtual void prepare() {
|
|
insert( nsd()->capExtent, 0 );
|
|
nsd()->capFirstNewRecord.writing() = insert( nsd()->capExtent, 1 );
|
|
}
|
|
virtual int count() const {
|
|
return 2;
|
|
}
|
|
};
|
|
|
|
class NewCapMiddle : public Base {
|
|
virtual void prepare() {
|
|
insert( nsd()->capExtent, 0 );
|
|
nsd()->capFirstNewRecord.writing() = insert( nsd()->capExtent, 1 );
|
|
insert( nsd()->capExtent, 2 );
|
|
}
|
|
virtual int count() const {
|
|
return 3;
|
|
}
|
|
};
|
|
|
|
class FirstExtent : public Base {
|
|
virtual void prepare() {
|
|
insert( nsd()->capExtent, 0 );
|
|
insert( nsd()->lastExtent, 1 );
|
|
nsd()->capFirstNewRecord.writing() = insert( nsd()->capExtent, 2 );
|
|
insert( nsd()->capExtent, 3 );
|
|
}
|
|
virtual int count() const {
|
|
return 4;
|
|
}
|
|
virtual int nExtents() const {
|
|
return 2;
|
|
}
|
|
};
|
|
|
|
class LastExtent : public Base {
|
|
virtual void prepare() {
|
|
nsd()->capExtent.writing() = nsd()->lastExtent;
|
|
insert( nsd()->capExtent, 0 );
|
|
insert( nsd()->firstExtent, 1 );
|
|
nsd()->capFirstNewRecord.writing() = insert( nsd()->capExtent, 2 );
|
|
insert( nsd()->capExtent, 3 );
|
|
}
|
|
virtual int count() const {
|
|
return 4;
|
|
}
|
|
virtual int nExtents() const {
|
|
return 2;
|
|
}
|
|
};
|
|
|
|
class MidExtent : public Base {
|
|
virtual void prepare() {
|
|
nsd()->capExtent.writing() = nsd()->firstExtent.ext()->xnext;
|
|
insert( nsd()->capExtent, 0 );
|
|
insert( nsd()->lastExtent, 1 );
|
|
insert( nsd()->firstExtent, 2 );
|
|
nsd()->capFirstNewRecord.writing() = insert( nsd()->capExtent, 3 );
|
|
insert( nsd()->capExtent, 4 );
|
|
}
|
|
virtual int count() const {
|
|
return 5;
|
|
}
|
|
virtual int nExtents() const {
|
|
return 3;
|
|
}
|
|
};
|
|
|
|
class AloneInExtent : public Base {
|
|
virtual void prepare() {
|
|
nsd()->capExtent.writing() = nsd()->firstExtent.ext()->xnext;
|
|
insert( nsd()->lastExtent, 0 );
|
|
insert( nsd()->firstExtent, 1 );
|
|
nsd()->capFirstNewRecord.writing() = insert( nsd()->capExtent, 2 );
|
|
}
|
|
virtual int count() const {
|
|
return 3;
|
|
}
|
|
virtual int nExtents() const {
|
|
return 3;
|
|
}
|
|
};
|
|
|
|
class FirstInExtent : public Base {
|
|
virtual void prepare() {
|
|
nsd()->capExtent.writing() = nsd()->firstExtent.ext()->xnext;
|
|
insert( nsd()->lastExtent, 0 );
|
|
insert( nsd()->firstExtent, 1 );
|
|
nsd()->capFirstNewRecord.writing() = insert( nsd()->capExtent, 2 );
|
|
insert( nsd()->capExtent, 3 );
|
|
}
|
|
virtual int count() const {
|
|
return 4;
|
|
}
|
|
virtual int nExtents() const {
|
|
return 3;
|
|
}
|
|
};
|
|
|
|
class LastInExtent : public Base {
|
|
virtual void prepare() {
|
|
nsd()->capExtent.writing() = nsd()->firstExtent.ext()->xnext;
|
|
insert( nsd()->capExtent, 0 );
|
|
insert( nsd()->lastExtent, 1 );
|
|
insert( nsd()->firstExtent, 2 );
|
|
nsd()->capFirstNewRecord.writing() = insert( nsd()->capExtent, 3 );
|
|
}
|
|
virtual int count() const {
|
|
return 4;
|
|
}
|
|
virtual int nExtents() const {
|
|
return 3;
|
|
}
|
|
};
|
|
|
|
} // namespace ScanCapped
|
|
|
|
namespace Insert {
|
|
class Base {
|
|
public:
|
|
Base() : _context( ns() ) {
|
|
}
|
|
virtual ~Base() {
|
|
if ( !nsd() )
|
|
return;
|
|
string n( ns() );
|
|
dropNS( n );
|
|
}
|
|
protected:
|
|
static const char *ns() {
|
|
return "unittests.pdfiletests.Insert";
|
|
}
|
|
static NamespaceDetails *nsd() {
|
|
return nsdetails( ns() );
|
|
}
|
|
private:
|
|
dblock lk_;
|
|
Client::Context _context;
|
|
};
|
|
|
|
class UpdateDate : public Base {
|
|
public:
|
|
void run() {
|
|
BSONObjBuilder b;
|
|
b.appendTimestamp( "a" );
|
|
BSONObj o = b.done();
|
|
ASSERT( 0 == o.getField( "a" ).date() );
|
|
theDataFileMgr.insertWithObjMod( ns(), o );
|
|
ASSERT( 0 != o.getField( "a" ).date() );
|
|
}
|
|
};
|
|
} // namespace Insert
|
|
|
|
class ExtentSizing {
|
|
public:
|
|
struct SmallFilesControl {
|
|
SmallFilesControl() {
|
|
old = cmdLine.smallfiles;
|
|
cmdLine.smallfiles = false;
|
|
}
|
|
~SmallFilesControl() {
|
|
cmdLine.smallfiles = old;
|
|
}
|
|
bool old;
|
|
};
|
|
void run() {
|
|
SmallFilesControl c;
|
|
// test that no matter what we start with, we always get to max extent size
|
|
for ( int obj=16; obj<BSONObjMaxUserSize; obj += 111 ) {
|
|
int sz = Extent::initialSize( obj );
|
|
for ( int i=0; i<100; i++ ) {
|
|
sz = Extent::followupSize( obj , sz );
|
|
}
|
|
ASSERT_EQUALS( Extent::maxSize() , sz );
|
|
}
|
|
}
|
|
};
|
|
|
|
class ExtentAllocOrder {
|
|
public:
|
|
void run() {
|
|
string dbname = "unittest_ex";
|
|
|
|
string c1 = dbname + ".x1";
|
|
string c2 = dbname + ".x2";
|
|
|
|
{
|
|
DBDirectClient db;
|
|
db.dropDatabase( dbname );
|
|
}
|
|
|
|
dblock mylock;
|
|
Client::Context cx( dbname );
|
|
|
|
bool isnew;
|
|
Database * d = dbHolder.getOrCreate( dbname , dbpath , isnew );
|
|
assert( d );
|
|
|
|
int big = 10 * 1024;
|
|
//int small = 1024;
|
|
|
|
unsigned long long l = 0;
|
|
while ( 1 ) {
|
|
MongoDataFile * f = d->addAFile( big , false );
|
|
cout << f->length() << endl;
|
|
if ( f->length() == l )
|
|
break;
|
|
l = f->length();
|
|
}
|
|
|
|
int start = d->numFiles();
|
|
for ( int i=0; i<start; i++ )
|
|
d->allocExtent( c1.c_str() , d->getFile( i )->getHeader()->unusedLength , false );
|
|
ASSERT_EQUALS( start , d->numFiles() );
|
|
|
|
{
|
|
DBDirectClient db;
|
|
db.dropDatabase( dbname );
|
|
}
|
|
}
|
|
};
|
|
|
|
|
|
class All : public Suite {
|
|
public:
|
|
All() : Suite( "pdfile" ) {}
|
|
|
|
void setupTests() {
|
|
add< ScanCapped::Empty >();
|
|
add< ScanCapped::EmptyLooped >();
|
|
add< ScanCapped::EmptyMultiExtentLooped >();
|
|
add< ScanCapped::Single >();
|
|
add< ScanCapped::NewCapFirst >();
|
|
add< ScanCapped::NewCapLast >();
|
|
add< ScanCapped::NewCapMiddle >();
|
|
add< ScanCapped::FirstExtent >();
|
|
add< ScanCapped::LastExtent >();
|
|
add< ScanCapped::MidExtent >();
|
|
add< ScanCapped::AloneInExtent >();
|
|
add< ScanCapped::FirstInExtent >();
|
|
add< ScanCapped::LastInExtent >();
|
|
add< Insert::UpdateDate >();
|
|
add< ExtentSizing >();
|
|
add< ExtentAllocOrder >();
|
|
}
|
|
} myall;
|
|
|
|
} // namespace PdfileTests
|
|
|