Implement support for fallback update check urls in the component updater

components/component_updater/test/url_request_post_interceptor.h

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef COMPONENTS_COMPONENT_UPDATER_TEST_URL_REQUEST_POST_INTERCEPTOR_H_
 #define COMPONENTS_COMPONENT_UPDATER_TEST_URL_REQUEST_POST_INTERCEPTOR_H_
 
 #include <map>
 #include <queue>
 #include <string>
@@ skipping 48 matching lines @@
     virtual bool Match(const std::string& actual) const = 0;
     virtual ~RequestMatcher() {}
   };
 
   // Returns the url that is intercepted.
   GURL GetUrl() const;
 
   // Sets an expection for the body of the POST request and optionally,
   // provides a canned response identified by a |file_path| to be returned when
   // the expectation is met. If no |file_path| is provided, then an empty
-  // response body is served. This class takes ownership of the
-  // |request_matcher| object. Returns |true| if the expectation was set.
+  // response body is served. If |response_code| is provided, then an empty
+  // response body with that response code is returned.
+  // Returns |true| if the expectation was set. This class takes ownership of
+  // the |request_matcher| object.
   bool ExpectRequest(class RequestMatcher* request_matcher);
+  bool ExpectRequest(class RequestMatcher* request_matcher, int response_code);
   bool ExpectRequest(class RequestMatcher* request_matcher,
                      const base::FilePath& filepath);
 
   // Returns how many requests have been intercepted and matched by
   // an expectation. One expectation can only be matched by one request.
   int GetHitCount() const;
 
   // Returns how many requests in total have been captured by the interceptor.
   int GetCount() const;
 
   // Returns all requests that have been intercepted, matched or not.
   std::vector<std::string> GetRequests() const;
 
   // Returns all requests as a string for debugging purposes.
   std::string GetRequestsAsString() const;
 
   // Resets the state of the interceptor so that new expectations can be set.
   void Reset();
 
   class Delegate;
 
  private:
   friend class URLRequestPostInterceptorFactory;
-  typedef std::pair<const RequestMatcher*, std::string> Expectation;
+
+  static const int kResponseCode200 = 200;
+
+  struct ExpectationResponse {
+    ExpectationResponse(int code, const std::string& body)
+        : response_code(code), response_body(body) {}
+    const int response_code;
+    const std::string response_body;
+  };
+  typedef std::pair<const RequestMatcher*, ExpectationResponse> Expectation;
 
   URLRequestPostInterceptor(
       const GURL& url,
       const scoped_refptr<base::SequencedTaskRunner>& io_task_runner);
   ~URLRequestPostInterceptor();
 
   void ClearExpectations();
+
   const GURL url_;
   scoped_refptr<base::SequencedTaskRunner> io_task_runner_;
 
   mutable base::Lock interceptor_lock_;
   mutable int hit_count_;
   mutable std::vector<std::string> requests_;
   mutable std::queue<Expectation> expectations_;
 
   DISALLOW_COPY_AND_ASSIGN(URLRequestPostInterceptor);
 };
@@ skipping 24 matching lines @@
   DISALLOW_COPY_AND_ASSIGN(URLRequestPostInterceptorFactory);
 };
 
 // Intercepts HTTP POST requests sent to "localhost2".
 class InterceptorFactory : public URLRequestPostInterceptorFactory {
  public:
   explicit InterceptorFactory(
       const scoped_refptr<base::SequencedTaskRunner>& io_task_runner);
   ~InterceptorFactory();
 
+  // Creates an interceptor for the url path defined by POST_INTERCEPT_PATH.
   URLRequestPostInterceptor* CreateInterceptor();
 
+  // Creates an interceptor for the given url path.
+  URLRequestPostInterceptor* CreateInterceptorForPath(const char* url_path);
+
  private:
   DISALLOW_COPY_AND_ASSIGN(InterceptorFactory);
 };
 
 class PartialMatch : public URLRequestPostInterceptor::RequestMatcher {
  public:
   explicit PartialMatch(const std::string& expected) : expected_(expected) {}
   virtual bool Match(const std::string& actual) const OVERRIDE;
 
  private:
   const std::string expected_;
 
   DISALLOW_COPY_AND_ASSIGN(PartialMatch);
 };
 
 }  // namespace component_updater
 
 #endif  // COMPONENTS_COMPONENT_UPDATER_TEST_URL_REQUEST_POST_INTERCEPTOR_H_