Home | History | Annotate | Download | only in tab_model 
      1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "chrome/browser/chrome_notification_types.h"
      6 #include "chrome/browser/ui/android/tab_model/tab_model.h"
      7 #include "chrome/test/base/testing_profile.h"
      8 #include "content/public/browser/notification_service.h"
      9 #include "content/public/browser/notification_source.h"
     10 #include "testing/gmock/include/gmock/gmock.h"
     11 #include "testing/gtest/include/gtest/gtest.h"
     12 
     13 typedef testing::Test TabModelTest;
     14 
     15 namespace {
     16 class TabModelAndroidProfileMock : public TestingProfile {
     17  public:
     18   TabModelAndroidProfileMock() {}
     19   virtual ~TabModelAndroidProfileMock() {}
     20 
     21   MOCK_METHOD0(GetOffTheRecordProfile, Profile*());
     22   MOCK_METHOD0(HasOffTheRecordProfile, bool());
     23 };
     24 }  // namespace
     25 
     26 class TestTabModel : public TabModel {
     27  public:
     28   explicit TestTabModel(Profile* profile)
     29     : TabModel(profile) {}
     30 
     31   virtual int GetTabCount() const OVERRIDE { return 0; }
     32   virtual int GetActiveIndex() const OVERRIDE { return 0; }
     33   virtual content::WebContents* GetWebContentsAt(int index) const OVERRIDE {
     34     return NULL;
     35   }
     36   virtual void CreateTab(content::WebContents* web_contents) OVERRIDE {}
     37   virtual content::WebContents* CreateTabForTesting(const GURL& url) OVERRIDE {
     38     return NULL;
     39   }
     40   virtual bool IsSessionRestoreInProgress() const OVERRIDE { return false; }
     41   virtual void OpenClearBrowsingData() const OVERRIDE {}
     42   virtual browser_sync::SyncedTabDelegate* GetTabAt(int index) const OVERRIDE {
     43     return NULL;
     44   }
     45 
     46 };
     47 
     48 TEST_F(TabModelTest, TestProfileHandling) {
     49   // Construct TabModel with standard Profile.
     50   TestingProfile testing_profile;
     51   TestTabModel tab_model(&testing_profile);
     52 
     53   // Verify TabModel has the correct profile and profile type.
     54   EXPECT_EQ(&testing_profile, tab_model.GetProfile());
     55   EXPECT_FALSE(tab_model.IsOffTheRecord());
     56 
     57   // Notify profile is being destroyed and verify pointer is cleared.
     58   content::NotificationService::current()->Notify(
     59     chrome::NOTIFICATION_PROFILE_DESTROYED,
     60     content::Source<Profile>(&testing_profile),
     61     content::NotificationService::NoDetails());
     62   EXPECT_EQ(NULL, tab_model.GetProfile());
     63 }
     64 
     65 TEST_F(TabModelTest, TestProfileHandlingOffTheRecord) {
     66   // Construct TabModel with off-the-record Profile.
     67   TabModelAndroidProfileMock testing_profile;
     68   EXPECT_CALL(testing_profile, HasOffTheRecordProfile())
     69     .WillOnce(testing::Return(true));
     70   EXPECT_CALL(testing_profile, GetOffTheRecordProfile())
     71     .WillOnce(testing::Return(&testing_profile));
     72   TestTabModel tab_model(&testing_profile);
     73 
     74   // Verify TabModel has the correct profile and profile type.
     75   EXPECT_EQ(&testing_profile, tab_model.GetProfile());
     76   EXPECT_TRUE(tab_model.IsOffTheRecord());
     77 
     78   // Notify profile is being destroyed and verify pointer is cleared.
     79   content::NotificationService::current()->Notify(
     80     chrome::NOTIFICATION_PROFILE_DESTROYED,
     81     content::Source<Profile>(&testing_profile),
     82     content::NotificationService::NoDetails());
     83   EXPECT_EQ(NULL, tab_model.GetProfile());
     84 }
     85