/*
 * Copyright (c) 2022 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.
 */

#include "hdi_service_test.h"
#include "battery_log.h"
#include "battery_thread_test.h"
#include "hdf_base.h"
#include "power_supply_provider.h"
#include <chrono>
#include <condition_variable>
#include <csignal>
#include <cstring>
#include <dirent.h>
#include <fcntl.h>
#include <fstream>
#include <iostream>
#include <memory>
#include <mutex>
#include <streambuf>
#include <sys/stat.h>
#include <thread>
#include <vector>

using namespace testing::ext;
using namespace OHOS;
using namespace OHOS::HDI::Battery;
using namespace OHOS::HDI::Battery::V2_0;
using namespace std;

namespace HdiServiceTest {
const std::string SYSTEM_BATTERY_PATH = "/sys/class/power_supply";
const std::string MOCK_BATTERY_PATH = "/data/service/el0/battery/";
static std::vector<std::string> g_filenodeName;
static std::map<std::string, std::string> g_nodeInfo;
const int STR_TO_LONG_LEN = 10;
const int NUM_ZERO = 0;
constexpr int32_t ERROR = -1;
constexpr int32_t MAX_BUFF_SIZE = 128;
constexpr int32_t MAX_SYSFS_SIZE = 64;
std::unique_ptr<PowerSupplyProvider> giver_ = nullptr;

void HdiServiceTest::SetUpTestCase(void)
{
    giver_ = std::make_unique<PowerSupplyProvider>();
    if (giver_ == nullptr) {
        BATTERY_HILOGI(LABEL_TEST, "Failed to get PowerSupplyProvider");
    }
}

void HdiServiceTest::TearDownTestCase(void) {}

void HdiServiceTest::SetUp(void) {}

void HdiServiceTest::TearDown(void) {}

struct StringEnumMap {
    const char* str;
    int32_t enumVal;
};

std::string CreateFile(std::string path, std::string content)
{
    std::ofstream stream(path.c_str());
    if (!stream.is_open()) {
        BATTERY_HILOGI(LABEL_TEST, "Cannot create file");
        return nullptr;
    }
    stream << content.c_str() << std::endl;
    stream.close();
    return path;
}

static void CheckSubfolderNode(const std::string& path)
{
    DIR* dir = nullptr;
    struct dirent* entry = nullptr;
    std::string batteryPath = SYSTEM_BATTERY_PATH + "/" + path;

    dir = opendir(batteryPath.c_str());
    if (dir == nullptr) {
        BATTERY_HILOGI(LABEL_TEST, "subfolder file is not exist.");
        return;
    }

    while (true) {
        entry = readdir(dir);
        if (entry == nullptr) {
            break;
        }

        if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0) {
            continue;
        }

        if (entry->d_type == DT_DIR || entry->d_type == DT_LNK) {
            continue;
        }

        if ((strcmp(entry->d_name, "type") == 0) && (g_nodeInfo["type"] == "") &&
            (strcasecmp(path.c_str(), "battery") != 0)) {
            g_nodeInfo["type"] = path;
        }

        for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
            if ((strcmp(entry->d_name, iter->first.c_str()) == 0) && (g_nodeInfo[iter->first] == "")) {
                g_nodeInfo[iter->first] = path;
            }
        }
    }
    closedir(dir);
}

static void TraversalBaseNode()
{
    g_nodeInfo.insert(std::make_pair("type", ""));
    g_nodeInfo.insert(std::make_pair("online", ""));
    g_nodeInfo.insert(std::make_pair("current_max", ""));
    g_nodeInfo.insert(std::make_pair("voltage_max", ""));
    g_nodeInfo.insert(std::make_pair("capacity", ""));
    g_nodeInfo.insert(std::make_pair("voltage_now", ""));
    g_nodeInfo.insert(std::make_pair("temp", ""));
    g_nodeInfo.insert(std::make_pair("health", ""));
    g_nodeInfo.insert(std::make_pair("status", ""));
    g_nodeInfo.insert(std::make_pair("present", ""));
    g_nodeInfo.insert(std::make_pair("charge_counter", ""));
    g_nodeInfo.insert(std::make_pair("technology", ""));
    g_nodeInfo.insert(std::make_pair("charge_full", ""));
    g_nodeInfo.insert(std::make_pair("current_avg", ""));
    g_nodeInfo.insert(std::make_pair("current_now", ""));
    g_nodeInfo.insert(std::make_pair("charge_now", ""));

    auto iter = g_filenodeName.begin();
    while (iter != g_filenodeName.end()) {
        if (*iter == "battery") {
            CheckSubfolderNode(*iter);
            iter = g_filenodeName.erase(iter);
        } else {
            iter++;
        }
    }

    iter = g_filenodeName.begin();
    while (iter != g_filenodeName.end()) {
        if (*iter == "Battery") {
            CheckSubfolderNode(*iter);
            iter = g_filenodeName.erase(iter);
        } else {
            iter++;
        }
    }

    for (auto it = g_filenodeName.begin(); it != g_filenodeName.end(); ++it) {
        CheckSubfolderNode(*it);
    }
}

