1/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2/*
3 *     Copyright 2019 Couchbase, Inc.
4 *
5 *   Licensed under the Apache License, Version 2.0 (the "License");
6 *   you may not use this file except in compliance with the License.
7 *   You may obtain a copy of the License at
8 *
9 *       http://www.apache.org/licenses/LICENSE-2.0
10 *
11 *   Unless required by applicable law or agreed to in writing, software
12 *   distributed under the License is distributed on an "AS IS" BASIS,
13 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 *   See the License for the specific language governing permissions and
15 *   limitations under the License.
16 */
17#pragma once
18
19#include "executorpool.h"
20
21/*
22 * Wrapper to ExecutorPool, exposes protected members for testing.
23 */
24class MockExecutorPool : public ExecutorPool {
25public:
26    MockExecutorPool(size_t numTaskSets)
27        : ExecutorPool(4 /*maxThreads*/,
28                       numTaskSets,
29                       ThreadPoolConfig::ThreadCount(1) /*maxReaders*/,
30                       ThreadPoolConfig::ThreadCount(1) /*maxWriters*/,
31                       1 /*maxAuxIO*/,
32                       1 /*maxNonIO*/) {
33    }
34
35    /**
36     * Replaces the ExecutorPool instance with a MockExecutorPool one.
37     * Useful for checking the internal state of ExecutorPool.
38     */
39    static void replaceExecutorPoolWithMock();
40
41    /**
42     * @param queueType
43     * @param taskName
44     * @return true if the given task (of the given type) is scheduled,
45     *     false otherwise
46     */
47    bool isTaskScheduled(const task_type_t queueType,
48                         const std::string& taskName);
49};
50