62 lines · cpp
1//===--- rtsan_test_assertions.cpp - Realtime Sanitizer ---------*- C++ -*-===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8//9// Part of the RealtimeSanitizer runtime library test suite10//11//===----------------------------------------------------------------------===//12 13#include "rtsan_test_utilities.h"14 15#include "rtsan/rtsan_assertions.h"16 17#include "sanitizer_common/sanitizer_stacktrace.h"18 19#include <gmock/gmock.h>20 21using namespace __sanitizer;22using namespace __rtsan;23 24class TestRtsanAssertions : public ::testing::Test {25protected:26 void SetUp() override { __rtsan_ensure_initialized(); }27};28 29static void ExpectViolationAction(Context &context,30 bool expect_violation_callback) {31 ::testing::MockFunction<void(const BufferedStackTrace &stack,32 const DiagnosticsInfo &info)>33 mock_on_violation;34 EXPECT_CALL(mock_on_violation, Call).Times(expect_violation_callback ? 1 : 0);35 DiagnosticsInfo info{};36 ExpectNotRealtime(context, info, mock_on_violation.AsStdFunction());37}38 39TEST_F(TestRtsanAssertions,40 ExpectNotRealtimeDoesNotCallViolationActionIfNotInRealtimeContext) {41 Context context{};42 ASSERT_FALSE(context.InRealtimeContext());43 ExpectViolationAction(context, false);44}45 46TEST_F(TestRtsanAssertions,47 ExpectNotRealtimeCallsViolationActionIfInRealtimeContext) {48 Context context{};49 context.RealtimePush();50 ASSERT_TRUE(context.InRealtimeContext());51 ExpectViolationAction(context, true);52}53 54TEST_F(TestRtsanAssertions,55 ExpectNotRealtimeDoesNotCallViolationActionIfRealtimeButBypassed) {56 Context context{};57 context.RealtimePush();58 context.BypassPush();59 ASSERT_TRUE(context.IsBypassed());60 ExpectViolationAction(context, false);61}62