/*
 * Copyright (c) 2020-2021 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 "layout/grid_layout.h"

#include <climits>
#include <gtest/gtest.h>

using namespace testing::ext;
namespace OHOS {
namespace {
UIView* CreatView()
{
    uint16_t width = 100; // 100  view width
    uint16_t height = 100; // 100 view height
    auto view = new UIView();
    view->Resize(width, height);
    EXPECT_EQ(view->GetX(), 0);
    EXPECT_EQ(view->GetY(), 0);
    return view;
}
}
class GridLayoutTest : public testing::Test {
public:
    static void SetUpTestCase(void);
    static void TearDownTestCase(void);
    static GridLayout* gridLayout_;
};

GridLayout* GridLayoutTest::gridLayout_ = nullptr;

void GridLayoutTest::SetUpTestCase(void)
{
    if (gridLayout_ == nullptr) {
        gridLayout_ = new GridLayout();
        gridLayout_->SetPosition(0, 0, 600, 300); // 600: layout width; 300: layout height
        gridLayout_->SetRows(2); // 2: row
        gridLayout_->SetCols(3); // 3: col
    }
}

void GridLayoutTest::TearDownTestCase(void)
{
    if (gridLayout_ != nullptr) {
        delete gridLayout_;
        gridLayout_ = nullptr;
    }
}

/**
 * @tc.name: GridLayout_001
 * @tc.desc: Normal Process.
 * @tc.type: FUNC
 * @tc.require: AR000DSMR7
 */
HWTEST_F(GridLayoutTest, GridLayout_001, TestSize.Level1)
{
    if (gridLayout_ == nullptr) {
        EXPECT_EQ(1, 0);
        return;
    }
    /*
     * 1 2 3
     * 4 5 6
     */
    gridLayout_->SetLayoutDirection(LAYOUT_HOR);
    auto view1 = CreatView();
    gridLayout_->Add(view1);
    auto view2 = CreatView();
    gridLayout_->Add(view2);
    auto view3 = CreatView();
    gridLayout_->Add(view3);
    auto view4 = CreatView();
    gridLayout_->Add(view4);
    gridLayout_->LayoutChildren();
    EXPECT_EQ(view1->GetX(), 50);  // 50: view x after layout
    EXPECT_EQ(view1->GetY(), 25);  // 25: view y after layout
    EXPECT_EQ(view2->GetX(), 250); // 250: view x after layout
    EXPECT_EQ(view2->GetY(), view1->GetY());
    EXPECT_EQ(view3->GetX(), 450); // 450: view x after layout
    EXPECT_EQ(view3->GetY(), view1->GetY());
    EXPECT_EQ(view4->GetX(), view1->GetX());
    EXPECT_EQ(view4->GetY(), 175); // 175: view x after layout
    gridLayout_->RemoveAll();
    delete view1;
    delete view2;
    delete view3;
    delete view4;
}

/**
 * @tc.name: GridLayout_002
 * @tc.desc: Normal Process.
 * @tc.type: FUNC
 * @tc.require: AR000DSMR7
 */
HWTEST_F(GridLayoutTest, GridLayout_002, TestSize.Level1)
{
    if (gridLayout_ == nullptr) {
        EXPECT_EQ(1, 0);
        return;
    }
    /*
     * 1 3 5
     * 2 4 6
     */
    gridLayout_->SetLayoutDirection(LAYOUT_HOR_R);
    auto view1 = CreatView();
    gridLayout_->Add(view1);
    auto view2 = CreatView();
    gridLayout_->Add(view2);
    auto view3 = CreatView();
    gridLayout_->Add(view3);
    auto view4 = CreatView();
    gridLayout_->Add(view4);
    gridLayout_->LayoutChildren();
    EXPECT_EQ(view1->GetX(), 50);  // 50: view x after layout
    EXPECT_EQ(view1->GetY(), 25);  // 25: view y after layout
    EXPECT_EQ(view2->GetX(), view1->GetX());
    EXPECT_EQ(view2->GetY(), 175); // 175: view y afert layout
    EXPECT_EQ(view3->GetX(), 250); // 250: view x after layout
    EXPECT_EQ(view3->GetY(), view1->GetY());
    EXPECT_EQ(view4->GetX(), view3->GetX());
    EXPECT_EQ(view4->GetY(), view2->GetY());
    gridLayout_->RemoveAll();
    delete view1;
    delete view2;
    delete view3;
    delete view4;
}

/**
 * @tc.name: GridLayout_003
 * @tc.desc: Normal Process.
 * @tc.type: FUNC
 * @tc.require: AR000DSMR7
 */
HWTEST_F(GridLayoutTest, GridLayout_003, TestSize.Level1)
{
    if (gridLayout_ == nullptr) {
        EXPECT_EQ(1, 0);
        return;
    }
    /*
     * 1 3 5
     * 2 4 6
     */
    gridLayout_->SetLayoutDirection(LAYOUT_VER);
    auto view1 = CreatView();
    gridLayout_->Add(view1);
    auto view2 = CreatView();
    gridLayout_->Add(view2);
    auto view3 = CreatView();
    gridLayout_->Add(view3);
    auto view4 = CreatView();
    gridLayout_->Add(view4);
    gridLayout_->LayoutChildren();
    EXPECT_EQ(view1->GetX(), 50);  // 50: view x after layout
    EXPECT_EQ(view1->GetY(), 25);  // 25: view y after layout
    EXPECT_EQ(view2->GetX(), view1->GetX());
    EXPECT_EQ(view2->GetY(), 175); // 175: view y afert layout
    EXPECT_EQ(view3->GetX(), 250); // 250: view x after layout
    EXPECT_EQ(view3->GetY(), view1->GetY());
    EXPECT_EQ(view4->GetX(), view3->GetX());
    EXPECT_EQ(view4->GetY(), view2->GetY());
    gridLayout_->RemoveAll();
    delete view1;
    delete view2;
    delete view3;
    delete view4;
}

/**
 * @tc.name: GridLayout_004
 * @tc.desc: Normal Process.
 * @tc.type: FUNC
 * @tc.require: AR000DSMR7
 */
HWTEST_F(GridLayoutTest, GridLayout_004, TestSize.Level1)
{
    if (gridLayout_ == nullptr) {
        EXPECT_EQ(1, 0);
        return;
    }
    /*
     * 1 2 3
     * 4 5 6
     */
    gridLayout_->SetLayoutDirection(LAYOUT_VER_R);
    auto view1 = CreatView();
    gridLayout_->Add(view1);
    auto view2 = CreatView();
    gridLayout_->Add(view2);
    auto view3 = CreatView();
    gridLayout_->Add(view3);
    auto view4 = CreatView();
    gridLayout_->Add(view4);
    gridLayout_->LayoutChildren();
    EXPECT_EQ(view1->GetX(), 50);  // 50: view x after layout
    EXPECT_EQ(view1->GetY(), 25);  // 25: view y after layout
    EXPECT_EQ(view2->GetX(), view1->GetX());
    EXPECT_EQ(view2->GetY(), 175); // 175: view y afert layout
    EXPECT_EQ(view3->GetX(), 250); // 250: view x after layout
    EXPECT_EQ(view3->GetY(), view1->GetY());
    EXPECT_EQ(view4->GetX(), view3->GetX());
    EXPECT_EQ(view4->GetY(), view2->GetY());
    gridLayout_->RemoveAll();
    delete view1;
    delete view2;
    delete view3;
    delete view4;
}
} // namespace OHOS