// Copyright 2020 The Android Open Source Project
//
// 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 "android/base/misc/FileUtils.h"

#include "android/base/files/ScopedFd.h"
#include "android/utils/eintr_wrapper.h"
#include "android/utils/file_io.h"
#include "android/utils/tempfile.h"

#include <fcntl.h>
#include <gtest/gtest.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <limits>
#include <string>

using android::base::ScopedFd;

namespace android {

TEST(FileUtils, stringToFile) {
    const char test_pattern[] = "test pattern";
    TempFile* tf = tempfile_create();

    ScopedFd fd(HANDLE_EINTR(open(tempfile_path(tf), O_RDWR, 0600)));
    EXPECT_NE(-1, fd.get());

    EXPECT_FALSE(writeStringToFile(-1, test_pattern));
    EXPECT_TRUE(writeStringToFile(fd.get(), test_pattern));

    std::string dummy_buffer("dummy");
    EXPECT_FALSE(readFileIntoString(-1, &dummy_buffer));
    EXPECT_STREQ("dummy", dummy_buffer.c_str());

    std::string read_buffer;
    EXPECT_TRUE(readFileIntoString(fd.get(), &read_buffer));
    EXPECT_STREQ(test_pattern, read_buffer.c_str());

    fd.close();

    auto readFailedRes = readFileIntoString("!@#%R#$%W$*@#$*");
    EXPECT_FALSE(readFailedRes);

    auto readSucceededRes = readFileIntoString(tempfile_path(tf));
    EXPECT_TRUE(readSucceededRes);
    EXPECT_STREQ(test_pattern, readSucceededRes->c_str());

    tempfile_close(tf);
}

// Tests readFileIntoString
TEST(FileUtils, fileToString) {
    const int bytesToTest = 100;

    std::vector<uint8_t> testPattern;

    for (int i = 0; i < bytesToTest; i++) {
        testPattern.push_back(i % 3 == 0 ? 0x0 : 0x1a);
    }

    TempFile* tf = tempfile_create();

    ScopedFd fd(HANDLE_EINTR(open(tempfile_path(tf), O_RDWR, 0600)));
    EXPECT_NE(-1, fd.get());

    HANDLE_EINTR(write(fd.get(), testPattern.data(), bytesToTest));

    fd.close();

    const auto testOutput = readFileIntoString(tempfile_path(tf));

    EXPECT_TRUE(testOutput);
    EXPECT_EQ(bytesToTest, testOutput->size());

    for (int i = 0; i < bytesToTest; i++) {
        EXPECT_EQ((char)testPattern[i], (char)testOutput->at(i));
    }
}

}  // namespace android