| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "ash/shelf/scoped_observer_with_duplicated_sources.h" | 5 #include "ash/shelf/scoped_observer_with_duplicated_sources.h" |
| 6 | 6 |
| 7 #include "testing/gtest/include/gtest/gtest.h" | 7 #include "testing/gtest/include/gtest/gtest.h" |
| 8 | 8 |
| 9 class TestObserver { | 9 class TestObserver { |
| 10 public: | 10 public: |
| 11 TestObserver() {} | 11 TestObserver() {} |
| 12 ~TestObserver() {} | 12 ~TestObserver() {} |
| 13 | 13 |
| 14 private: | 14 private: |
| 15 DISALLOW_COPY_AND_ASSIGN(TestObserver); | 15 DISALLOW_COPY_AND_ASSIGN(TestObserver); |
| 16 }; | 16 }; |
| 17 | 17 |
| 18 class TestSource { | 18 class TestSource { |
| 19 public: | 19 public: |
| 20 TestSource() : observer_count_(0) {} | 20 TestSource() : observer_count_(0) {} |
| 21 ~TestSource() {} | 21 ~TestSource() {} |
| 22 | 22 |
| 23 void AddObserver(TestObserver* observer) { | 23 void AddObserver(TestObserver* observer) { observer_count_++; } |
| 24 observer_count_++; | 24 void RemoveObserver(TestObserver* observer) { observer_count_--; } |
| 25 } | |
| 26 void RemoveObserver(TestObserver* observer) { | |
| 27 observer_count_--; | |
| 28 } | |
| 29 | 25 |
| 30 int GetObserverCount() { | 26 int GetObserverCount() { return observer_count_; } |
| 31 return observer_count_; | |
| 32 } | |
| 33 | 27 |
| 34 private: | 28 private: |
| 35 int observer_count_; | 29 int observer_count_; |
| 36 | 30 |
| 37 DISALLOW_COPY_AND_ASSIGN(TestSource); | 31 DISALLOW_COPY_AND_ASSIGN(TestSource); |
| 38 }; | 32 }; |
| 39 | 33 |
| 40 TEST(ScopedObserverWithDuplicatedSourcesTest, DuplicatedSource) { | 34 TEST(ScopedObserverWithDuplicatedSourcesTest, DuplicatedSource) { |
| 41 TestObserver observer; | 35 TestObserver observer; |
| 42 TestSource source1; | 36 TestSource source1; |
| 43 TestSource source2; | 37 TestSource source2; |
| 44 | 38 |
| 45 ScopedObserverWithDuplicatedSources<TestSource, TestObserver> | 39 ScopedObserverWithDuplicatedSources<TestSource, TestObserver> observers( |
| 46 observers(&observer); | 40 &observer); |
| 47 EXPECT_EQ(0, source1.GetObserverCount()); | 41 EXPECT_EQ(0, source1.GetObserverCount()); |
| 48 EXPECT_FALSE(observers.IsObserving(&source1)); | 42 EXPECT_FALSE(observers.IsObserving(&source1)); |
| 49 EXPECT_EQ(0, source2.GetObserverCount()); | 43 EXPECT_EQ(0, source2.GetObserverCount()); |
| 50 EXPECT_FALSE(observers.IsObserving(&source2)); | 44 EXPECT_FALSE(observers.IsObserving(&source2)); |
| 51 | 45 |
| 52 // Add |source1|. | 46 // Add |source1|. |
| 53 observers.Add(&source1); | 47 observers.Add(&source1); |
| 54 EXPECT_EQ(1, source1.GetObserverCount()); | 48 EXPECT_EQ(1, source1.GetObserverCount()); |
| 55 EXPECT_TRUE(observers.IsObserving(&source1)); | 49 EXPECT_TRUE(observers.IsObserving(&source1)); |
| 56 // AddObserver of TestSource is called only once. | 50 // AddObserver of TestSource is called only once. |
| (...skipping 15 matching lines...) Expand all Loading... |
| 72 observers.Remove(&source2); | 66 observers.Remove(&source2); |
| 73 EXPECT_EQ(0, source2.GetObserverCount()); | 67 EXPECT_EQ(0, source2.GetObserverCount()); |
| 74 EXPECT_FALSE(observers.IsObserving(&source2)); | 68 EXPECT_FALSE(observers.IsObserving(&source2)); |
| 75 | 69 |
| 76 // Remove |source1| again. | 70 // Remove |source1| again. |
| 77 observers.Remove(&source1); | 71 observers.Remove(&source1); |
| 78 // In this time, |observer| is removed from |source1|. | 72 // In this time, |observer| is removed from |source1|. |
| 79 EXPECT_EQ(0, source1.GetObserverCount()); | 73 EXPECT_EQ(0, source1.GetObserverCount()); |
| 80 EXPECT_FALSE(observers.IsObserving(&source1)); | 74 EXPECT_FALSE(observers.IsObserving(&source1)); |
| 81 } | 75 } |
| OLD | NEW |