// Copyright 2021 Google LLC
//
// 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 "util.hpp"

#include <flasher/device/fake.hpp>
#include <flasher/mutate/asymmetric.hpp>
#include <flasher/ops.hpp>

#include <optional>
#include <stdexcept>

#include <gtest/gtest.h>

namespace flasher
{
namespace ops
{

class VerifyTest : public OpTest<device::Fake>
{
  protected:
    VerifyTest()
    {
        fillFileInc(f, 0, 12);
    }
};

TEST_F(VerifyTest, InvalidOffset)
{
    EXPECT_THROW(verify(d, /*dev_offset=*/13, f, /*file_offset=*/0, m,
                        /*max_size=*/0, /*stride_size=*/std::nullopt),
                 std::invalid_argument);
}

TEST_F(VerifyTest, InvalidStride)
{
    EXPECT_THROW(verify(d, /*dev_offset=*/0, f, /*file_offset=*/0, m,
                        /*max_size=*/0, /*stride_size=*/0),
                 std::invalid_argument);
}

TEST_F(VerifyTest, ValidWhole)
{
    verify(d, /*dev_offset=*/0, f, /*file_offset=*/0, m, /*max_size=*/20,
           /*stride_size=*/std::nullopt);
    verify(d, /*dev_offset=*/0, f, /*file_offset=*/0, m, /*max_size=*/12,
           /*stride_size=*/1);
    verify(d, /*dev_offset=*/0, f, /*file_offset=*/0, m, /*max_size=*/11,
           /*stride_size=*/std::nullopt);
}

TEST_F(VerifyTest, SmallFile)
{
    f.data.resize(4);
    verify(d, /*dev_offset=*/0, f, /*file_offset=*/0, m, /*max_size=*/20,
           /*stride_size=*/std::nullopt);
}

TEST_F(VerifyTest, FileTooBig)
{
    fillFileInc(f, 0, 13);
    EXPECT_THROW(verify(d, /*dev_offset=*/0, f, /*file_offset=*/0, m,
                        /*max_size=*/20, /*stride_size=*/std::nullopt),
                 std::runtime_error);
}

TEST_F(VerifyTest, ValidSubset)
{
    fillFileInc(f, 2, 10);
    verify(d, /*dev_offset=*/3, f, /*file_offset=*/1, m, /*max_size=*/20,
           /*stride_size=*/std::nullopt);
}

TEST_F(VerifyTest, TooBigSubset)
{
    fillFileInc(f, 2, 11);
    EXPECT_THROW(verify(d, /*dev_offset=*/3, f, /*file_offset=*/1, m,
                        /*max_size=*/20, /*stride_size=*/std::nullopt),
                 std::runtime_error);
}

TEST_F(VerifyTest, RespectSize)
{
    fillFileInc(f, 2, 11);
    f.data[5] = std::byte{0xff};
    verify(d, /*dev_offset=*/3, f, /*file_offset=*/1, m, /*max_size=*/4,
           /*stride_size=*/std::nullopt);
}

TEST_F(VerifyTest, MutateApplied)
{
    m.mutations.push_back(std::make_unique<mutate::Asymmetric>());
    m.forward(df.data, 0);
    verify(d, /*dev_offset=*/1, f, /*file_offset=*/1, m, /*max_size=*/4,
           /*stride_size=*/std::nullopt);
}

} // namespace ops
} // namespace flasher