static int32_t InitBaseSysfs(void)
{
    DIR* dir = nullptr;
    struct dirent* entry = nullptr;
    int32_t index = 0;

    dir = opendir(SYSTEM_BATTERY_PATH.c_str());
    if (dir == nullptr) {
        BATTERY_HILOGE(LABEL_TEST, "cannot open POWER_SUPPLY_BASE_PATH");
        return HDF_ERR_IO;
    }

    while (true) {
        entry = readdir(dir);
        if (entry == nullptr) {
            break;
        }

        if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0) {
            continue;
        }

        if (entry->d_type == DT_DIR || entry->d_type == DT_LNK) {
            BATTERY_HILOGI(LABEL_TEST, "init sysfs info of %{public}s", entry->d_name);
            if (index >= MAX_SYSFS_SIZE) {
                BATTERY_HILOGE(LABEL_TEST, "too many plugged types");
                break;
            }
            g_filenodeName.emplace_back(entry->d_name);
            index++;
        }
    }

    TraversalBaseNode();
    BATTERY_HILOGI(LABEL_TEST, "index is %{public}d", index);
    closedir(dir);

    return HDF_SUCCESS;
}

static int32_t ReadTemperatureSysfs()
{
    int strlen = 10;
    char buf[128] = {0};
    int32_t readSize;
    InitBaseSysfs();
    std::string tempNode = "battery";
    for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
        if (iter->first == "temp") {
            tempNode = iter->second;
            break;
        }
    }
    std::string sysBattTemPath = SYSTEM_BATTERY_PATH + "/" + tempNode + "/" + "temp";

    int fd = open(sysBattTemPath.c_str(), O_RDONLY);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open TemperatureSysfs");
        return HDF_FAILURE;
    }

    readSize = read(fd, buf, sizeof(buf) - 1);
    if (readSize < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read TemperatureSysfs");
        close(fd);
        return HDF_FAILURE;
    }

    buf[readSize] = '\0';
    int32_t battTemperature = strtol(buf, nullptr, strlen);
    if (battTemperature < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "read system file temperature is %{public}d", battTemperature);
    }
    BATTERY_HILOGE(LABEL_TEST, "read system file temperature is %{public}d", battTemperature);
    close(fd);
    return battTemperature;
}

static int32_t ReadVoltageSysfs()
{
    int strlen = 10;
    char buf[128] = {0};
    int32_t readSize;
    std::string voltageNode = "battery";
    for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
        if (iter->first == "voltage_now") {
            voltageNode = iter->second;
            break;
        }
    }
    std::string sysBattVolPath = SYSTEM_BATTERY_PATH + "/" + voltageNode + "/" + "voltage_now";

    int fd = open(sysBattVolPath.c_str(), O_RDONLY);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open VoltageSysfs");
        return HDF_FAILURE;
    }

    readSize = read(fd, buf, sizeof(buf) - 1);
    if (readSize < HDF_SUCCESS) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read VoltageSysfs");
        close(fd);
        return HDF_FAILURE;
    }

    buf[readSize] = '\0';
    int32_t battVoltage = strtol(buf, nullptr, strlen);
    if (battVoltage < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "read system file voltage is %{public}d", battVoltage);
    }
    BATTERY_HILOGE(LABEL_TEST, "read system file voltage is %{public}d", battVoltage);
    close(fd);
    return battVoltage;
}

static int32_t ReadCapacitySysfs()
{
    int strlen = 10;
    char buf[128] = {0};
    int32_t readSize;
    std::string capacityNode = "battery";
    for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
        if (iter->first == "capacity") {
            capacityNode = iter->second;
            break;
        }
    }
    std::string sysBattCapPath = SYSTEM_BATTERY_PATH + "/" + capacityNode + "/" + "capacity";

    int fd = open(sysBattCapPath.c_str(), O_RDONLY);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open CapacitySysfs");
        return HDF_FAILURE;
    }

    readSize = read(fd, buf, sizeof(buf) - 1);
    if (readSize < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read CapacitySysfs");
        close(fd);
        return HDF_FAILURE;
    }

    buf[readSize] = '\0';
    int32_t battCapacity = strtol(buf, nullptr, strlen);
    if (battCapacity < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "read system file capacity is %{public}d", battCapacity);
    }
    BATTERY_HILOGE(LABEL_TEST, "read system file capacity is %{public}d", battCapacity);
    close(fd);
    return battCapacity;
}

static int32_t ReadTotalEnergySysfs()
{
    int strlen = 10;
    char buf[128] = {0};
    int32_t readSize;
    InitBaseSysfs();
    std::string totalEnergyNode = "battery";
    for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
        if (iter->first == "charge_full") {
            totalEnergyNode = iter->second;
            break;
        }
    }
    std::string sysBattTotalEnergyPath = SYSTEM_BATTERY_PATH + "/" + totalEnergyNode + "/" + "charge_full";

    int fd = open(sysBattTotalEnergyPath.c_str(), O_RDONLY);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open TotalEnergySysfs");
        return HDF_FAILURE;
    }

    readSize = read(fd, buf, sizeof(buf) - 1);
    if (readSize < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read TotalEnergySysfs");
        close(fd);
        return HDF_FAILURE;
    }

    buf[readSize] = '\0';
    int32_t totalEnergy = strtol(buf, nullptr, strlen);
    if (totalEnergy < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "read system file totalEnergy is %{public}d", totalEnergy);
    }
    BATTERY_HILOGE(LABEL_TEST, "read system file totalEnergy is %{public}d", totalEnergy);
    close(fd);
    return totalEnergy;
}

