#pragma once

#include <balancer/kernel/balancing/backend.h>
#include <balancer/kernel/balancing/per_worker_backend.h>

#include <balancer/kernel/module/module_face.h>

#include <library/cpp/testing/gmock_in_unittest/gmock.h>

namespace NSrvKernel::NTesting {

class TPerWorkerBackendMock : public TPerWorkerBaseBackend {
public:
    using TPerWorkerBaseBackend::TPerWorkerBaseBackend;

    MOCK_METHOD(void, DoOnCompleteRequest, (const TDuration&), (override));
    MOCK_METHOD(void, DoOnFailRequest, (const TError&, const TDuration&), (override));
};

}  // namespace NSrvKernel::NTesting
