/*
 * Copyright (C) 2021 Intel Corporation
 *
 * SPDX-License-Identifier: MIT
 *
 */

#include "shared/source/helpers/file_io.h"
#include "shared/source/utilities/cpu_info.h"

#include "gtest/gtest.h"

#include <cstdio>
#include <fstream>

using namespace NEO;

TEST(CpuInfo, givenProcCpuinfoFileExistsWhenIsCpuFlagPresentIsCalledThenValidValueIsReturned) {
    std::string cpuinfoFile = "test_files/linux/proc/cpuinfo";
    EXPECT_FALSE(fileExists(cpuinfoFile));

    {
        std::ofstream cpuinfo(cpuinfoFile);
        cpuinfo << "processor\t\t: 0\nflags\t\t: flag1 flag2 flag3\n";
    }

    EXPECT_TRUE(fileExists(cpuinfoFile));

    CpuInfo testCpuInfo;
    EXPECT_TRUE(testCpuInfo.isCpuFlagPresent("flag1"));
    EXPECT_TRUE(testCpuInfo.isCpuFlagPresent("flag2"));
    EXPECT_FALSE(testCpuInfo.isCpuFlagPresent("nonExistingCpuFlag"));

    std::remove(cpuinfoFile.c_str());
}

TEST(CpuInfo, givenProcCpuinfoFileIsNotExistsWhenIsCpuFlagPresentIsCalledThenValidValueIsReturned) {
    std::string cpuinfoFile = "test_files/linux/proc/cpuinfo";
    EXPECT_FALSE(fileExists(cpuinfoFile));

    CpuInfo testCpuInfo;
    EXPECT_FALSE(testCpuInfo.isCpuFlagPresent("flag1"));
    EXPECT_FALSE(testCpuInfo.isCpuFlagPresent("flag2"));
    EXPECT_FALSE(testCpuInfo.isCpuFlagPresent("nonExistingCpuFlag"));
}