static int32_t ReadCurrentAverageSysfs()
{
    int strlen = 10;
    char buf[128] = {0};
    int32_t readSize;
    InitBaseSysfs();
    std::string currentAvgNode = "battery";
    for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
        if (iter->first == "current_avg") {
            currentAvgNode = iter->second;
            break;
        }
    }
    std::string sysBattCurrentAvgPath = SYSTEM_BATTERY_PATH + "/" + currentAvgNode + "/" + "current_avg";

    int fd = open(sysBattCurrentAvgPath.c_str(), O_RDONLY);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open CurrentAverageSysfs");
        return HDF_FAILURE;
    }

    readSize = read(fd, buf, sizeof(buf) - 1);
    if (readSize < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read CurrentAverageSysfs");
        close(fd);
        return HDF_FAILURE;
    }

    buf[readSize] = '\0';
    int32_t currentAvg = strtol(buf, nullptr, strlen);
    if (currentAvg < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "read system file currentAvg is %{public}d", currentAvg);
    }
    BATTERY_HILOGE(LABEL_TEST, "read system file currentAvg is %{public}d", currentAvg);
    close(fd);
    return currentAvg;
}

static int32_t ReadCurrentNowSysfs()
{
    int strlen = 10;
    char buf[128] = {0};
    int32_t readSize;
    InitBaseSysfs();
    std::string currentNowNode = "battery";
    for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
        if (iter->first == "current_now") {
            currentNowNode = iter->second;
            break;
        }
    }
    std::string sysBattCurrentNowPath = SYSTEM_BATTERY_PATH + "/" + currentNowNode + "/" + "current_now";

    int fd = open(sysBattCurrentNowPath.c_str(), O_RDONLY);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open CurrentNowSysfs");
        return HDF_FAILURE;
    }

    readSize = read(fd, buf, sizeof(buf) - 1);
    if (readSize < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read CurrentNowSysfs");
        close(fd);
        return HDF_FAILURE;
    }

    buf[readSize] = '\0';
    int32_t currentNow = strtol(buf, nullptr, strlen);
    if (currentNow < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "read system file currentNow is %{public}d", currentNow);
    }
    BATTERY_HILOGE(LABEL_TEST, "read system file currentNow is %{public}d", currentNow);
    close(fd);
    return currentNow;
}

static int32_t ReadRemainEnergySysfs()
{
    int strlen = 10;
    char buf[128] = {0};
    int32_t readSize;
    InitBaseSysfs();
    std::string chargeNowNode = "battery";
    for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
        if (iter->first == "charge_now") {
            chargeNowNode = iter->second;
            break;
        }
    }
    std::string sysBattChargeNowPath = SYSTEM_BATTERY_PATH + "/" + chargeNowNode + "/" + "charge_now";

    int fd = open(sysBattChargeNowPath.c_str(), O_RDONLY);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open RemainEnergySysfs");
        return HDF_FAILURE;
    }

    readSize = read(fd, buf, sizeof(buf) - 1);
    if (readSize < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read RemainEnergySysfs");
        close(fd);
        return HDF_FAILURE;
    }

    buf[readSize] = '\0';
    int32_t chargeNow = strtol(buf, nullptr, strlen);
    if (chargeNow < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "read system file chargeNow is %{public}d", chargeNow);
    }
    BATTERY_HILOGE(LABEL_TEST, "read system file chargeNow is %{public}d", chargeNow);
    close(fd);
    return chargeNow;
}

static void Trim(char* str)
{
    if (str == nullptr) {
        return;
    }
    str[strcspn(str, "\n")] = 0;
}

static int32_t HealthStateEnumConverter(const char* str)
{
    struct StringEnumMap healthStateEnumMap[] = {
        {"Good",                PowerSupplyProvider::BATTERY_HEALTH_GOOD       },
        {"Cold",                PowerSupplyProvider::BATTERY_HEALTH_COLD       },
        {"Warm",                PowerSupplyProvider::BATTERY_HEALTH_GOOD       }, // JEITA specification
        {"Cool",                PowerSupplyProvider::BATTERY_HEALTH_GOOD       }, // JEITA specification
        {"Hot",                 PowerSupplyProvider::BATTERY_HEALTH_OVERHEAT   }, // JEITA specification
        {"Overheat",            PowerSupplyProvider::BATTERY_HEALTH_OVERHEAT   },
        {"Over voltage",        PowerSupplyProvider::BATTERY_HEALTH_OVERVOLTAGE},
        {"Dead",                PowerSupplyProvider::BATTERY_HEALTH_DEAD       },
        {"Unknown",             PowerSupplyProvider::BATTERY_HEALTH_UNKNOWN    },
        {"Unspecified failure", PowerSupplyProvider::BATTERY_HEALTH_UNKNOWN    },
        {NULL,                  PowerSupplyProvider::BATTERY_HEALTH_UNKNOWN    },
    };

    for (int i = 0; healthStateEnumMap[i].str; ++i) {
        if (strcmp(str, healthStateEnumMap[i].str) == 0) {
            return healthStateEnumMap[i].enumVal;
        }
    }

    return PowerSupplyProvider::BATTERY_HEALTH_UNKNOWN;
}

