chromium/base/mac/scoped_sending_event_unittest.mm

// Copyright 2011 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#import "base/mac/scoped_sending_event.h"

#import <Foundation/Foundation.h>

#include "testing/gtest/include/gtest/gtest.h"

#ifdef LEAK_SANITIZER
#include <sanitizer/lsan_interface.h>
#endif

@interface ScopedSendingEventTestCrApp : NSApplication <CrAppControlProtocol> {
 @private
  BOOL _handlingSendEvent;
}
@property(nonatomic, assign, getter=isHandlingSendEvent) BOOL handlingSendEvent;
@end

@implementation ScopedSendingEventTestCrApp
@synthesize handlingSendEvent = _handlingSendEvent;
@end

namespace {

class ScopedSendingEventTest : public testing::Test {
 public:
  ScopedSendingEventTest() {
#ifdef LEAK_SANITIZER
    // NSApplication's `init` creates a helper object and writes it to an
    // AppKit-owned static unconditionally. This is not cleaned up on
    // NSApplication dealloc.
    // When we create a new NSApplication, as we do when we run multiple
    // tests in this suite, a new object is created and stomps on the old
    // static.
    // This needs a scoped disabler instead of just ignoring the app
    // object since the leak is a side-effect of object creation.
    __lsan::ScopedDisabler disable;
#endif
    app_ = [[ScopedSendingEventTestCrApp alloc] init];
    NSApp = app_;
  }
  ~ScopedSendingEventTest() override { NSApp = nil; }

 private:
  ScopedSendingEventTestCrApp* __strong app_;
};

// Sets the flag within scope, resets when leaving scope.
TEST_F(ScopedSendingEventTest, SetHandlingSendEvent) {
  id<CrAppProtocol> app = NSApp;
  EXPECT_FALSE([app isHandlingSendEvent]);
  {
    base::mac::ScopedSendingEvent is_handling_send_event;
    EXPECT_TRUE([app isHandlingSendEvent]);
  }
  EXPECT_FALSE([app isHandlingSendEvent]);
}

// Nested call restores previous value rather than resetting flag.
TEST_F(ScopedSendingEventTest, NestedSetHandlingSendEvent) {
  id<CrAppProtocol> app = NSApp;
  EXPECT_FALSE([app isHandlingSendEvent]);
  {
    base::mac::ScopedSendingEvent is_handling_send_event;
    EXPECT_TRUE([app isHandlingSendEvent]);
    {
      base::mac::ScopedSendingEvent nested_is_handling_send_event;
      EXPECT_TRUE([app isHandlingSendEvent]);
    }
    EXPECT_TRUE([app isHandlingSendEvent]);
  }
  EXPECT_FALSE([app isHandlingSendEvent]);
}

}  // namespace