| Index: chrome/browser/ui/global_error/global_error_service_unittest.cc
|
| diff --git a/chrome/browser/ui/global_error/global_error_service_unittest.cc b/chrome/browser/ui/global_error/global_error_service_unittest.cc
|
| index 333461eacf80883fe21be1b2395ff8670b221b62..4ca56a7432aaab64bdc848d296f1cc5ba1f52c68 100644
|
| --- a/chrome/browser/ui/global_error/global_error_service_unittest.cc
|
| +++ b/chrome/browser/ui/global_error/global_error_service_unittest.cc
|
| @@ -85,6 +85,9 @@ class BadgeError : public BaseError {
|
| public:
|
| explicit BadgeError(int resource_id) : resource_id_(resource_id) {}
|
| bool HasBadge() OVERRIDE { return true; }
|
| + Severity GetSeverity() OVERRIDE {
|
| + return static_cast<Severity>(resource_id_);
|
| + }
|
| virtual int GetBadgeResourceID() OVERRIDE { return resource_id_; }
|
|
|
| private:
|
| @@ -173,8 +176,9 @@ TEST(GlobalErrorServiceTest, GetMenuItem) {
|
| // Test getting the badge icon resource ID of the first error.
|
| TEST(GlobalErrorServiceTest, GetBadgeID) {
|
| BaseError* error1 = new BaseError;
|
| - BadgeError error2(2);
|
| + BadgeError* error2 = new BadgeError(2);
|
| BadgeError* error3 = new BadgeError(3);
|
| + BadgeError* error4 = new BadgeError(4);
|
|
|
| GlobalErrorService service(NULL);
|
| EXPECT_EQ(0, service.GetFirstBadgeResourceID());
|
| @@ -182,14 +186,19 @@ TEST(GlobalErrorServiceTest, GetBadgeID) {
|
| service.AddGlobalError(error1);
|
| EXPECT_EQ(0, service.GetFirstBadgeResourceID());
|
|
|
| - service.AddGlobalError(&error2);
|
| + service.AddGlobalError(error2);
|
| EXPECT_EQ(2, service.GetFirstBadgeResourceID());
|
|
|
| + service.AddGlobalError(error4);
|
| + EXPECT_EQ(4, service.GetFirstBadgeResourceID());
|
| +
|
| service.AddGlobalError(error3);
|
| - EXPECT_EQ(2, service.GetFirstBadgeResourceID());
|
| + EXPECT_EQ(4, service.GetFirstBadgeResourceID());
|
| +
|
| + // Remove the highest-severity error with a badge.
|
| + service.RemoveGlobalError(error4);
|
| + delete error4;
|
|
|
| - // Remove the first error with a badge.
|
| - service.RemoveGlobalError(&error2);
|
| // Now error3 should be the first error with a badge.
|
| EXPECT_EQ(3, service.GetFirstBadgeResourceID());
|
| }
|
|
|