xref: /base/msdp/device_status/libs/include/devicestatus_msdp_mock.h

/*
 * Copyright (c) 2022-2023 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef DEVICESTATUS_MSDP_MOCK_H
#define DEVICESTATUS_MSDP_MOCK_H

#include <atomic>
#include <map>
#include <memory>
#include <mutex>
#include <thread>
#include <vector>

#include "devicestatus_data_parse.h"
#include "stationary_data.h"
#include "devicestatus_msdp_interface.h"

namespace OHOS {
namespace Msdp {
namespace DeviceStatus {
class DeviceStatusMsdpMock final : public IMsdp {
public:
    enum EventType {
        EVENT_UEVENT_FD,
        EVENT_TIMER_FD
    };

    DeviceStatusMsdpMock();
    ~DeviceStatusMsdpMock();

    bool Init();
    void InitMockStore();
    int32_t SetTimerInterval(int32_t interval);
    void CloseTimer();
    void InitTimer();
    void TimerCallback();
    int32_t RegisterTimerCallback(int32_t fd, const EventType et);
    void StartThread();
    void LoopingThreadEntry();
    ErrCode Enable(Type type) override;
    ErrCode Disable(Type type) override;
    ErrCode DisableCount(Type type) override;
    ErrCode RegisterCallback(std::shared_ptr<IMsdp::MsdpAlgoCallback> callback) override;
    ErrCode UnregisterCallback() override;
    ErrCode NotifyMsdpImpl(const Data &data);
    int32_t GetDeviceStatusData();
    std::shared_ptr<MsdpAlgoCallback> GetCallbackImpl()
    {
        std::unique_lock lock(mutex_);
        return callback_;
    }

private:
    using Callback = std::function<void(DeviceStatusMsdpMock*)>;
    std::shared_ptr<MsdpAlgoCallback> callback_ { nullptr };
    std::map<int32_t, Callback> callbacks_;
    std::unique_ptr<DeviceStatusDataParse> dataParse_ { nullptr };
    int32_t timerInterval_ { -1 };
    int32_t timerFd_ { -1 };
    int32_t epFd_ { -1 };
    std::mutex mutex_;
    std::vector<Type> enabledType_;
    std::atomic<bool> alive_ { false };
    std::thread thread_;
};
} // namespace DeviceStatus
} // namespace Msdp
} // namespace OHOS
#endif // DEVICESTATUS_MSDP_MOCK_H