diff options
Diffstat (limited to 'tests/unit/unittest/notification.h')
-rw-r--r-- | tests/unit/unittest/notification.h | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/tests/unit/unittest/notification.h b/tests/unit/unittest/notification.h new file mode 100644 index 0000000000..a27d9acb76 --- /dev/null +++ b/tests/unit/unittest/notification.h @@ -0,0 +1,56 @@ +/**************************************************************************** +** +** Copyright (C) 2016 The Qt Company Ltd. +** Contact: https://www.qt.io/licensing/ +** +** This file is part of Qt Creator. +** +** Commercial License Usage +** Licensees holding valid commercial Qt licenses may use this file in +** accordance with the commercial license agreement provided with the +** Software or, alternatively, in accordance with the terms contained in +** a written agreement between you and The Qt Company. For licensing terms +** and conditions see https://www.qt.io/terms-conditions. For further +** information use the contact form at https://www.qt.io/contact-us. +** +** GNU General Public License Usage +** Alternatively, this file may be used under the terms of the GNU +** General Public License version 3 as published by the Free Software +** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT +** included in the packaging of this file. Please review the following +** information to ensure the GNU General Public License requirements will +** be met: https://www.gnu.org/licenses/gpl-3.0.html. +** +****************************************************************************/ + +#include "googletest.h" + +#include <condition_variable> +#include <mutex> + +class Notification +{ +public: + void wait(int count = 1) + { + std::unique_lock<std::mutex> lock{m_mutex}; + m_waitCount += count; + if (m_waitCount > 0) + m_condition.wait(lock, [&] { return m_waitCount <= 0; }); + } + + void notify() + { + { + std::unique_lock<std::mutex> lock{m_mutex}; + --m_waitCount; + } + + m_condition.notify_all(); + } + +private: + std::mutex m_mutex; + std::condition_variable m_condition; + int m_waitCount = 0; +}; |