Fixing memory leak in DistilledPagePrefs unittests

components/dom_distiller/core/distilled_page_prefs_unittests.cc

 // Copyright 2014 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 "components/dom_distiller/core/distilled_page_prefs.h"
 
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "components/pref_registry/testing_pref_service_syncable.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ skipping 14 matching lines @@
 
  private:
   DistilledPagePrefs::Theme theme_;
 };
 
 }  // namespace
 
 class DistilledPagePrefsTest : public testing::Test {
  protected:
   virtual void SetUp() OVERRIDE {
-    user_prefs::TestingPrefServiceSyncable* pref_service =
-        new user_prefs::TestingPrefServiceSyncable();
-    DistilledPagePrefs::RegisterProfilePrefs(pref_service->registry());
-    distilled_page_prefs_ = new DistilledPagePrefs(pref_service);
+    pref_service_.reset(new user_prefs::TestingPrefServiceSyncable());
+    DistilledPagePrefs::RegisterProfilePrefs(pref_service_->registry());
+    distilled_page_prefs_.reset(new DistilledPagePrefs(pref_service_.get()));
   }
 
-  DistilledPagePrefs* distilled_page_prefs_;
+  scoped_ptr<user_prefs::TestingPrefServiceSyncable> pref_service_;
+  scoped_ptr<DistilledPagePrefs> distilled_page_prefs_;
 
  private:
   base::MessageLoop message_loop_;
 };
 
 TEST_F(DistilledPagePrefsTest, TestingOnChangeThemeIsBeingCalled) {
-  TestingObserver* obs = new TestingObserver();
-  distilled_page_prefs_->AddObserver(obs);
+  TestingObserver obs;
+  distilled_page_prefs_->AddObserver(&obs);
   distilled_page_prefs_->SetTheme(DistilledPagePrefs::SEPIA);
-  EXPECT_EQ(DistilledPagePrefs::LIGHT, obs->GetTheme());
+  EXPECT_EQ(DistilledPagePrefs::LIGHT, obs.GetTheme());
   base::RunLoop().RunUntilIdle();
-  EXPECT_EQ(DistilledPagePrefs::SEPIA, obs->GetTheme());
+  EXPECT_EQ(DistilledPagePrefs::SEPIA, obs.GetTheme());
   distilled_page_prefs_->SetTheme(DistilledPagePrefs::DARK);
   base::RunLoop().RunUntilIdle();
-  EXPECT_EQ(DistilledPagePrefs::DARK, obs->GetTheme());
+  EXPECT_EQ(DistilledPagePrefs::DARK, obs.GetTheme());
+  distilled_page_prefs_->RemoveObserver(&obs);
 }
 
 TEST_F(DistilledPagePrefsTest, TestingMultipleObservers) {
-  TestingObserver* obs = new TestingObserver();
-  distilled_page_prefs_->AddObserver(obs);
-  TestingObserver* obs2 = new TestingObserver();
-  distilled_page_prefs_->AddObserver(obs2);
+  TestingObserver obs;
+  distilled_page_prefs_->AddObserver(&obs);
+  TestingObserver obs2;
+  distilled_page_prefs_->AddObserver(&obs2);
   distilled_page_prefs_->SetTheme(DistilledPagePrefs::SEPIA);
   base::RunLoop().RunUntilIdle();
-  EXPECT_EQ(DistilledPagePrefs::SEPIA, obs->GetTheme());
-  EXPECT_EQ(DistilledPagePrefs::SEPIA, obs2->GetTheme());
-  distilled_page_prefs_->RemoveObserver(obs);
+  EXPECT_EQ(DistilledPagePrefs::SEPIA, obs.GetTheme());
+  EXPECT_EQ(DistilledPagePrefs::SEPIA, obs2.GetTheme());
+  distilled_page_prefs_->RemoveObserver(&obs);
   distilled_page_prefs_->SetTheme(DistilledPagePrefs::LIGHT);
   base::RunLoop().RunUntilIdle();
-  EXPECT_EQ(DistilledPagePrefs::SEPIA, obs->GetTheme());
-  EXPECT_EQ(DistilledPagePrefs::LIGHT, obs2->GetTheme());
+  EXPECT_EQ(DistilledPagePrefs::SEPIA, obs.GetTheme());
+  EXPECT_EQ(DistilledPagePrefs::LIGHT, obs2.GetTheme());
+  distilled_page_prefs_->RemoveObserver(&obs2);
 }
 
 }  // namespace dom_distiller