/*
 * 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 <cmath>
#include <cstdio>
#include <unistd.h>
#include <gtest/gtest.h>
#include <securec.h>
#include <vector>
#include "hdf_base.h"
#include "osal_time.h"
#include "v1_1/imotion_interface.h"
#include "motion_callback_impl.h"

#define DATA_NUM 12
#define DATA_VALUE 6

using namespace OHOS::HDI::Motion::V1_1;
using namespace testing::ext;

namespace {
    sptr<OHOS::HDI::Motion::V1_1::IMotionInterface> g_motionInterface = nullptr;
    sptr<IMotionCallback> g_motionCallback = new MotionCallbackImpl();
    sptr<IMotionCallback> g_motionCallbackUnregistered = new MotionCallbackImpl();
    std::vector<uint8_t> g_motionConfigData(DATA_NUM, DATA_VALUE);
}

class HdfMotionTest : public testing::Test {
public:
    static void SetUpTestCase();
    static void TearDownTestCase();
    void SetUp();
    void TearDown();
};

void HdfMotionTest::SetUpTestCase()
{
    g_motionInterface = OHOS::HDI::Motion::V1_1::IMotionInterface::Get();
}

void HdfMotionTest::TearDownTestCase()
{
}

void HdfMotionTest::SetUp()
{
}

void HdfMotionTest::TearDown()
{
}

HWTEST_F(HdfMotionTest, GetMotionClient_001, TestSize.Level1)
{
    ASSERT_NE(nullptr, g_motionInterface);
}

HWTEST_F(HdfMotionTest, RegisterMotionDataCb_001, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    int32_t ret = g_motionInterface->Register(g_motionCallback);
    EXPECT_EQ(HDF_SUCCESS, ret);
    ret = g_motionInterface->Unregister(g_motionCallback);
    EXPECT_EQ(0, ret);
}

HWTEST_F(HdfMotionTest, RegisterMotionDataCb_002, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    int32_t ret = g_motionInterface->Register(g_motionCallback);
    EXPECT_EQ(HDF_SUCCESS, ret);
    ret = g_motionInterface->Register(g_motionCallback);
    EXPECT_EQ(HDF_FAILURE, ret);
    ret = g_motionInterface->Unregister(g_motionCallback);
    EXPECT_EQ(HDF_SUCCESS, ret);
}

HWTEST_F(HdfMotionTest, RegisterMotionDataCb_003, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    sptr<IMotionCallback> motionCallback = nullptr;
    int32_t ret = g_motionInterface->Register(motionCallback);
    EXPECT_NE(HDF_SUCCESS, ret);
}

HWTEST_F(HdfMotionTest, RegisterMotionDataCb_004, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    int32_t ret = g_motionInterface->Unregister(g_motionCallbackUnregistered);
    EXPECT_EQ(HDF_FAILURE, ret);
}

HWTEST_F(HdfMotionTest, RegisterMotionDataCb_005, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    sptr<IMotionCallback> motionCallback = nullptr;
    int32_t ret = g_motionInterface->Unregister(motionCallback);
    EXPECT_NE(HDF_SUCCESS, ret);
}

HWTEST_F(HdfMotionTest, EnableMotion_001, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }

    vector<int32_t> vec;
    vec.push_back(OHOS::HDI::Motion::V1_1::HDF_MOTION_TYPE_PICKUP);
    vec.push_back(OHOS::HDI::Motion::V1_1::HDF_MOTION_TYPE_FLIP);
    vec.push_back(OHOS::HDI::Motion::V1_1::HDF_MOTION_TYPE_SHAKE);
    vec.push_back(OHOS::HDI::Motion::V1_1::HDF_MOTION_TYPE_ROTATION);

    int32_t ret = g_motionInterface->Register(g_motionCallback);
    EXPECT_EQ(HDF_SUCCESS, ret);

    for (size_t i = 0; i < vec.size(); i++) {
        ret = g_motionInterface->EnableMotion(vec[i]);
        if (ret == HDF_SUCCESS) {
            printf("Motion %d enabled successfully\n", vec[i]);
        } else {
            printf("Motion %d enable failed\n", vec[i]);
        }
        EXPECT_EQ(HDF_SUCCESS, ret);
        OsalSleep(15);

        ret = g_motionInterface->DisableMotion(vec[i]);
        if (ret == HDF_SUCCESS) {
            printf("Motion %d disabled successfully\n", vec[i]);
        } else {
            printf("Motion %d disable failed\n", vec[i]);
        }
        EXPECT_EQ(HDF_SUCCESS, ret);
        OsalSleep(2);
    }

    ret = g_motionInterface->Unregister(g_motionCallback);
    EXPECT_EQ(HDF_SUCCESS, ret);
}

HWTEST_F(HdfMotionTest, EnableMotion_002, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    int32_t ret = g_motionInterface->EnableMotion(OHOS::HDI::Motion::V1_1::HDF_MOTION_TYPE_MAX);
    EXPECT_NE(HDF_SUCCESS, ret);
    OsalSleep(2);

    ret = g_motionInterface->DisableMotion(OHOS::HDI::Motion::V1_1::HDF_MOTION_TYPE_MAX);
    EXPECT_NE(HDF_SUCCESS, ret);
}

HWTEST_F(HdfMotionTest, EnableMotion_003, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    int32_t ret = g_motionInterface->EnableMotion(OHOS::HDI::Motion::V1_1::HDF_MOTION_TYPE_WRIST_UP);
    EXPECT_EQ(HDF_SUCCESS, ret);
    OsalSleep(2);

    ret = g_motionInterface->DisableMotion(OHOS::HDI::Motion::V1_1::HDF_MOTION_TYPE_WRIST_UP);
    EXPECT_EQ(HDF_SUCCESS, ret);
}

HWTEST_F(HdfMotionTest, EnableMotion_004, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    int32_t motionType = -1;
    int32_t ret = g_motionInterface->EnableMotion(motionType);
    EXPECT_NE(HDF_SUCCESS, ret);
    OsalSleep(2);

    ret = g_motionInterface->DisableMotion(motionType);
    EXPECT_NE(HDF_SUCCESS, ret);
}

HWTEST_F(HdfMotionTest, DisableMotion_001, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    int32_t ret = g_motionInterface->Register(g_motionCallback);
    EXPECT_EQ(HDF_SUCCESS, ret);

    ret = g_motionInterface->EnableMotion(OHOS::HDI::Motion::V1_1::HDF_MOTION_TYPE_PICKUP);
    EXPECT_EQ(HDF_SUCCESS, ret);
    ret = g_motionInterface->DisableMotion(OHOS::HDI::Motion::V1_1::HDF_MOTION_TYPE_PICKUP);
    EXPECT_EQ(HDF_SUCCESS, ret);

    ret = g_motionInterface->Unregister(g_motionCallback);
    EXPECT_EQ(HDF_SUCCESS, ret);
}

HWTEST_F(HdfMotionTest, DisableMotion_002, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    int32_t ret = g_motionInterface->DisableMotion(OHOS::HDI::Motion::V1_1::HDF_MOTION_TYPE_MAX);
    EXPECT_EQ(HDF_ERR_INVALID_PARAM, ret);
}

HWTEST_F(HdfMotionTest, DisableMotion_003, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    int32_t ret = g_motionInterface->DisableMotion(OHOS::HDI::Motion::V1_1::HDF_MOTION_TYPE_PICKUP);
    EXPECT_EQ(HDF_SUCCESS, ret);
}

HWTEST_F(HdfMotionTest, DisableMotion_004, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    int32_t motionType = -1;
    int32_t ret = g_motionInterface->DisableMotion(motionType);
    EXPECT_NE(HDF_SUCCESS, ret);
}

HWTEST_F(HdfMotionTest, SetMotionConfig_001, TestSize.Level1)
{
    if (g_motionInterface == nullptr) {
        ASSERT_NE(nullptr, g_motionInterface);
        return;
    }
    int32_t motionType = -1;
    int32_t ret = g_motionInterface->SetMotionConfig(motionType, g_motionConfigData);
    EXPECT_NE(HDF_SUCCESS, ret);
}