Tied up BlimpNavigationController to NavigationFeature

blimp/client/core/blimp_contents_impl_unittest.cc

 // 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/client/core/blimp_contents_impl.h"
 
+#include "base/macros.h"
+#include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "blimp/client/core/blimp_contents_impl.h"
 #include "blimp/client/core/public/blimp_contents_observer.h"
+#include "blimp/client/feature/mock_navigation_feature.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blimp {
 namespace client {
 namespace {
 
 const GURL kExampleURL = GURL("https://www.example.com/");
 const GURL kOtherExampleURL = GURL("https://www.otherexample.com/");
 
 class TestBlimpContentsObserver : public BlimpContentsObserver {
  public:
   TestBlimpContentsObserver() = default;
   ~TestBlimpContentsObserver() override = default;
 
   void OnURLUpdated(const GURL& url) override { last_url_ = url; }
 
   GURL GetLastURL() { return last_url_; }
 
  private:
   GURL last_url_;
 
   DISALLOW_COPY_AND_ASSIGN(TestBlimpContentsObserver);
 };
 
+class TestBlimpNavigationControllerDelegate
+    : public BlimpNavigationControllerDelegate {
+ public:
+  TestBlimpNavigationControllerDelegate() = default;
+  ~TestBlimpNavigationControllerDelegate() override = default;
+
+ private:
+  void OnUrlChanged(const GURL& url) override {}

[David Trainor- moved to gerrit, 2016/06/29 17:27:15]

// BlimpNavigationControllerDelegate implementation.

[shaktisahu, 2016/07/19 21:45:29]

Done.

+  void OnFaviconChanged(const SkBitmap& favicon) override {}
+  void OnTitleChanged(const std::string& title) override {}
+  void OnLoadingChanged(bool loading) override {}
+  void OnPageLoadStatusUpdate(bool completed) override {}
+  DISALLOW_COPY_AND_ASSIGN(TestBlimpNavigationControllerDelegate);

[David Trainor- moved to gerrit, 2016/06/29 17:27:15]

New line before this?

[shaktisahu, 2016/07/19 21:45:29]

Done.

+};
+
 TEST(BlimpContentsImplTest, Basic) {
   base::MessageLoop loop;
   BlimpContentsImpl blimp_contents;
 
   BlimpNavigationController& navigation_controller =
       blimp_contents.GetNavigationController();
 
   TestBlimpContentsObserver observer1;
   blimp_contents.AddObserver(&observer1);
   TestBlimpContentsObserver observer2;
   blimp_contents.AddObserver(&observer2);
 
+  BlimpNavigationControllerImpl* nav_controller_impl =
+      static_cast<BlimpNavigationControllerImpl*>(&navigation_controller);
+  MockNavigationFeature feature;
+  feature.SetDelegate(1, nav_controller_impl);
+  nav_controller_impl->SetNavigationFeatureForTesting(&feature);
+
   navigation_controller.LoadURL(kExampleURL);
   loop.RunUntilIdle();
 
   EXPECT_EQ(kExampleURL, navigation_controller.GetURL());
   EXPECT_EQ(kExampleURL, observer1.GetLastURL());
   EXPECT_EQ(kExampleURL, observer2.GetLastURL());
 
   // Observer should no longer receive callbacks.
   blimp_contents.RemoveObserver(&observer1);
 
   navigation_controller.LoadURL(kOtherExampleURL);
   loop.RunUntilIdle();
 
   EXPECT_EQ(kOtherExampleURL, navigation_controller.GetURL());
   EXPECT_EQ(kExampleURL, observer1.GetLastURL());
   EXPECT_EQ(kOtherExampleURL, observer2.GetLastURL());
 }
 
 }  // namespace
 }  // namespace client
 }  // namespace blimp