Index: blimp/client/session/assignment_source_unittest.cc |
diff --git a/blimp/client/session/assignment_source_unittest.cc b/blimp/client/session/assignment_source_unittest.cc |
index 4930154792211579034510b561475f1791b574ce..4c940876b418cd5bf2be89ee95730971ed9002b0 100644 |
--- a/blimp/client/session/assignment_source_unittest.cc |
+++ b/blimp/client/session/assignment_source_unittest.cc |
@@ -12,13 +12,14 @@ |
#include "base/message_loop/message_loop.h" |
#include "base/path_service.h" |
#include "base/run_loop.h" |
+#include "base/strings/string_number_conversions.h" |
#include "base/test/test_simple_task_runner.h" |
#include "base/threading/thread_task_runner_handle.h" |
#include "base/values.h" |
#include "blimp/client/app/blimp_client_switches.h" |
#include "blimp/common/get_client_token.h" |
-#include "blimp/common/protocol_version.h" |
#include "blimp/common/switches.h" |
+#include "blimp/common/version_info.h" |
#include "components/safe_json/testing_json_parser.h" |
#include "net/base/test_data_directory.h" |
#include "net/url_request/test_url_fetcher_factory.h" |
@@ -106,7 +107,7 @@ class AssignmentSourceTest : public testing::Test { |
int status, |
const std::string& response, |
const std::string& client_auth_token, |
- const std::string& protocol_version) { |
+ const int protocol_version) { |
source_.GetAssignment(client_auth_token, |
base::Bind(&AssignmentSourceTest::AssignmentResponse, |
base::Unretained(this))); |
@@ -128,7 +129,8 @@ class AssignmentSourceTest : public testing::Test { |
std::string uploaded_protocol_version; |
EXPECT_TRUE( |
dict->GetString("protocol_version", &uploaded_protocol_version)); |
- EXPECT_EQ(protocol_version, uploaded_protocol_version); |
+ std::string expected_protocol_version = base::IntToString(protocol_version); |
+ EXPECT_EQ(expected_protocol_version, uploaded_protocol_version); |
// Check that the request has a valid authentication header. |
net::HttpRequestHeaders headers; |
@@ -259,7 +261,7 @@ TEST_F(AssignmentSourceTest, TestSuccess) { |
GetNetworkAssignmentAndWaitForResponse( |
net::HTTP_OK, net::Error::OK, ValueToString(*BuildAssignerResponse()), |
- kTestAuthToken, kEngineVersion); |
+ kTestAuthToken, GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestValidAfterError) { |
@@ -276,28 +278,29 @@ TEST_F(AssignmentSourceTest, TestValidAfterError) { |
.Times(1) |
.RetiresOnSaturation(); |
- GetNetworkAssignmentAndWaitForResponse(net::HTTP_OK, |
- net::Error::ERR_INSUFFICIENT_RESOURCES, |
- "", kTestAuthToken, kEngineVersion); |
+ GetNetworkAssignmentAndWaitForResponse( |
+ net::HTTP_OK, net::Error::ERR_INSUFFICIENT_RESOURCES, "", kTestAuthToken, |
+ GetProtocolVersion()); |
GetNetworkAssignmentAndWaitForResponse( |
net::HTTP_OK, net::Error::OK, ValueToString(*BuildAssignerResponse()), |
- kTestAuthToken, kEngineVersion); |
+ kTestAuthToken, GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestNetworkFailure) { |
EXPECT_CALL(*this, AssignmentResponse( |
AssignmentSource::Result::RESULT_NETWORK_FAILURE, _)); |
- GetNetworkAssignmentAndWaitForResponse(net::HTTP_OK, |
- net::Error::ERR_INSUFFICIENT_RESOURCES, |
- "", kTestAuthToken, kEngineVersion); |
+ GetNetworkAssignmentAndWaitForResponse( |
+ net::HTTP_OK, net::Error::ERR_INSUFFICIENT_RESOURCES, "", kTestAuthToken, |
+ GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestBadRequest) { |
EXPECT_CALL(*this, AssignmentResponse( |
AssignmentSource::Result::RESULT_BAD_REQUEST, _)); |
GetNetworkAssignmentAndWaitForResponse(net::HTTP_BAD_REQUEST, net::Error::OK, |
- "", kTestAuthToken, kEngineVersion); |
+ "", kTestAuthToken, |
+ GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestUnauthorized) { |
@@ -305,14 +308,16 @@ TEST_F(AssignmentSourceTest, TestUnauthorized) { |
AssignmentResponse( |
AssignmentSource::Result::RESULT_EXPIRED_ACCESS_TOKEN, _)); |
GetNetworkAssignmentAndWaitForResponse(net::HTTP_UNAUTHORIZED, net::Error::OK, |
- "", kTestAuthToken, kEngineVersion); |
+ "", kTestAuthToken, |
+ GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestForbidden) { |
EXPECT_CALL(*this, AssignmentResponse( |
AssignmentSource::Result::RESULT_USER_INVALID, _)); |
GetNetworkAssignmentAndWaitForResponse(net::HTTP_FORBIDDEN, net::Error::OK, |
- "", kTestAuthToken, kEngineVersion); |
+ "", kTestAuthToken, |
+ GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestTooManyRequests) { |
@@ -320,7 +325,7 @@ TEST_F(AssignmentSourceTest, TestTooManyRequests) { |
AssignmentSource::Result::RESULT_OUT_OF_VMS, _)); |
GetNetworkAssignmentAndWaitForResponse(static_cast<net::HttpStatusCode>(429), |
net::Error::OK, "", kTestAuthToken, |
- kEngineVersion); |
+ GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestInternalServerError) { |
@@ -328,7 +333,7 @@ TEST_F(AssignmentSourceTest, TestInternalServerError) { |
AssignmentSource::Result::RESULT_SERVER_ERROR, _)); |
GetNetworkAssignmentAndWaitForResponse(net::HTTP_INTERNAL_SERVER_ERROR, |
net::Error::OK, "", kTestAuthToken, |
- kEngineVersion); |
+ GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestUnexpectedNetCodeFallback) { |
@@ -336,7 +341,7 @@ TEST_F(AssignmentSourceTest, TestUnexpectedNetCodeFallback) { |
AssignmentSource::Result::RESULT_BAD_RESPONSE, _)); |
GetNetworkAssignmentAndWaitForResponse(net::HTTP_NOT_IMPLEMENTED, |
net::Error::OK, "", kTestAuthToken, |
- kEngineVersion); |
+ GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestInvalidJsonResponse) { |
@@ -349,7 +354,7 @@ TEST_F(AssignmentSourceTest, TestInvalidJsonResponse) { |
EXPECT_CALL(*this, AssignmentResponse( |
AssignmentSource::Result::RESULT_BAD_RESPONSE, _)); |
GetNetworkAssignmentAndWaitForResponse(net::HTTP_OK, net::Error::OK, response, |
- kTestAuthToken, kEngineVersion); |
+ kTestAuthToken, GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestMissingResponsePort) { |
@@ -359,7 +364,7 @@ TEST_F(AssignmentSourceTest, TestMissingResponsePort) { |
AssignmentSource::Result::RESULT_BAD_RESPONSE, _)); |
GetNetworkAssignmentAndWaitForResponse(net::HTTP_OK, net::Error::OK, |
ValueToString(*response), |
- kTestAuthToken, kEngineVersion); |
+ kTestAuthToken, GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestInvalidIPAddress) { |
@@ -370,7 +375,7 @@ TEST_F(AssignmentSourceTest, TestInvalidIPAddress) { |
AssignmentSource::Result::RESULT_BAD_RESPONSE, _)); |
GetNetworkAssignmentAndWaitForResponse(net::HTTP_OK, net::Error::OK, |
ValueToString(*response), |
- kTestAuthToken, kEngineVersion); |
+ kTestAuthToken, GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestMissingCert) { |
@@ -380,7 +385,7 @@ TEST_F(AssignmentSourceTest, TestMissingCert) { |
AssignmentSource::Result::RESULT_BAD_RESPONSE, _)); |
GetNetworkAssignmentAndWaitForResponse(net::HTTP_OK, net::Error::OK, |
ValueToString(*response), |
- kTestAuthToken, kEngineVersion); |
+ kTestAuthToken, GetProtocolVersion()); |
} |
TEST_F(AssignmentSourceTest, TestInvalidCert) { |
@@ -390,7 +395,7 @@ TEST_F(AssignmentSourceTest, TestInvalidCert) { |
AssignmentSource::Result::RESULT_INVALID_CERT, _)); |
GetNetworkAssignmentAndWaitForResponse(net::HTTP_OK, net::Error::OK, |
ValueToString(*response), |
- kTestAuthToken, kEngineVersion); |
+ kTestAuthToken, GetProtocolVersion()); |
} |
} // namespace |