static int32_t ReadHealthStateSysfs()
{
    char buf[128] = {0};
    int32_t readSize;
    std::string healthNode = "battery";
    for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
        if (iter->first == "health") {
            healthNode = iter->second;
            break;
        }
    }
    std::string sysHealthStatePath = SYSTEM_BATTERY_PATH + "/" + healthNode + "/" + "health";

    int fd = open(sysHealthStatePath.c_str(), O_RDONLY);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open HealthStateSysfs");
        return HDF_FAILURE;
    }

    readSize = read(fd, buf, sizeof(buf) - 1);
    if (readSize < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read HealthStateSysfs");
        close(fd);
        return HDF_FAILURE;
    }

    Trim(buf);

    int32_t battHealthState = HealthStateEnumConverter(buf);
    BATTERY_HILOGE(LABEL_TEST, "read system file healthState is %{public}d", battHealthState);
    close(fd);
    return battHealthState;
}

static int32_t PluggedTypeEnumConverter(const char* str)
{
    struct StringEnumMap pluggedTypeEnumMap[] = {
        {"USB",        PowerSupplyProvider::PLUGGED_TYPE_USB     },
        {"USB_PD_DRP", PowerSupplyProvider::PLUGGED_TYPE_USB     },
        {"Wireless",   PowerSupplyProvider::PLUGGED_TYPE_WIRELESS},
        {"Mains",      PowerSupplyProvider::PLUGGED_TYPE_AC      },
        {"UPS",        PowerSupplyProvider::PLUGGED_TYPE_AC      },
        {"USB_ACA",    PowerSupplyProvider::PLUGGED_TYPE_AC      },
        {"USB_C",      PowerSupplyProvider::PLUGGED_TYPE_AC      },
        {"USB_CDP",    PowerSupplyProvider::PLUGGED_TYPE_AC      },
        {"USB_DCP",    PowerSupplyProvider::PLUGGED_TYPE_AC      },
        {"USB_HVDCP",  PowerSupplyProvider::PLUGGED_TYPE_AC      },
        {"USB_PD",     PowerSupplyProvider::PLUGGED_TYPE_AC      },
        {"Unknown",    PowerSupplyProvider::PLUGGED_TYPE_BUTT    },
        {NULL,         PowerSupplyProvider::PLUGGED_TYPE_BUTT    },
    };

    for (int i = 0; pluggedTypeEnumMap[i].str; ++i) {
        if (strcmp(str, pluggedTypeEnumMap[i].str) == 0) {
            return pluggedTypeEnumMap[i].enumVal;
        }
    }

    return PowerSupplyProvider::PLUGGED_TYPE_BUTT;
}

int32_t ReadSysfsFile(const char* path, char* buf, size_t size)
{
    int fd = open(path, O_RDONLY, S_IRUSR | S_IRGRP | S_IROTH);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open file");
        return HDF_ERR_IO;
    }

    int32_t readSize = read(fd, buf, size - 1);
    if (readSize < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read file");
        close(fd);
        return HDF_ERR_IO;
    }

    buf[readSize] = '\0';
    Trim(buf);
    close(fd);

    return HDF_SUCCESS;
}

static int32_t ReadPluggedTypeSysfs()
{
    std::string node = "USB";
    int32_t online = ERROR;
    char buf[MAX_BUFF_SIZE] = {0};

    auto iter = g_filenodeName.begin();
    while (iter != g_filenodeName.end()) {
        if (strcasecmp(iter->c_str(), "battery") == 0) {
            continue;
        }
        std::string onlinePath = SYSTEM_BATTERY_PATH + "/" + *iter + "/" + "online";
        if (ReadSysfsFile(onlinePath.c_str(), buf, MAX_BUFF_SIZE) != HDF_SUCCESS) {
            BATTERY_HILOGW(LABEL_TEST, "read online path failed in loop");
        }
        online = strtol(buf, nullptr, STR_TO_LONG_LEN);
        if (online) {
            node = *iter;
            break;
        }
        iter++;
    }
    if (online == ERROR) {
        return ERROR;
    }
    std::string typePath = SYSTEM_BATTERY_PATH + "/" + node + "/" + "type";
    if (ReadSysfsFile(typePath.c_str(), buf, MAX_BUFF_SIZE) != HDF_SUCCESS) {
        BATTERY_HILOGI(LABEL_TEST, "read type path failed");
        return ERROR;
    }
    Trim(buf);
    return PluggedTypeEnumConverter(buf);
}

