1fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Copyright 2007, Google Inc.
2fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// All rights reserved.
3fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
4fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Redistribution and use in source and binary forms, with or without
5fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// modification, are permitted provided that the following conditions are
6fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// met:
7fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
8fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//     * Redistributions of source code must retain the above copyright
9fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// notice, this list of conditions and the following disclaimer.
10fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//     * Redistributions in binary form must reproduce the above
11fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// copyright notice, this list of conditions and the following disclaimer
12fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// in the documentation and/or other materials provided with the
13fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// distribution.
14fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//     * Neither the name of Google Inc. nor the names of its
15fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// contributors may be used to endorse or promote products derived from
16fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// this software without specific prior written permission.
17fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
18fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
30fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Author: wan@google.com (Zhanyong Wan)
31fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
32fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Tests that SCOPED_TRACE() and various Google Test assertions can be
33fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// used in a large number of threads concurrently.
34fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
35fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <iostream>
36fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <gtest/gtest.h>
37fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
38fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// We must define this macro in order to #include
39fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// gtest-internal-inl.h.  This is how Google Test prevents a user from
40fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// accidentally depending on its internal implementation.
41fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#define GTEST_IMPLEMENTATION_ 1
42fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include "src/gtest-internal-inl.h"
43fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#undef GTEST_IMPLEMENTATION_
44fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
45fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace testing {
46fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace {
47fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
48fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleusing internal::String;
49fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleusing internal::TestPropertyKeyIs;
50fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleusing internal::Vector;
51fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
52fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// How many threads to create?
53fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst int kThreadCount = 50;
54fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
55fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleString IdToKey(int id, const char* suffix) {
56fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Message key;
57fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  key << "key_" << id << "_" << suffix;
58fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return key.GetString();
59fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
60fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
61fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleString IdToString(int id) {
62fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Message id_message;
63fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  id_message << id;
64fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return id_message.GetString();
65fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
66fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
67fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid ExpectKeyAndValueWereRecordedForId(const Vector<TestProperty>& properties,
68fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                        int id,
69fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                        const char* suffix) {
70fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  TestPropertyKeyIs matches_key(IdToKey(id, suffix).c_str());
71fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const TestProperty* property = properties.FindIf(matches_key);
72fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ASSERT_TRUE(property != NULL)
73fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      << "expecting " << suffix << " value for id " << id;
74fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  EXPECT_STREQ(IdToString(id).c_str(), property->value());
75fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
76fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
77fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Calls a large number of Google Test assertions, where exactly one of them
78fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// will fail.
79fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid ManyAsserts(int id) {
80fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ::std::cout << "Thread #" << id << " running...\n";
81fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
82fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  SCOPED_TRACE(Message() << "Thread #" << id);
83fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
84fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < kThreadCount; i++) {
85fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    SCOPED_TRACE(Message() << "Iteration #" << i);
86fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
87fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // A bunch of assertions that should succeed.
88fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    EXPECT_TRUE(true);
89fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    ASSERT_FALSE(false) << "This shouldn't fail.";
90fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    EXPECT_STREQ("a", "a");
91fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    ASSERT_LE(5, 6);
92fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    EXPECT_EQ(i, i) << "This shouldn't fail.";
93fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
94fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // RecordProperty() should interact safely with other threads as well.
95fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // The shared_key forces property updates.
96fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Test::RecordProperty(IdToKey(id, "string").c_str(), IdToString(id).c_str());
97fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Test::RecordProperty(IdToKey(id, "int").c_str(), id);
98fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Test::RecordProperty("shared_key", IdToString(id).c_str());
99fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
100fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // This assertion should fail kThreadCount times per thread.  It
101fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // is for testing whether Google Test can handle failed assertions in a
102fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // multi-threaded context.
103fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    EXPECT_LT(i, 0) << "This should always fail.";
104fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
105fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
106fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
107fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Tests using SCOPED_TRACE() and Google Test assertions in many threads
108fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// concurrently.
109fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleTEST(StressTest, CanUseScopedTraceAndAssertionsInManyThreads) {
110fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // TODO(wan): when Google Test is made thread-safe, run
111fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // ManyAsserts() in many threads here.
112fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
113fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
114fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleTEST(NoFatalFailureTest, ExpectNoFatalFailureIgnoresFailuresInOtherThreads) {
115fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // TODO(mheule@google.com): Test this works correctly when Google
116fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Test is made thread-safe.
117fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
118fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
119fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleTEST(NoFatalFailureTest, AssertNoFatalFailureIgnoresFailuresInOtherThreads) {
120fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // TODO(mheule@google.com): Test this works correctly when Google
121fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Test is made thread-safe.
122fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
123fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
124fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleTEST(FatalFailureTest, ExpectFatalFailureIgnoresFailuresInOtherThreads) {
125fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // TODO(mheule@google.com): Test this works correctly when Google
126fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Test is made thread-safe.
127fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
128fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
129fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleTEST(FatalFailureOnAllThreadsTest, ExpectFatalFailureOnAllThreads) {
130fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // TODO(wan@google.com): Test this works correctly when Google Test
131fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // is made thread-safe.
132fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
133fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
134fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleTEST(NonFatalFailureTest, ExpectNonFatalFailureIgnoresFailuresInOtherThreads) {
135fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // TODO(mheule@google.com): Test this works correctly when Google
136fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Test is made thread-safe.
137fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
138fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
139fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleTEST(NonFatalFailureOnAllThreadsTest, ExpectNonFatalFailureOnAllThreads) {
140fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // TODO(wan@google.com): Test this works correctly when Google Test
141fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // is made thread-safe.
142fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
143fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
144fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}  // namespace
145fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}  // namespace testing
146fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
147fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleint main(int argc, char **argv) {
148fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  testing::InitGoogleTest(&argc, argv);
149fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
150fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return RUN_ALL_TESTS();
151fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
152