Page Speed Optimization Libraries
1.4.26.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 ExpectStringAsyncFetch(bool expect_success, 00037 const RequestContextPtr& request_context) 00038 : StringAsyncFetch(request_context), expect_success_(expect_success) {} 00039 virtual ~ExpectStringAsyncFetch() { 00040 EXPECT_TRUE(done()); 00041 } 00042 00043 virtual void HandleDone(bool success) { 00044 EXPECT_FALSE(done()) << "Already Done; perhaps you reused without Reset()"; 00045 StringAsyncFetch::HandleDone(success); 00046 EXPECT_EQ(expect_success_, success); 00047 } 00048 00049 void set_expect_success(bool x) { expect_success_ = x; } 00050 00051 private: 00052 bool expect_success_; 00053 00054 DISALLOW_COPY_AND_ASSIGN(ExpectStringAsyncFetch); 00055 }; 00056 00057 } 00058 00059 #endif ///< NET_INSTAWEB_HTTP_PUBLIC_MOCK_CALLBACK_H_