int32_t ChargeStateEnumConverter(const char* str)
{
    struct StringEnumMap chargeStateEnumMap[] = {
        {"Discharging",  PowerSupplyProvider::CHARGE_STATE_NONE    },
        {"Charging",     PowerSupplyProvider::CHARGE_STATE_ENABLE  },
        {"Full",         PowerSupplyProvider::CHARGE_STATE_FULL    },
        {"Not charging", PowerSupplyProvider::CHARGE_STATE_DISABLE },
        {"Unknown",      PowerSupplyProvider::CHARGE_STATE_RESERVED},
        {NULL,           PowerSupplyProvider::CHARGE_STATE_RESERVED},
    };

    for (int i = 0; chargeStateEnumMap[i].str; ++i) {
        if (strcmp(str, chargeStateEnumMap[i].str) == 0) {
            return chargeStateEnumMap[i].enumVal;
        }
    }
    return PowerSupplyProvider::CHARGE_STATE_RESERVED;
}

static int32_t ReadChargeStateSysfs()
{
    char buf[128] = {0};
    int32_t readSize;
    std::string statusNode = "battery";
    for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
        if (iter->first == "status") {
            statusNode = iter->second;
            break;
        }
    }
    std::string sysChargeStatePath = SYSTEM_BATTERY_PATH + "/" + statusNode + "/" + "status";

    int fd = open(sysChargeStatePath.c_str(), O_RDONLY);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open ChargeStateSysfs");
        return HDF_FAILURE;
    }

    readSize = read(fd, buf, sizeof(buf) - 1);
    if (readSize < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read ChargeStateSysfs");
        close(fd);
        return HDF_FAILURE;
    }

    Trim(buf);
    int32_t battChargeState = ChargeStateEnumConverter(buf);
    BATTERY_HILOGE(LABEL_TEST, "read system file chargeState is %{public}d", battChargeState);
    close(fd);

    return battChargeState;
}

static int32_t ReadChargeCounterSysfs()
{
    int strlen = 10;
    char buf[128] = {0};
    int32_t readSize;
    std::string counterNode = "battery";
    for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
        if (iter->first == "charge_counter") {
            counterNode = iter->second;
            break;
        }
    }
    std::string sysChargeCounterPath = SYSTEM_BATTERY_PATH + "/" + counterNode + "/" + "charge_counter";

    int fd = open(sysChargeCounterPath.c_str(), O_RDONLY);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open ChargeCounterSysfs");
        return HDF_FAILURE;
    }

    readSize = read(fd, buf, sizeof(buf) - 1);
    if (readSize < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read ChargeCounterSysfs");
        close(fd);
        return HDF_FAILURE;
    }

    buf[readSize] = '\0';
    int32_t battChargeCounter = strtol(buf, nullptr, strlen);
    if (battChargeCounter < 0) {
        BATTERY_HILOGE(LABEL_TEST, "read system file chargeState is %{public}d", battChargeCounter);
    }
    BATTERY_HILOGE(LABEL_TEST, "read system file chargeState is %{public}d", battChargeCounter);
    close(fd);

    return battChargeCounter;
}

static int32_t ReadPresentSysfs()
{
    int strlen = 10;
    char buf[128] = {0};
    int32_t readSize;
    std::string presentNode = "battery";
    for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
        if (iter->first == "present") {
            presentNode = iter->second;
            break;
        }
    }
    std::string sysPresentPath = SYSTEM_BATTERY_PATH + "/" + presentNode + "/" + "present";

    int fd = open(sysPresentPath.c_str(), O_RDONLY);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open PresentSysfs");
        return HDF_FAILURE;
    }

    readSize = read(fd, buf, sizeof(buf) - 1);
    if (readSize < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read PresentSysfs");
        close(fd);
        return HDF_FAILURE;
    }

    buf[readSize] = '\0';
    int32_t battPresent = strtol(buf, nullptr, strlen);
    if (battPresent < 0) {
        BATTERY_HILOGE(LABEL_TEST, "read system file chargeState is %{public}d", battPresent);
    }
    BATTERY_HILOGE(LABEL_TEST, "read system file chargeState is %{public}d", battPresent);
    close(fd);
    return battPresent;
}

static std::string ReadTechnologySysfs(std::string& battTechnology)
{
    char buf[128] = {0};
    int32_t readSize;
    std::string technologyNode = "battery";
    for (auto iter = g_nodeInfo.begin(); iter != g_nodeInfo.end(); ++iter) {
        if (iter->first == "technology") {
            technologyNode = iter->second;
            break;
        }
    }
    std::string sysTechnologyPath = SYSTEM_BATTERY_PATH + "/" + technologyNode + "/" + "technology";

    int fd = open(sysTechnologyPath.c_str(), O_RDONLY);
    if (fd < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to open TechnologySysfs");
        return "";
    }

    readSize = read(fd, buf, sizeof(buf) - 1);
    if (readSize < NUM_ZERO) {
        BATTERY_HILOGE(LABEL_TEST, "failed to read TechnologySysfs");
        close(fd);
        return "";
    }
    buf[readSize] = '\0';
    Trim(buf);

    battTechnology.assign(buf, strlen(buf));
    BATTERY_HILOGE(LABEL_TEST, "read system file technology is %{public}s.", battTechnology.c_str());
    close(fd);
    return battTechnology;
}

