Adds EngineGeolocationFeature for Blimp Geolocation project.

blimp/engine/feature/geolocation/engine_geolocation_feature_unittest.cc

Index: blimp/engine/feature/geolocation/engine_geolocation_feature_unittest.cc
diff --git a/blimp/engine/feature/geolocation/engine_geolocation_feature_unittest.cc b/blimp/engine/feature/geolocation/engine_geolocation_feature_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..4e684a23b104f4320c8d3af5ade584b2d5ea09fa
--- /dev/null
+++ b/blimp/engine/feature/geolocation/engine_geolocation_feature_unittest.cc
@@ -0,0 +1,204 @@
+// Copyright 2016 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.
+
+#include "blimp/engine/feature/geolocation/engine_geolocation_feature.h"
+
+#include <memory>
+#include <string>
+#include <utility>
+
+#include "base/memory/ptr_util.h"
+#include "blimp/common/create_blimp_message.h"
+#include "blimp/common/proto/blimp_message.pb.h"
+#include "blimp/common/proto/geolocation.pb.h"
+#include "blimp/net/test_common.h"
+#include "content/public/browser/location_provider.h"
+#include "content/public/common/geoposition.h"
+#include "net/base/test_completion_callback.h"
+#include "net/test/gtest_util.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using testing::_;
+
+namespace blimp {
+namespace engine {
+
+void SendMockLocationMessage(BlimpMessageProcessor* processor) {
+  GeolocationMessage* geolocation_message;
+  std::unique_ptr<BlimpMessage> message =
+      CreateBlimpMessage(&geolocation_message);
+
+  GeolocationCoordinatesMessage* coordinates_message =
+      geolocation_message->mutable_coordinates();
+  coordinates_message->set_latitude(-42.0);
+  coordinates_message->set_longitude(17.3);
+  coordinates_message->set_altitude(123.4);
+  coordinates_message->set_accuracy(73.7);
+
+  net::TestCompletionCallback cb;
+  processor->ProcessMessage(std::move(message), cb.callback());
+  EXPECT_EQ(net::OK, cb.WaitForResult());
+}
+
+void SendMockErrorMessage(BlimpMessageProcessor* processor,
+                          GeolocationErrorMessage::ErrorCode error,
+                          std::string error_message) {
+  GeolocationMessage* geolocation_message;
+  std::unique_ptr<BlimpMessage> message =
+      CreateBlimpMessage(&geolocation_message);
+  GeolocationErrorMessage* geolocation_error_message =
+      geolocation_message->mutable_error();
+  geolocation_error_message->set_error_code(error);
+  geolocation_error_message->set_error_message(error_message);
+
+  net::TestCompletionCallback cb;
+  processor->ProcessMessage(std::move(message), cb.callback());
+  EXPECT_EQ(net::OK, cb.WaitForResult());
+}
+
+void SendMalformedMessage(BlimpMessageProcessor* processor) {
+  GeolocationMessage* geolocation_message;
+  std::unique_ptr<BlimpMessage> message =
+      CreateBlimpMessage(&geolocation_message);
+
+  net::TestCompletionCallback cb;
+  processor->ProcessMessage(std::move(message), cb.callback());
+  EXPECT_EQ(net::OK, cb.WaitForResult());
+}
+
+void SendUnexpectedMessage(BlimpMessageProcessor* processor) {
+  GeolocationMessage* geolocation_message;
+  std::unique_ptr<BlimpMessage> message =
+      CreateBlimpMessage(&geolocation_message);
+  GeolocationSetInterestLevelMessage* geolocation_set_interest_level =
+      geolocation_message->mutable_set_interest_level();
+  geolocation_set_interest_level->set_level(
+      GeolocationSetInterestLevelMessage::NO_INTEREST);
+
+  net::TestCompletionCallback cb;
+  processor->ProcessMessage(std::move(message), cb.callback());
+  EXPECT_EQ(net::ERR_UNEXPECTED, cb.WaitForResult());

[Wez, 2016/07/22 22:09:44]

nit: I would prefer that the callback and EXPECT are in the test itself, so it's
clear to the reader how we're verifying the expected behaviour.  Personally I'd
have this function be CreateUnexpectedMessage() and then put these three lines
in the test itself. :)

+}
+
+MATCHER_P(EqualsUpdatedRequestLevel, level, "") {
+  return arg.feature_case() == BlimpMessage::kGeolocation &&
+         arg.geolocation().type_case() ==
+             GeolocationMessage::kSetInterestLevel &&
+         arg.geolocation().set_interest_level().level() == level;
+}
+
+MATCHER(EqualsRequestRefresh, "") {
+  return arg.feature_case() == BlimpMessage::kGeolocation &&
+         arg.geolocation().type_case() == GeolocationMessage::kRequestRefresh;
+}
+
+class EngineGeolocationFeatureTest : public testing::Test {
+ public:
+  EngineGeolocationFeatureTest()
+      : out_processor_(nullptr),
+        location_provider_(feature_.CreateGeolocationDelegate()
+                               ->OverrideSystemLocationProvider()),
+        mock_callback_(
+            base::Bind(&EngineGeolocationFeatureTest::OnLocationUpdate,
+                       base::Unretained(this))) {}
+
+  void SetUp() override {
+    out_processor_ = new MockBlimpMessageProcessor();
+    feature_.set_outgoing_message_processor(base::WrapUnique(out_processor_));
+    location_provider_->SetUpdateCallback(mock_callback_);
+  }
+
+  void OnLocationUpdate(const content::LocationProvider* provider,
+                        const content::Geoposition& geoposition) {
+    received_position = geoposition;
+  }
+
+ protected:
+  // This is a raw pointer to a class that is owned by the GeolocationFeature.
+  MockBlimpMessageProcessor* out_processor_;
+
+  EngineGeolocationFeature feature_;
+  std::unique_ptr<content::LocationProvider> location_provider_;
+  content::LocationProvider::LocationProviderUpdateCallback mock_callback_;
+  content::Geoposition received_position;
+};
+
+TEST_F(EngineGeolocationFeatureTest, LocationReceived) {
+  SendMockLocationMessage(&feature_);
+  EXPECT_EQ(content::Geoposition::ERROR_CODE_NONE,
+            received_position.error_code);
+  EXPECT_EQ(-42.0, received_position.latitude);
+  EXPECT_EQ(17.3, received_position.longitude);
+  EXPECT_EQ(123.4, received_position.altitude);
+  EXPECT_EQ(73.7, received_position.accuracy);
+}
+
+TEST_F(EngineGeolocationFeatureTest, ErrorRecieved) {
+  SendMockErrorMessage(&feature_, GeolocationErrorMessage::PERMISSION_DENIED,
+                       "PERMISSION_DENIED");
+  EXPECT_EQ(content::Geoposition::ERROR_CODE_PERMISSION_DENIED,
+            received_position.error_code);
+  EXPECT_EQ("PERMISSION_DENIED", received_position.error_message);
+
+  SendMockErrorMessage(&feature_, GeolocationErrorMessage::POSITION_UNAVAILABLE,
+                       "POSITION_UNAVAILABLE");
+  EXPECT_EQ(content::Geoposition::ERROR_CODE_POSITION_UNAVAILABLE,
+            received_position.error_code);
+  EXPECT_EQ("POSITION_UNAVAILABLE", received_position.error_message);
+}
+
+TEST_F(EngineGeolocationFeatureTest, UpdateRequestLevel) {
+  EXPECT_CALL(*out_processor_,
+              MockableProcessMessage(
+                  EqualsUpdatedRequestLevel(
+                      GeolocationSetInterestLevelMessage::HIGH_ACCURACY),
+                  _))
+      .Times(1);
+  EXPECT_CALL(*out_processor_,
+              MockableProcessMessage(
+                  EqualsUpdatedRequestLevel(
+                      GeolocationSetInterestLevelMessage::LOW_ACCURACY),
+                  _))
+      .Times(1);
+  EXPECT_CALL(*out_processor_,
+              MockableProcessMessage(
+                  EqualsUpdatedRequestLevel(
+                      GeolocationSetInterestLevelMessage::NO_INTEREST),
+                  _))
+      .Times(1);
+
+  location_provider_->StartProvider(true);
+  location_provider_->StartProvider(false);
+  location_provider_->StopProvider();
+}
+
+TEST_F(EngineGeolocationFeatureTest, UnexpectedMessageReceived) {
+  SendUnexpectedMessage(&feature_);
+}
+
+TEST_F(EngineGeolocationFeatureTest, RequestRefresh) {
+  EXPECT_CALL(*out_processor_,
+              MockableProcessMessage(
+                  EqualsUpdatedRequestLevel(
+                      GeolocationSetInterestLevelMessage::HIGH_ACCURACY),
+                  _))
+      .Times(1);
+
+  EXPECT_CALL(*out_processor_,
+              MockableProcessMessage(EqualsRequestRefresh(), _))
+      .Times(1);
+  EXPECT_CALL(*out_processor_,
+              MockableProcessMessage(
+                  EqualsUpdatedRequestLevel(
+                      GeolocationSetInterestLevelMessage::NO_INTEREST),
+                  _))
+      .Times(1);
+
+  location_provider_->StartProvider(true);
+  location_provider_->RequestRefresh();
+}
+
+}  // namespace engine
+}  // namespace blimp