Upstream Chrome on iOS source code [1/11].

ios/chrome/browser/crash_report/breakpad_helper_unittest.mm

+// Copyright 2013 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 "base/mac/scoped_nsobject.h"
+#import "breakpad/src/client/ios/BreakpadController.h"
+#import "ios/chrome/browser/crash_report/breakpad_helper.h"
+#import "ios/chrome/test/base/scoped_block_swizzler.h"
+#import "ios/chrome/test/ocmock/OCMockObject+BreakpadControllerTesting.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/platform_test.h"
+#import "third_party/ocmock/OCMock/OCMock.h"
+#include "third_party/ocmock/gtest_support.h"
+
+namespace {
+
+const int kCrashReportCount = 3;
+NSString* const kUploadedInRecoveryMode = @"uploaded_in_recovery_mode";
+
+class BreakpadHelperTest : public PlatformTest {
+ public:
+  void SetUp() override {
+    PlatformTest::SetUp();
+
+    mock_breakpad_controller_.reset(
+        [[OCMockObject mockForClass:[BreakpadController class]] retain]);
+
+    // Swizzle +[BreakpadController sharedInstance] to return
+    // |mock_breakpad_controller_| instead of the normal singleton instance.
+    id implementation_block = ^BreakpadController*(id self) {
+      return mock_breakpad_controller_.get();
+    };
+    breakpad_controller_shared_instance_swizzler_.reset(new ScopedBlockSwizzler(
+        [BreakpadController class], @selector(sharedInstance),
+        implementation_block));
+  }
+
+  void TearDown() override {
+    [[mock_breakpad_controller_ stub] stop];
+    breakpad_helper::SetEnabled(false);
+
+    PlatformTest::TearDown();
+  }
+
+ protected:
+  base::scoped_nsobject<id> mock_breakpad_controller_;
+  std::unique_ptr<ScopedBlockSwizzler>
+      breakpad_controller_shared_instance_swizzler_;
+};
+
+TEST_F(BreakpadHelperTest, BrowserState) {
+  // Check that we do not overflow the size of a breakpad record.
+  breakpad_helper::SetCurrentOrientation(3, 7);
+}
+
+TEST_F(BreakpadHelperTest, HorizontalSizeClass) {
+  // Check that we do not overflow the size of a breakpad record.
+  breakpad_helper::SetCurrentHorizontalSizeClass(2);
+}
+
+TEST_F(BreakpadHelperTest, GetCrashReportCount) {
+  [mock_breakpad_controller_ cr_expectGetCrashReportCount:kCrashReportCount];
+
+  // Verify that breakpad_helper::GetCrashReportCount() returns the
+  // crash report count that we arranged to pass to the result block that was
+  // passed to -[BreakpadController getCrashReportCount:].
+  EXPECT_EQ(kCrashReportCount, breakpad_helper::GetCrashReportCount());
+  EXPECT_OCMOCK_VERIFY(mock_breakpad_controller_.get());
+}
+
+TEST_F(BreakpadHelperTest, HasReportToUpload) {
+  [mock_breakpad_controller_ cr_expectGetCrashReportCount:kCrashReportCount];
+  EXPECT_TRUE(breakpad_helper::HasReportToUpload());
+  EXPECT_OCMOCK_VERIFY(mock_breakpad_controller_.get());
+
+  [mock_breakpad_controller_ cr_expectGetCrashReportCount:0];
+  EXPECT_FALSE(breakpad_helper::HasReportToUpload());
+  EXPECT_OCMOCK_VERIFY(mock_breakpad_controller_.get());
+}
+
+TEST_F(BreakpadHelperTest, IsUploadingEnabled) {
+  // Test when crash reporter is disabled.
+  breakpad_helper::SetEnabled(false);
+  EXPECT_FALSE(breakpad_helper::IsUploadingEnabled());
+
+  breakpad_helper::SetUploadingEnabled(false);
+  EXPECT_FALSE(breakpad_helper::IsUploadingEnabled());
+
+  breakpad_helper::SetUploadingEnabled(true);
+  EXPECT_FALSE(breakpad_helper::IsUploadingEnabled());
+
+  // Test when crash reporter is enabled.
+  [[mock_breakpad_controller_ expect] start:NO];
+  breakpad_helper::SetEnabled(true);
+  EXPECT_FALSE(breakpad_helper::IsUploadingEnabled());
+  EXPECT_OCMOCK_VERIFY(mock_breakpad_controller_.get());
+
+  [[mock_breakpad_controller_ expect] setUploadingEnabled:NO];
+  breakpad_helper::SetUploadingEnabled(false);
+  EXPECT_FALSE(breakpad_helper::IsUploadingEnabled());
+  EXPECT_OCMOCK_VERIFY(mock_breakpad_controller_.get());
+
+  [[mock_breakpad_controller_ expect] setUploadingEnabled:YES];
+  breakpad_helper::SetUploadingEnabled(true);
+  EXPECT_TRUE(breakpad_helper::IsUploadingEnabled());
+  EXPECT_OCMOCK_VERIFY(mock_breakpad_controller_.get());
+}
+
+TEST_F(BreakpadHelperTest, StartUploadingReportsInRecoveryMode) {
+  // Test when crash reporter is disabled.
+  breakpad_helper::SetEnabled(false);
+  breakpad_helper::StartUploadingReportsInRecoveryMode();
+
+  // Test when crash reporter is enabled.
+  [[mock_breakpad_controller_ expect] start:NO];
+  breakpad_helper::SetEnabled(true);
+  EXPECT_OCMOCK_VERIFY(mock_breakpad_controller_.get());
+
+  [[mock_breakpad_controller_ expect] stop];
+  [[mock_breakpad_controller_ expect]
+      setParametersToAddAtUploadTime:[OCMArg checkWithBlock:^(id value) {
+        return [value isKindOfClass:[NSDictionary class]] &&
+                       [value[kUploadedInRecoveryMode] isEqualToString:@"yes"]
+                   ? YES
+                   : NO;
+      }]];
+  [[mock_breakpad_controller_ expect] setUploadInterval:1];
+  [[mock_breakpad_controller_ expect] start:NO];
+  [[mock_breakpad_controller_ expect] setUploadingEnabled:YES];
+  breakpad_helper::StartUploadingReportsInRecoveryMode();
+  EXPECT_OCMOCK_VERIFY(mock_breakpad_controller_.get());
+}
+
+TEST_F(BreakpadHelperTest, RestoreDefaultConfiguration) {
+  // Test when crash reporter is disabled.
+  breakpad_helper::SetEnabled(false);
+  breakpad_helper::RestoreDefaultConfiguration();
+
+  // Test when crash reporter is enabled.
+  [[mock_breakpad_controller_ expect] start:NO];
+  breakpad_helper::SetEnabled(true);
+  EXPECT_OCMOCK_VERIFY(mock_breakpad_controller_.get());
+
+  [[mock_breakpad_controller_ expect] stop];
+  [[mock_breakpad_controller_ expect] resetConfiguration];
+  [[mock_breakpad_controller_ expect] start:NO];
+  [[mock_breakpad_controller_ expect] setUploadingEnabled:NO];
+  breakpad_helper::RestoreDefaultConfiguration();
+  EXPECT_OCMOCK_VERIFY(mock_breakpad_controller_.get());
+}
+
+}  // namespace