Page Speed Optimization Libraries
1.3.25.1
|
00001 /* 00002 * Copyright 2011 Google Inc. 00003 * 00004 * Licensed under the Apache License, Version 2.0 (the "License"); 00005 * you may not use this file except in compliance with the License. 00006 * You may obtain a copy of the License at 00007 * 00008 * http:///www.apache.org/licenses/LICENSE-2.0 00009 * 00010 * Unless required by applicable law or agreed to in writing, software 00011 * distributed under the License is distributed on an "AS IS" BASIS, 00012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 00013 * See the License for the specific language governing permissions and 00014 * limitations under the License. 00015 */ 00016 00018 00020 00021 #ifndef NET_INSTAWEB_HTTP_PUBLIC_MOCK_CALLBACK_H_ 00022 #define NET_INSTAWEB_HTTP_PUBLIC_MOCK_CALLBACK_H_ 00023 00024 #include "net/instaweb/http/public/async_fetch.h" 00025 #include "net/instaweb/http/public/request_context.h" 00026 #include "net/instaweb/util/public/basictypes.h" 00027 #include "net/instaweb/util/public/gtest.h" 00028 00029 namespace net_instaweb { 00030 00034 class ExpectStringAsyncFetch : public StringAsyncFetch { 00035 public: 00036 explicit ExpectStringAsyncFetch(bool expect_success) 00037 : expect_success_(expect_success) {} 00038 ExpectStringAsyncFetch(bool expect_success, 00039 const RequestContextPtr& request_context) 00040 : StringAsyncFetch(request_context), expect_success_(expect_success) {} 00041 virtual ~ExpectStringAsyncFetch() { 00042 EXPECT_TRUE(done()); 00043 } 00044 00045 virtual void HandleDone(bool success) { 00046 EXPECT_FALSE(done()) << "Already Done; perhaps you reused without Reset()"; 00047 StringAsyncFetch::HandleDone(success); 00048 EXPECT_EQ(expect_success_, success); 00049 } 00050 00051 void set_expect_success(bool x) { expect_success_ = x; } 00052 00053 private: 00054 bool expect_success_; 00055 00056 DISALLOW_COPY_AND_ASSIGN(ExpectStringAsyncFetch); 00057 }; 00058 00059 } 00060 00061 #endif ///< NET_INSTAWEB_HTTP_PUBLIC_MOCK_CALLBACK_H_