81 lines
2.0 KiB
C++
81 lines
2.0 KiB
C++
|
// Copyright 2012 The Chromium Authors
|
||
|
|
// Use of this source code is governed by a BSD-style license that can be
|
||
|
|
// found in the LICENSE file.
|
||
|
|
|
||
|
|
#include "net/proxy_resolution/mock_pac_file_fetcher.h"
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
#include <utility>
|
||
|
|
|
||
|
|
#include "base/check.h"
|
||
|
|
#include "base/run_loop.h"
|
||
|
|
#include "base/strings/utf_string_conversions.h"
|
||
|
|
#include "net/base/net_errors.h"
|
||
|
|
|
||
|
|
namespace net {
|
||
|
|
|
||
|
|
MockPacFileFetcher::MockPacFileFetcher() = default;
|
||
|
|
|
||
|
|
MockPacFileFetcher::~MockPacFileFetcher() = default;
|
||
|
|
|
||
|
|
// PacFileFetcher implementation.
|
||
|
|
int MockPacFileFetcher::Fetch(
|
||
|
|
const GURL& url,
|
||
|
|
std::u16string* text,
|
||
|
|
CompletionOnceCallback callback,
|
||
|
|
const NetworkTrafficAnnotationTag traffic_annotation) {
|
||
|
|
DCHECK(!has_pending_request());
|
||
|
|
|
||
|
|
if (on_fetch_complete_)
|
||
|
|
std::move(on_fetch_complete_).Run();
|
||
|
|
|
||
|
|
if (is_shutdown_)
|
||
|
|
return ERR_CONTEXT_SHUT_DOWN;
|
||
|
|
|
||
|
|
// Save the caller's information, and have them wait.
|
||
|
|
pending_request_url_ = url;
|
||
|
|
pending_request_callback_ = std::move(callback);
|
||
|
|
pending_request_text_ = text;
|
||
|
|
|
||
|
|
return ERR_IO_PENDING;
|
||
|
|
}
|
||
|
|
|
||
|
|
void MockPacFileFetcher::NotifyFetchCompletion(int result,
|
||
|
|
const std::string& ascii_text) {
|
||
|
|
DCHECK(has_pending_request());
|
||
|
|
*pending_request_text_ = base::ASCIIToUTF16(ascii_text);
|
||
|
|
std::move(pending_request_callback_).Run(result);
|
||
|
|
}
|
||
|
|
|
||
|
|
void MockPacFileFetcher::Cancel() {
|
||
|
|
pending_request_callback_.Reset();
|
||
|
|
}
|
||
|
|
|
||
|
|
void MockPacFileFetcher::OnShutdown() {
|
||
|
|
is_shutdown_ = true;
|
||
|
|
if (pending_request_callback_) {
|
||
|
|
std::move(pending_request_callback_).Run(ERR_CONTEXT_SHUT_DOWN);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
URLRequestContext* MockPacFileFetcher::GetRequestContext() const {
|
||
|
|
return nullptr;
|
||
|
|
}
|
||
|
|
|
||
|
|
const GURL& MockPacFileFetcher::pending_request_url() const {
|
||
|
|
return pending_request_url_;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool MockPacFileFetcher::has_pending_request() const {
|
||
|
|
return !pending_request_callback_.is_null();
|
||
|
|
}
|
||
|
|
|
||
|
|
void MockPacFileFetcher::WaitUntilFetch() {
|
||
|
|
DCHECK(!has_pending_request());
|
||
|
|
base::RunLoop run_loop;
|
||
|
|
on_fetch_complete_ = run_loop.QuitClosure();
|
||
|
|
run_loop.Run();
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace net
|