static bool IsNotMock()
{
    bool rootExist = access(SYSTEM_BATTERY_PATH.c_str(), F_OK) == 0;
    bool lowerExist = access((SYSTEM_BATTERY_PATH + "/battery").c_str(), F_OK) == 0;
    bool upperExist = access((SYSTEM_BATTERY_PATH + "/Battery").c_str(), F_OK) == 0;
    return rootExist && (lowerExist || upperExist);
}

/**
 * @tc.name: ProviderIsNotNull
 * @tc.desc: Test functions of PowerSupplyProvider
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, ProviderIsNotNull, TestSize.Level1)
{
    ASSERT_TRUE(giver_ != nullptr);
    if (!IsNotMock()) {
        giver_->SetSysFilePath(MOCK_BATTERY_PATH);
        BATTERY_HILOGI(LABEL_TEST, "Is mock test");
    }
    giver_->InitPowerSupplySysfs();
}

/**
 * @tc.name: HdiService001
 * @tc.desc: Test functions of ParseTemperature
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService001, TestSize.Level1)
{
    int32_t temperature = 0;
    if (IsNotMock()) {
        giver_->ParseTemperature(&temperature);
        int32_t sysfsTemp = ReadTemperatureSysfs();
        BATTERY_HILOGI(
            LABEL_TEST, "Not Mock HdiService001::temperature=%{public}d, t=%{public}d", temperature, sysfsTemp);
        ASSERT_TRUE(temperature == sysfsTemp);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "battery/temp", "567");
        giver_->ParseTemperature(&temperature);
        BATTERY_HILOGI(LABEL_TEST, "HdiService001::temperature=%{public}d.", temperature);
        ASSERT_TRUE(temperature == 567);
    }
}

/**
 * @tc.name: HdiService002
 * @tc.desc: Test functions of ParseVoltage
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService002, TestSize.Level1)
{
    int32_t voltage = 0;
    if (IsNotMock()) {
        giver_->ParseVoltage(&voltage);
        int32_t sysfsVoltage = ReadVoltageSysfs();
        BATTERY_HILOGI(LABEL_TEST, "Not Mock HdiService002::voltage=%{public}d, v=%{public}d", voltage, sysfsVoltage);
        ASSERT_TRUE(voltage == sysfsVoltage);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "battery/voltage_avg", "4123456");
        CreateFile(MOCK_BATTERY_PATH + "battery/voltage_now", "4123456");
        giver_->ParseVoltage(&voltage);
        BATTERY_HILOGI(LABEL_TEST, "Not Mock HdiService002::voltage=%{public}d", voltage);
        ASSERT_TRUE(voltage == 4123456);
    }
}

/**
 * @tc.name: HdiService003
 * @tc.desc: Test functions of ParseCapacity
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService003, TestSize.Level1)
{
    int32_t capacity = -1;
    if (IsNotMock()) {
        giver_->ParseCapacity(&capacity);
        int32_t sysfsCapacity = ReadCapacitySysfs();
        BATTERY_HILOGI(
            LABEL_TEST, "Not Mcok HdiService003::capacity=%{public}d, l=%{public}d", capacity, sysfsCapacity);
        ASSERT_TRUE(capacity == sysfsCapacity);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "battery/capacity", "11");
        giver_->ParseCapacity(&capacity);
        BATTERY_HILOGI(LABEL_TEST, "HdiService003::capacity=%{public}d", capacity);
        ASSERT_TRUE(capacity == 11);
    }
}

/**
 * @tc.name: HdiService004
 * @tc.desc: Test functions of ParseHealthState
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService004, TestSize.Level1)
{
    int32_t healthState = -1;
    if (IsNotMock()) {
        giver_->ParseHealthState(&healthState);
        int32_t sysfsHealthState = ReadHealthStateSysfs();
        BATTERY_HILOGI(
            LABEL_TEST, "Not Mock HdiService004::healthState=%{public}d, h=%{public}d", healthState, sysfsHealthState);
        ASSERT_TRUE(healthState == sysfsHealthState);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "battery/health", "Good");
        giver_->ParseHealthState(&healthState);
        BATTERY_HILOGI(LABEL_TEST, "HdiService004::healthState=%{public}d.", healthState);
        ASSERT_TRUE(PowerSupplyProvider::BatteryHealthState(healthState) ==
            PowerSupplyProvider::BatteryHealthState::BATTERY_HEALTH_GOOD);
    }
}

/**
 * @tc.name: HdiService005
 * @tc.desc: Test functions of ParsePluggedType
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService005, TestSize.Level1)
{
    int32_t pluggedType = PowerSupplyProvider::PLUGGED_TYPE_NONE;
    if (IsNotMock()) {
        giver_->ParsePluggedType(&pluggedType);
        int32_t sysfsPluggedType = ReadPluggedTypeSysfs();
        BATTERY_HILOGI(
            LABEL_TEST, "Not Mock HdiService005::pluggedType=%{public}d, p=%{public}d", pluggedType, sysfsPluggedType);
        ASSERT_TRUE(pluggedType == sysfsPluggedType);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "ohos_charger/online", "1");
        CreateFile(MOCK_BATTERY_PATH + "ohos_charger/type", "Wireless");
        giver_->ParsePluggedType(&pluggedType);
        BATTERY_HILOGI(LABEL_TEST, "HdiService005::pluggedType=%{public}d.", pluggedType);
        ASSERT_TRUE(PowerSupplyProvider::BatteryPluggedType(pluggedType) ==
            PowerSupplyProvider::BatteryPluggedType::PLUGGED_TYPE_WIRELESS);
    }
}

/**
 * @tc.name: HdiService006
 * @tc.desc: Test functions of ParseChargeState
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService006, TestSize.Level1)
{
    int32_t chargeState = PowerSupplyProvider::CHARGE_STATE_RESERVED;
    if (IsNotMock()) {
        giver_->ParseChargeState(&chargeState);
        int32_t sysfsChargeState = ReadChargeStateSysfs();
        BATTERY_HILOGI(
            LABEL_TEST, "Not Mock HdiService006::chargeState=%{public}d, cs=%{public}d", chargeState, sysfsChargeState);
        ASSERT_TRUE(chargeState == sysfsChargeState);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "battery/status", "Not charging");
        giver_->ParseChargeState(&chargeState);
        BATTERY_HILOGI(LABEL_TEST, "HdiService006::chargeState=%{public}d.", chargeState);
        ASSERT_TRUE(PowerSupplyProvider::BatteryChargeState(chargeState) ==
            PowerSupplyProvider::BatteryChargeState::CHARGE_STATE_DISABLE);
    }
}

/**
 * @tc.name: HdiService007
 * @tc.desc: Test functions of ParseChargeCounter
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService007, TestSize.Level1)
{
    int32_t chargeCounter = -1;
    if (IsNotMock()) {
        giver_->ParseChargeCounter(&chargeCounter);
        int32_t sysfsChargeCounter = ReadChargeCounterSysfs();
        BATTERY_HILOGI(LABEL_TEST, "Not Mcok HdiService007::chargeCounter=%{public}d, cc=%{public}d", chargeCounter,
            sysfsChargeCounter);
        ASSERT_TRUE(chargeCounter == sysfsChargeCounter);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "battery/charge_counter", "12345");
        giver_->ParseChargeCounter(&chargeCounter);
        BATTERY_HILOGI(LABEL_TEST, "HdiService007::chargeCounter=%{public}d.", chargeCounter);
        ASSERT_TRUE(chargeCounter == 12345);
    }
}

/**
 * @tc.name: HdiService008
 * @tc.desc: Test functions of ParsePresent
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService008, TestSize.Level1)
{
    int8_t present = -1;
    if (IsNotMock()) {
        giver_->ParsePresent(&present);
        int32_t sysfsPresent = ReadPresentSysfs();
        BATTERY_HILOGI(LABEL_TEST, "Not Mock HdiService008::present=%{public}d, p=%{public}d", present, sysfsPresent);
        ASSERT_TRUE(present == sysfsPresent);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "battery/present", "1");
        giver_->ParsePresent(&present);
        BATTERY_HILOGI(LABEL_TEST, "HdiService008::present=%{public}d.", present);
        ASSERT_TRUE(present == 1);
    }
}

/**
 * @tc.name: HdiService009
 * @tc.desc: Test functions to get value of technology
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService009, TestSize.Level1)
{
    std::string technology = "invalid";
    if (IsNotMock()) {
        giver_->ParseTechnology(technology);
        std::string sysfsTechnology = "";
        ReadTechnologySysfs(sysfsTechnology);
        BATTERY_HILOGI(LABEL_TEST, "HdiService009::technology=%{public}s, ty=%{public}s", technology.c_str(),
            sysfsTechnology.c_str());
        ASSERT_TRUE(technology == sysfsTechnology);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "ohos-fgu/technology", "Li");
        giver_->ParseTechnology(technology);
        BATTERY_HILOGI(LABEL_TEST, "HdiService009::technology=%{public}s.", technology.c_str());
        ASSERT_TRUE(technology == "Li");
    }
}

/**
 * @tc.name: HdiService010
 * @tc.desc: Test functions to get fd of socket
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService010, TestSize.Level1)
{
    using namespace OHOS::HDI::Battery::V2_0;

    BatteryThread bt;
    auto fd = OpenUeventSocketTest(bt);
    BATTERY_HILOGI(LABEL_TEST, "HdiService010::fd=%{public}d.", fd);

    ASSERT_TRUE(fd > 0);
    close(fd);
}

/**
 * @tc.name: HdiService011
 * @tc.desc: Test functions UpdateEpollInterval when charge-online
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService011, TestSize.Level1)
{
    const int32_t CHARGE_STATE_ENABLE = 1;
    BatteryThread bt;

    UpdateEpollIntervalTest(CHARGE_STATE_ENABLE, bt);
    auto epollInterval = GetEpollIntervalTest(bt);
    BATTERY_HILOGI(LABEL_TEST, "HdiService011::epollInterval=%{public}d.", epollInterval);

    ASSERT_TRUE(epollInterval == 2000);
}

/**
 * @tc.name: HdiService012
 * @tc.desc: Test functions UpdateEpollInterval when charge-offline
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService012, TestSize.Level1)
{
    const int32_t CHARGE_STATE_NONE = 0;
    BatteryThread bt;

    UpdateEpollIntervalTest(CHARGE_STATE_NONE, bt);
    auto epollInterval = GetEpollIntervalTest(bt);
    BATTERY_HILOGI(LABEL_TEST, "HdiService012::epollInterval=%{public}d.", epollInterval);

    ASSERT_TRUE(epollInterval == -1);
}

/**
 * @tc.name: HdiService013
 * @tc.desc: Test functions Init
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService013, TestSize.Level1)
{
    void* service = nullptr;
    BatteryThread bt;

    InitTest(service, bt);
    BATTERY_HILOGI(LABEL_TEST, "HdiService013::InitTest success");
    auto epollFd = GetEpollFdTest(bt);
    BATTERY_HILOGI(LABEL_TEST, "HdiService013::epollFd=%{public}d", epollFd);

    ASSERT_TRUE(epollFd > 0);
}

/**
 * @tc.name: HdiService023
 * @tc.desc: Test functions of ParseTotalEnergy
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService023, TestSize.Level1)
{
    int32_t totalEnergy = 0;
    if (IsNotMock()) {
        giver_->ParseTotalEnergy(&totalEnergy);
        int32_t sysfsTotalEnergy = ReadTotalEnergySysfs();
        BATTERY_HILOGI(
            LABEL_TEST, "Not Mock HdiService023::totalEnergy=%{public}d, t=%{public}d", totalEnergy, sysfsTotalEnergy);
        ASSERT_TRUE(totalEnergy == sysfsTotalEnergy);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "battery/charge_full", "4000000");
        giver_->ParseTotalEnergy(&totalEnergy);
        BATTERY_HILOGI(LABEL_TEST, "HdiService023::totalEnergy=%{public}d.", totalEnergy);
        ASSERT_TRUE(totalEnergy == 4000000);
    }
}

/**
 * @tc.name: HdiService024
 * @tc.desc: Test functions of ParseCurrentAverage
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService024, TestSize.Level1)
{
    int32_t currentAvg = HDF_FAILURE;
    if (IsNotMock()) {
        giver_->ParseCurrentAverage(&currentAvg);
        int32_t sysfsCurrentAvg = ReadCurrentAverageSysfs();
        BATTERY_HILOGI(
            LABEL_TEST, "Not Mock HdiService024::currentAvg=%{public}d, t=%{public}d", currentAvg, sysfsCurrentAvg);
        ASSERT_TRUE(currentAvg == sysfsCurrentAvg);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "battery/current_avg", "1000");
        giver_->ParseCurrentAverage(&currentAvg);
        BATTERY_HILOGI(LABEL_TEST, "HdiService024::currentAvg=%{public}d.", currentAvg);
        ASSERT_TRUE(currentAvg == 1000);
    }
}

/**
 * @tc.name: HdiService025
 * @tc.desc: Test functions of ParseCurrentNow
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService025, TestSize.Level1)
{
    int32_t currentNow = 0;
    if (IsNotMock()) {
        giver_->ParseCurrentNow(&currentNow);
        int32_t sysfsCurrentNow = ReadCurrentNowSysfs();
        BATTERY_HILOGI(
            LABEL_TEST, "Not Mock HdiService025::currentNow=%{public}d, t=%{public}d", currentNow, sysfsCurrentNow);
        ASSERT_TRUE(currentNow == sysfsCurrentNow);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "battery/current_now", "1000");
        giver_->ParseCurrentNow(&currentNow);
        BATTERY_HILOGI(LABEL_TEST, "HdiService025::currentNow=%{public}d.", currentNow);
        ASSERT_TRUE(currentNow == 1000);
    }
}

/**
 * @tc.name: HdiService026
 * @tc.desc: Test functions of ParseChargeNow
 * @tc.type: FUNC
 */
HWTEST_F(HdiServiceTest, HdiService026, TestSize.Level1)
{
    int32_t chargeNow = 0;
    if (IsNotMock()) {
        giver_->ParseRemainEnergy(&chargeNow);
        int32_t sysfsChargeNow = ReadRemainEnergySysfs();
        BATTERY_HILOGI(
            LABEL_TEST, "Not Mock HdiService026::chargeNow=%{public}d, t=%{public}d", chargeNow, sysfsChargeNow);
        ASSERT_TRUE(chargeNow == sysfsChargeNow);
    } else {
        CreateFile(MOCK_BATTERY_PATH + "battery/charge_now", "1000");
        giver_->ParseRemainEnergy(&chargeNow);
        BATTERY_HILOGI(LABEL_TEST, "HdiService026::chargeNow=%{public}d.", chargeNow);
        ASSERT_TRUE(chargeNow == 1000);
    }
}
} // namespace HdiServiceTest