Index: chrome/browser/local_discovery/privet_http_unittest.cc |
diff --git a/chrome/browser/local_discovery/privet_http_unittest.cc b/chrome/browser/local_discovery/privet_http_unittest.cc |
index 1a524feb8f66f039d1912189a30b7559bdc7111b..03db69c433b4712d8efd6c042b649bee83db11d4 100644 |
--- a/chrome/browser/local_discovery/privet_http_unittest.cc |
+++ b/chrome/browser/local_discovery/privet_http_unittest.cc |
@@ -202,6 +202,8 @@ class MockTestURLFetcherFactoryDelegate |
class PrivetHTTPTest : public ::testing::Test { |
public: |
PrivetHTTPTest() { |
+ PrivetURLFetcher::ResetTokenMapForTests(); |
+ |
request_context_= new net::TestURLRequestContextGetter( |
base::MessageLoopProxy::current()); |
privet_client_.reset(new PrivetHTTPClientImpl( |
@@ -442,37 +444,6 @@ TEST_F(PrivetInfoTest, SuccessfulInfo) { |
EXPECT_CALL(info_callback_, OnPrivetJSONDoneInternal()); |
fetcher->delegate()->OnURLFetchComplete(fetcher); |
- |
- std::string name; |
- |
- privet_client_->GetCachedInfo()->GetString("name", &name); |
- EXPECT_EQ("Common printer", name); |
-}; |
- |
-TEST_F(PrivetInfoTest, InfoSaveToken) { |
- info_operation_->Start(); |
- |
- net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0); |
- ASSERT_TRUE(fetcher != NULL); |
- fetcher->SetResponseString(kSampleInfoResponse); |
- fetcher->set_status(net::URLRequestStatus(net::URLRequestStatus::SUCCESS, |
- net::OK)); |
- fetcher->set_response_code(200); |
- |
- EXPECT_CALL(info_callback_, OnPrivetJSONDoneInternal()); |
- fetcher->delegate()->OnURLFetchComplete(fetcher); |
- |
- info_operation_ = |
- privet_client_->CreateInfoOperation(info_callback_.callback()); |
- info_operation_->Start(); |
- |
- fetcher = fetcher_factory_.GetFetcherByID(0); |
- ASSERT_TRUE(fetcher != NULL); |
- net::HttpRequestHeaders headers; |
- fetcher->GetExtraRequestHeaders(&headers); |
- std::string header_token; |
- ASSERT_TRUE(headers.GetHeader("X-Privet-Token", &header_token)); |
- EXPECT_EQ("SampleTokenForTesting", header_token); |
}; |
TEST_F(PrivetInfoTest, InfoFailureHTTP) { |
@@ -486,7 +457,6 @@ TEST_F(PrivetInfoTest, InfoFailureHTTP) { |
EXPECT_CALL(info_callback_, OnPrivetJSONDoneInternal()); |
fetcher->delegate()->OnURLFetchComplete(fetcher); |
- EXPECT_EQ(NULL, privet_client_->GetCachedInfo()); |
}; |
class PrivetRegisterTest : public PrivetHTTPTest { |
@@ -528,15 +498,12 @@ class PrivetRegisterTest : public PrivetHTTPTest { |
}; |
TEST_F(PrivetRegisterTest, RegisterSuccessSimple) { |
- // Start with info request first to populate XSRF token. |
- info_operation_->Start(); |
+ register_operation_->Start(); |
EXPECT_TRUE(SuccessfulResponseToURL( |
GURL("http://10.0.0.8:6006/privet/info"), |
kSampleInfoResponse)); |
- register_operation_->Start(); |
- |
EXPECT_TRUE(SuccessfulResponseToURL( |
GURL("http://10.0.0.8:6006/privet/register?" |
"action=start&user=example%40google.com"), |
@@ -566,19 +533,6 @@ TEST_F(PrivetRegisterTest, RegisterSuccessSimple) { |
kSampleInfoResponseRegistered)); |
} |
-TEST_F(PrivetRegisterTest, RegisterNoInfoCall) { |
- register_operation_->Start(); |
- |
- EXPECT_TRUE(SuccessfulResponseToURL( |
- GURL("http://10.0.0.8:6006/privet/info"), |
- kSampleInfoResponse)); |
- |
- EXPECT_TRUE(SuccessfulResponseToURL( |
- GURL("http://10.0.0.8:6006/privet/register?" |
- "action=start&user=example%40google.com"), |
- kSampleRegisterStartResponse)); |
-} |
- |
TEST_F(PrivetRegisterTest, RegisterXSRFFailure) { |
register_operation_->Start(); |
@@ -672,15 +626,12 @@ TEST_F(PrivetRegisterTest, InfoFailure) { |
} |
TEST_F(PrivetRegisterTest, RegisterCancel) { |
- // Start with info request first to populate XSRF token. |
- info_operation_->Start(); |
+ register_operation_->Start(); |
EXPECT_TRUE(SuccessfulResponseToURL( |
GURL("http://10.0.0.8:6006/privet/info"), |
kSampleInfoResponse)); |
- register_operation_->Start(); |
- |
EXPECT_TRUE(SuccessfulResponseToURL( |
GURL("http://10.0.0.8:6006/privet/register?" |
"action=start&user=example%40google.com"), |
@@ -786,6 +737,8 @@ class PrivetLocalPrintTest : public PrivetHTTPTest { |
virtual ~PrivetLocalPrintTest() {} |
virtual void SetUp() OVERRIDE { |
+ PrivetURLFetcher::ResetTokenMapForTests(); |
+ |
local_print_operation_ = privet_client_->CreateLocalPrintOperation( |
&local_print_delegate_); |
@@ -817,9 +770,12 @@ TEST_F(PrivetLocalPrintTest, SuccessfulLocalPrint) { |
GURL("http://10.0.0.8:6006/privet/info"), |
kSampleInfoResponse)); |
- EXPECT_TRUE(SuccessfulResponseToURL( |
- GURL("http://10.0.0.8:6006/privet/capabilities"), |
- kSampleCapabilitiesResponse)); |
+ EXPECT_TRUE(SuccessfulResponseToURL(GURL("http://10.0.0.8:6006/privet/info"), |
+ kSampleInfoResponse)); |
+ |
+ EXPECT_TRUE( |
+ SuccessfulResponseToURL(GURL("http://10.0.0.8:6006/privet/capabilities"), |
+ kSampleCapabilitiesResponse)); |
EXPECT_CALL(local_print_delegate_, OnPrivetPrintingDoneInternal()); |
@@ -843,9 +799,12 @@ TEST_F(PrivetLocalPrintTest, SuccessfulLocalPrintWithAnyMimetype) { |
GURL("http://10.0.0.8:6006/privet/info"), |
kSampleInfoResponse)); |
- EXPECT_TRUE(SuccessfulResponseToURL( |
- GURL("http://10.0.0.8:6006/privet/capabilities"), |
- kSampleCapabilitiesResponseWithAnyMimetype)); |
+ EXPECT_TRUE(SuccessfulResponseToURL(GURL("http://10.0.0.8:6006/privet/info"), |
+ kSampleInfoResponse)); |
+ |
+ EXPECT_TRUE( |
+ SuccessfulResponseToURL(GURL("http://10.0.0.8:6006/privet/capabilities"), |
+ kSampleCapabilitiesResponseWithAnyMimetype)); |
EXPECT_CALL(local_print_delegate_, OnPrivetPrintingDoneInternal()); |
@@ -869,9 +828,12 @@ TEST_F(PrivetLocalPrintTest, SuccessfulPWGLocalPrint) { |
GURL("http://10.0.0.8:6006/privet/info"), |
kSampleInfoResponse)); |
- EXPECT_TRUE(SuccessfulResponseToURL( |
- GURL("http://10.0.0.8:6006/privet/capabilities"), |
- kSampleCapabilitiesResponsePWGOnly)); |
+ EXPECT_TRUE(SuccessfulResponseToURL(GURL("http://10.0.0.8:6006/privet/info"), |
+ kSampleInfoResponse)); |
+ |
+ EXPECT_TRUE( |
+ SuccessfulResponseToURL(GURL("http://10.0.0.8:6006/privet/capabilities"), |
+ kSampleCapabilitiesResponsePWGOnly)); |
EXPECT_CALL(local_print_delegate_, OnPrivetPrintingDoneInternal()); |
@@ -896,9 +858,12 @@ TEST_F(PrivetLocalPrintTest, SuccessfulLocalPrintWithCreatejob) { |
GURL("http://10.0.0.8:6006/privet/info"), |
kSampleInfoResponseWithCreatejob)); |
- EXPECT_TRUE(SuccessfulResponseToURL( |
- GURL("http://10.0.0.8:6006/privet/capabilities"), |
- kSampleCapabilitiesResponse)); |
+ EXPECT_TRUE(SuccessfulResponseToURL(GURL("http://10.0.0.8:6006/privet/info"), |
+ kSampleInfoResponse)); |
+ |
+ EXPECT_TRUE( |
+ SuccessfulResponseToURL(GURL("http://10.0.0.8:6006/privet/capabilities"), |
+ kSampleCapabilitiesResponse)); |
EXPECT_TRUE(SuccessfulResponseToURLAndData( |
GURL("http://10.0.0.8:6006/privet/printer/createjob"), |
@@ -928,9 +893,12 @@ TEST_F(PrivetLocalPrintTest, PDFPrintInvalidDocumentTypeRetry) { |
GURL("http://10.0.0.8:6006/privet/info"), |
kSampleInfoResponseWithCreatejob)); |
- EXPECT_TRUE(SuccessfulResponseToURL( |
- GURL("http://10.0.0.8:6006/privet/capabilities"), |
- kSampleCapabilitiesResponse)); |
+ EXPECT_TRUE(SuccessfulResponseToURL(GURL("http://10.0.0.8:6006/privet/info"), |
+ kSampleInfoResponse)); |
+ |
+ EXPECT_TRUE( |
+ SuccessfulResponseToURL(GURL("http://10.0.0.8:6006/privet/capabilities"), |
+ kSampleCapabilitiesResponse)); |
EXPECT_TRUE(SuccessfulResponseToURLAndData( |
GURL("http://10.0.0.8:6006/privet/printer/createjob"), |
@@ -967,6 +935,9 @@ TEST_F(PrivetLocalPrintTest, LocalPrintRetryOnInvalidJobID) { |
GURL("http://10.0.0.8:6006/privet/info"), |
kSampleInfoResponseWithCreatejob)); |
+ EXPECT_TRUE(SuccessfulResponseToURL(GURL("http://10.0.0.8:6006/privet/info"), |
+ kSampleInfoResponse)); |
+ |
EXPECT_TRUE(SuccessfulResponseToURL( |
GURL("http://10.0.0.8:6006/privet/capabilities"), |
kSampleCapabilitiesResponse)); |