1 files changed, 196 insertions, 0 deletions

diff --git a/linden/indra/test/mock_http_client.h b/linden/indra/test/mock_http_client.h
new file mode 100644
index 0000000..c3f1ade
--- /dev/null
+++ b/linden/indra/test/mock_http_client.h
@@ -0,0 +1,196 @@
+/** 
+ * @file mock_http_client.cpp
+ * @brief Framework for testing HTTP requests
+ *
+ * $LicenseInfo:firstyear=2007&license=viewergpl$
+ * 
+ * Copyright (c) 2007-2008, Linden Research, Inc.
+ * 
+ * Second Life Viewer Source Code
+ * The source code in this file ("Source Code") is provided by Linden Lab
+ * to you under the terms of the GNU General Public License, version 2.0
+ * ("GPL"), unless you have obtained a separate licensing agreement
+ * ("Other License"), formally executed by you and Linden Lab.  Terms of
+ * the GPL can be found in doc/GPL-license.txt in this distribution, or
+ * online at http://secondlifegrid.net/programs/open_source/licensing/gplv2
+ * 
+ * There are special exceptions to the terms and conditions of the GPL as
+ * it is applied to this Source Code. View the full text of the exception
+ * in the file doc/FLOSS-exception.txt in this software distribution, or
+ * online at http://secondlifegrid.net/programs/open_source/licensing/flossexception
+ * 
+ * By copying, modifying or distributing this software, you acknowledge
+ * that you have read and understood your obligations described above,
+ * and agree to abide by those obligations.
+ * 
+ * ALL LINDEN LAB SOURCE CODE IS PROVIDED "AS IS." LINDEN LAB MAKES NO
+ * WARRANTIES, EXPRESS, IMPLIED OR OTHERWISE, REGARDING ITS ACCURACY,
+ * COMPLETENESS OR PERFORMANCE.
+ * $/LicenseInfo$
+ */
+
+#include "llsdhttpserver.h"
+#include "lliohttpserver.h"
+#include "llhttpclient.h"
+#include "llformat.h"
+#include "llpipeutil.h"
+#include "llpumpio.h"
+
+namespace tut
+{
+        struct MockHttpClient
+        {
+        public:
+                MockHttpClient()
+                {
+                        apr_pool_create(&mPool, NULL);
+                        mServerPump = new LLPumpIO(mPool);
+                        mClientPump = new LLPumpIO(mPool);
+                        
+                        LLHTTPClient::setPump(*mClientPump);
+                }
+                
+                ~MockHttpClient()
+                {
+                        delete mServerPump;
+                        delete mClientPump;
+                        apr_pool_destroy(mPool);
+                }
+
+                void setupTheServer()
+                {
+                        LLHTTPNode& root = LLIOHTTPServer::create(mPool, *mServerPump, 8888);
+
+                        LLHTTPStandardServices::useServices();
+                        LLHTTPRegistrar::buildAllServices(root);
+                }
+                
+                void runThePump(float timeout = 100.0f)
+                {
+                        LLTimer timer;
+                        timer.setTimerExpirySec(timeout);
+
+                        while(!mSawCompleted && !timer.hasExpired())
+                        {
+                                if (mServerPump)
+                                {
+                                        mServerPump->pump();
+                                        mServerPump->callback();
+                                }
+                                if (mClientPump)
+                                {
+                                        mClientPump->pump();
+                                        mClientPump->callback();
+                                }
+                        }
+                }
+
+                void killServer()
+                {
+                        delete mServerPump;
+                        mServerPump = NULL;
+                }
+        
+        private:
+                apr_pool_t* mPool;
+                LLPumpIO* mServerPump;
+                LLPumpIO* mClientPump;
+
+                
+        protected:
+                void ensureStatusOK()
+                {
+                        if (mSawError)
+                        {
+                                std::string msg =
+                                        llformat("error() called when not expected, status %d",
+                                                mStatus); 
+                                fail(msg);
+                        }
+                }
+        
+                void ensureStatusError()
+                {
+                        if (!mSawError)
+                        {
+                                fail("error() wasn't called");
+                        }
+                }
+                
+                LLSD getResult()
+                {
+                        return mResult;
+                }
+        
+        protected:
+                bool mSawError;
+                U32 mStatus;
+                std::string mReason;
+                bool mSawCompleted;
+                LLSD mResult;
+                bool mResultDeleted;
+
+                class Result : public LLHTTPClient::Responder
+                {
+                protected:
+                        Result(MockHttpClient& client)
+                                : mClient(client)
+                        {
+                        }
+                
+                public:
+                        static boost::intrusive_ptr<Result> build(MockHttpClient& client)
+                        {
+                                return boost::intrusive_ptr<Result>(new Result(client));
+                        }
+                        
+                        ~Result()
+                        {
+                                mClient.mResultDeleted = true;
+                        }
+                        
+                        virtual void error(U32 status, const std::string& reason)
+                        {
+                                mClient.mSawError = true;
+                                mClient.mStatus = status;
+                                mClient.mReason = reason;
+                        }
+
+                        virtual void result(const LLSD& content)
+                        {
+                                mClient.mResult = content;
+                        }
+
+                        virtual void completed(
+                                                        U32 status, const std::string& reason,
+                                                        const LLSD& content)
+                        {
+                                LLHTTPClient::Responder::completed(status, reason, content);
+                                
+                                mClient.mSawCompleted = true;
+                        }
+
+                private:
+                        MockHttpClient& mClient;
+                };
+
+                friend class Result;
+
+        protected:
+
+                void reset()
+                {
+                        mSawError = false;
+                        mStatus = 0;
+                        mSawCompleted = false;
+                        mResult.clear();
+                        mResultDeleted = false;
+                }
+
+                LLHTTPClient::ResponderPtr newResult()
+                {
+                        reset();
+                        return Result::build(*this);
+                }
+        };
+}