Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(721)

Unified Diff: components/update_client/test/url_request_post_interceptor.h

Issue 808773005: Move most of the component updater artifacts to update_client. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: components/update_client/test/url_request_post_interceptor.h
diff --git a/components/update_client/test/url_request_post_interceptor.h b/components/update_client/test/url_request_post_interceptor.h
new file mode 100644
index 0000000000000000000000000000000000000000..d261160f611db1b5f76437cdb9a3190714b99e34
--- /dev/null
+++ b/components/update_client/test/url_request_post_interceptor.h
@@ -0,0 +1,165 @@
+// 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_UPDATE_CLIENT_TEST_URL_REQUEST_POST_INTERCEPTOR_H_
+#define COMPONENTS_UPDATE_CLIENT_TEST_URL_REQUEST_POST_INTERCEPTOR_H_
+
+#include <stdint.h>
+#include <map>
+#include <queue>
+#include <string>
+#include <utility>
+#include <vector>
+
+#include "base/macros.h"
+#include "base/memory/ref_counted.h"
+#include "base/synchronization/lock.h"
+#include "url/gurl.h"
+
+namespace base {
+class FilePath;
+class SequencedTaskRunner;
+}
+
+namespace net {
+class URLRequest;
+}
+
+namespace update_client {
+
+// Intercepts requests to a file path, counts them, and captures the body of
+// the requests. Optionally, for each request, it can return a canned response
+// from a given file. The class maintains a queue of expectations, and returns
+// one and only one response for each request that matches and it is
+// intercepted.
+class URLRequestPostInterceptor {
+ public:
+ // Allows a generic string maching interface when setting up expectations.
+ class RequestMatcher {
+ public:
+ 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. 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;
+
+ 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);
+};
+
+class URLRequestPostInterceptorFactory {
+ public:
+ URLRequestPostInterceptorFactory(
+ const std::string& scheme,
+ const std::string& hostname,
+ const scoped_refptr<base::SequencedTaskRunner>& io_task_runner);
+ ~URLRequestPostInterceptorFactory();
+
+ // Creates an interceptor object for the specified url path. Returns NULL
+ // in case of errors or a valid interceptor object otherwise. The caller
+ // does not own the returned object.
+ URLRequestPostInterceptor* CreateInterceptor(const base::FilePath& filepath);
+
+ private:
+ const std::string scheme_;
+ const std::string hostname_;
+ scoped_refptr<base::SequencedTaskRunner> io_task_runner_;
+
+ // After creation, |delegate_| lives on the IO thread and it is owned by
+ // a URLRequestFilter after registration. A task to unregister it and
+ // implicitly destroy it is posted from ~URLRequestPostInterceptorFactory().
+ URLRequestPostInterceptor::Delegate* delegate_;
+
+ 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) {}
+ bool Match(const std::string& actual) const override;
+
+ private:
+ const std::string expected_;
+
+ DISALLOW_COPY_AND_ASSIGN(PartialMatch);
+};
+
+} // namespace update_client
+
+#endif // COMPONENTS_UPDATE_CLIENT_TEST_URL_REQUEST_POST_INTERCEPTOR_H_

Powered by Google App Engine
This is Rietveld 408576698