This change alters where we unwind the stack. We now do it in ExpectNotRealtime, and pass in the DiagnosticInfo and Stack to OnViolation.
62 lines
1.9 KiB
C++
62 lines
1.9 KiB
C++
//===--- rtsan_test_assertions.cpp - Realtime Sanitizer ---------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Part of the RealtimeSanitizer runtime library test suite
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "rtsan_test_utilities.h"
|
|
|
|
#include "rtsan/rtsan_assertions.h"
|
|
|
|
#include "sanitizer_common/sanitizer_stacktrace.h"
|
|
|
|
#include <gmock/gmock.h>
|
|
|
|
using namespace __sanitizer;
|
|
using namespace __rtsan;
|
|
|
|
class TestRtsanAssertions : public ::testing::Test {
|
|
protected:
|
|
void SetUp() override { __rtsan_ensure_initialized(); }
|
|
};
|
|
|
|
static void ExpectViolationAction(Context &context,
|
|
bool expect_violation_callback) {
|
|
::testing::MockFunction<void(const BufferedStackTrace &stack,
|
|
const DiagnosticsInfo &info)>
|
|
mock_on_violation;
|
|
EXPECT_CALL(mock_on_violation, Call).Times(expect_violation_callback ? 1 : 0);
|
|
DiagnosticsInfo info{};
|
|
ExpectNotRealtime(context, info, mock_on_violation.AsStdFunction());
|
|
}
|
|
|
|
TEST_F(TestRtsanAssertions,
|
|
ExpectNotRealtimeDoesNotCallViolationActionIfNotInRealtimeContext) {
|
|
Context context{};
|
|
ASSERT_FALSE(context.InRealtimeContext());
|
|
ExpectViolationAction(context, false);
|
|
}
|
|
|
|
TEST_F(TestRtsanAssertions,
|
|
ExpectNotRealtimeCallsViolationActionIfInRealtimeContext) {
|
|
Context context{};
|
|
context.RealtimePush();
|
|
ASSERT_TRUE(context.InRealtimeContext());
|
|
ExpectViolationAction(context, true);
|
|
}
|
|
|
|
TEST_F(TestRtsanAssertions,
|
|
ExpectNotRealtimeDoesNotCallViolationActionIfRealtimeButBypassed) {
|
|
Context context{};
|
|
context.RealtimePush();
|
|
context.BypassPush();
|
|
ASSERT_TRUE(context.IsBypassed());
|
|
ExpectViolationAction(context, false);
|
|
}
|