1 files changed, 91 insertions, 0 deletions
diff --git a/dbtests/mockdbclient.h b/dbtests/mockdbclient.h
new file mode 100644
index 0000000..26f6250
--- /dev/null
+++ b/dbtests/mockdbclient.h
@@ -0,0 +1,91 @@
+// mockdbclient.h - mocked out client for testing.
+
+/**
+ *    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/>.
+ */
+
+#pragma once
+
+#include "../client/dbclient.h"
+#include "../db/commands.h"
+
+class MockDBClientConnection : public DBClientConnection {
+public:
+    MockDBClientConnection() : connect_() {}
+    virtual
+    BSONObj findOne(const string &ns, Query query, const BSONObj *fieldsToReturn = 0, int queryOptions = 0) {
+        return one_;
+    }
+    virtual
+    bool connect(const string &serverHostname, string& errmsg) {
+        return connect_;
+    }
+    virtual
+    bool isMaster(bool& isMaster, BSONObj *info=0) {
+        return isMaster_;
+    }
+    void one( const BSONObj &one ) {
+        one_ = one;
+    }
+    void connect( bool val ) {
+        connect_ = val;
+    }
+    void setIsMaster( bool val ) {
+        isMaster_ = val;
+    }
+private:
+    BSONObj one_;
+    bool connect_;
+    bool isMaster_;
+};
+
+class DirectDBClientConnection : public DBClientConnection {
+public:
+    struct ConnectionCallback {
+        virtual ~ConnectionCallback() {}
+        virtual void beforeCommand() {}
+        virtual void afterCommand() {}
+    };
+    DirectDBClientConnection( ReplPair *rp, ConnectionCallback *cc = 0 ) :
+            rp_( rp ),
+            cc_( cc ) {
+    }
+    virtual BSONObj findOne(const string &ns, Query query, const BSONObj *fieldsToReturn = 0, int queryOptions = 0) {
+        if ( cc_ ) cc_->beforeCommand();
+        SetGlobalReplPair s( rp_ );
+        BSONObjBuilder result;
+        result.append( "ok", Command::runAgainstRegistered( "admin.$cmd", query.obj, result ) ? 1.0 : 0.0 );
+        if ( cc_ ) cc_->afterCommand();
+        return result.obj();
+    }
+    virtual bool connect( const string &serverHostname, string& errmsg ) {
+        return true;
+    }
+private:
+    ReplPair *rp_;
+    ConnectionCallback *cc_;
+    class SetGlobalReplPair {
+    public:
+        SetGlobalReplPair( ReplPair *rp ) {
+            backup_ = replPair;
+            replPair = rp;
+        }
+        ~SetGlobalReplPair() {
+            replPair = backup_;
+        }
+    private:
+        ReplPair *backup_;
+    };
+};