/*
 * Copyright (c) 2023 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 "asset_add_test.h"

#include <gtest/gtest.h>

#include "asset_api.h"
#include "asset_test_common.h"

using namespace testing::ext;
namespace UnitTest::AssetAddTest {
class AssetAddTest : public testing::Test {
public:
    static void SetUpTestCase(void);

    static void TearDownTestCase(void);

    void SetUp(void);

    void TearDown(void);
};

void AssetAddTest::SetUpTestCase(void)
{
}

void AssetAddTest::TearDownTestCase(void)
{
}

void AssetAddTest::SetUp(void)
{
}

void AssetAddTest::TearDown(void)
{
}

/**
 * @tc.name: AssetAddTest.AssetAddTest001
 * @tc.desc: Add asset with all attrs, then query, expect success and match
 * @tc.type: FUNC
 * @tc.result:0
 */
HWTEST_F(AssetAddTest, AssetAddTest001, TestSize.Level0)
{
    Asset_Blob funcName = { .size = strlen(__func__), .data = reinterpret_cast<uint8_t*>(const_cast<char*>(__func__)) };
    Asset_Attr attr[] = {
        { .tag = ASSET_TAG_ALIAS, .value.blob = funcName },
        { .tag = ASSET_TAG_SECRET, .value.blob = funcName },
        { .tag = ASSET_TAG_ACCESSIBILITY, .value.u32 = ASSET_ACCESSIBILITY_DEVICE_POWERED_ON },
        { .tag = ASSET_TAG_REQUIRE_PASSWORD_SET, .value.boolean = false },
        { .tag = ASSET_TAG_AUTH_TYPE, .value.u32 = ASSET_AUTH_TYPE_NONE },
        { .tag = ASSET_TAG_SYNC_TYPE, .value.u32 = ASSET_SYNC_TYPE_NEVER },
        { .tag = ASSET_TAG_CONFLICT_RESOLUTION, .value.u32 = ASSET_CONFLICT_OVERWRITE },
        { .tag = ASSET_TAG_DATA_LABEL_NORMAL_1, .value.blob = funcName },
        { .tag = ASSET_TAG_DATA_LABEL_NORMAL_2, .value.blob = funcName },
        { .tag = ASSET_TAG_DATA_LABEL_NORMAL_3, .value.blob = funcName },
        { .tag = ASSET_TAG_DATA_LABEL_NORMAL_4, .value.blob = funcName },
        { .tag = ASSET_TAG_DATA_LABEL_NORMAL_LOCAL_1, .value.blob = funcName },
        { .tag = ASSET_TAG_DATA_LABEL_NORMAL_LOCAL_2, .value.blob = funcName },
        { .tag = ASSET_TAG_DATA_LABEL_NORMAL_LOCAL_3, .value.blob = funcName },
        { .tag = ASSET_TAG_DATA_LABEL_NORMAL_LOCAL_4, .value.blob = funcName },
        { .tag = ASSET_TAG_DATA_LABEL_CRITICAL_1, .value.blob = funcName },
        { .tag = ASSET_TAG_DATA_LABEL_CRITICAL_2, .value.blob = funcName },
        { .tag = ASSET_TAG_DATA_LABEL_CRITICAL_3, .value.blob = funcName },
        { .tag = ASSET_TAG_DATA_LABEL_CRITICAL_4, .value.blob = funcName }
    };
    ASSERT_EQ(ASSET_SUCCESS, OH_Asset_Add(attr, ARRAY_SIZE(attr)));

    Asset_ResultSet resultSet = { 0 };
    ASSERT_EQ(ASSET_SUCCESS, QueryByAliasNdk(__func__, &resultSet));
    ASSERT_EQ(1, resultSet.count);
    Asset_Result result = resultSet.results[0];
    ASSERT_EQ(true, CheckMatchAttrResultNdk(attr, ARRAY_SIZE(attr), &result));

    OH_Asset_FreeResultSet(&resultSet);
    ASSERT_EQ(ASSET_SUCCESS, RemoveByAliasNdk(__func__));
}

/**
 * @tc.name: AssetAddTest.AssetAddTest002
 * @tc.desc: Add empty alias and secret, expect ASSET_INVALID_ARGUMENT
 * @tc.type: FUNC
 * @tc.result:0
 */
HWTEST_F(AssetAddTest, AssetAddTest002, TestSize.Level0)
{
    Asset_Blob alias = { .size = strlen(__func__), .data = nullptr };
    Asset_Blob secret = { .size = 0, .data = reinterpret_cast<uint8_t*>(const_cast<char*>(__func__)) };
    Asset_Attr attr[] = {
        { .tag = ASSET_TAG_ALIAS, .value.blob = alias },
        { .tag = ASSET_TAG_SECRET, .value.blob = secret },
        { .tag = ASSET_TAG_ACCESSIBILITY, .value.u32 = ASSET_ACCESSIBILITY_DEVICE_POWERED_ON },
    };
    ASSERT_EQ(ASSET_INVALID_ARGUMENT, OH_Asset_Add(attr, ARRAY_SIZE(attr)));
}

/**
 * @tc.name: AssetAddTest.AssetAddTest003
 * @tc.desc: Add alias and secret with wrong blob-u32/blob-boolean data type, expect ASSET_INVALID_ARGUMENT
 * @tc.type: FUNC
 * @tc.result:0
 */
HWTEST_F(AssetAddTest, AssetAddTest003, TestSize.Level0)
{
    Asset_Attr attr[] = {
        { .tag = ASSET_TAG_ALIAS, .value.u32 = 1 },
        { .tag = ASSET_TAG_SECRET, .value.boolean = true },
        { .tag = ASSET_TAG_ACCESSIBILITY, .value.u32 = ASSET_ACCESSIBILITY_DEVICE_POWERED_ON },
    };
    ASSERT_EQ(ASSET_INVALID_ARGUMENT, OH_Asset_Add(attr, ARRAY_SIZE(attr)));
}

/**
 * @tc.name: AssetAddTest.AssetAddTest004
 * @tc.desc: Add alias and secret with wrong u32-boolean data type, expect ASSET_INVALID_ARGUMENT
 * @tc.type: FUNC
 * @tc.result:0
 */
HWTEST_F(AssetAddTest, AssetAddTest004, TestSize.Level0)
{
    Asset_Blob alias = { .size = strlen(__func__), .data = nullptr };
    Asset_Blob secret = { .size = 0, .data = reinterpret_cast<uint8_t*>(const_cast<char*>(__func__)) };
    Asset_Attr attr[] = {
        { .tag = ASSET_TAG_ALIAS, .value.blob = alias },
        { .tag = ASSET_TAG_SECRET, .value.blob = secret },
        { .tag = ASSET_TAG_AUTH_TYPE, .value.boolean = false },
        { .tag = ASSET_TAG_ACCESSIBILITY, .value.u32 = ASSET_ACCESSIBILITY_DEVICE_POWERED_ON },
    };
    ASSERT_EQ(ASSET_INVALID_ARGUMENT, OH_Asset_Add(attr, ARRAY_SIZE(attr)));
}

/**
 * @tc.name: AssetAddTest.AssetAddTest005
 * @tc.desc: Add alias and secret with wrong bool-blob data type, expect ASSET_INVALID_ARGUMENT
 * @tc.type: FUNC
 * @tc.result:0
 */
HWTEST_F(AssetAddTest, AssetAddTest005, TestSize.Level0)
{
    Asset_Blob alias = { .size = strlen(__func__), .data = nullptr };
    Asset_Blob secret = { .size = 0, .data = reinterpret_cast<uint8_t*>(const_cast<char*>(__func__)) };
    Asset_Attr attr[] = {
        { .tag = ASSET_TAG_ALIAS, .value.blob = alias },
        { .tag = ASSET_TAG_SECRET, .value.blob = secret },
        { .tag = ASSET_TAG_REQUIRE_PASSWORD_SET, .value.blob = secret },
        { .tag = ASSET_TAG_ACCESSIBILITY, .value.u32 = ASSET_ACCESSIBILITY_DEVICE_POWERED_ON },
    };
    ASSERT_EQ(ASSET_INVALID_ARGUMENT, OH_Asset_Add(attr, ARRAY_SIZE(attr)));
}

/**
 * @tc.name: AssetAddTest.AssetAddTest006
 * @tc.desc: Add alias and secret, then add again, expect duplicate
 * @tc.type: FUNC
 * @tc.result:0
 */
HWTEST_F(AssetAddTest, AssetAddTest006, TestSize.Level0)
{
    Asset_Blob funcName = { .size = strlen(__func__), .data = reinterpret_cast<uint8_t*>(const_cast<char*>(__func__)) };
    Asset_Attr attr[] = {
        { .tag = ASSET_TAG_ALIAS, .value.blob = funcName },
        { .tag = ASSET_TAG_SECRET, .value.blob = funcName },
        { .tag = ASSET_TAG_ACCESSIBILITY, .value.u32 = ASSET_ACCESSIBILITY_DEVICE_POWERED_ON },
    };
    ASSERT_EQ(ASSET_SUCCESS, OH_Asset_Add(attr, ARRAY_SIZE(attr)));
    ASSERT_EQ(ASSET_DUPLICATED, OH_Asset_Add(attr, ARRAY_SIZE(attr)));

    ASSERT_EQ(ASSET_SUCCESS, RemoveByAliasNdk(__func__));
}

/**
 * @tc.name: AssetAddTest.AssetAddTest008
 * @tc.desc: Add without attr, expect ASSET_INVALID_ARGUMENT
 * @tc.type: FUNC
 * @tc.result:0
 */
HWTEST_F(AssetAddTest, AssetAddTest008, TestSize.Level0)
{
    ASSERT_EQ(ASSET_INVALID_ARGUMENT, OH_Asset_Add(nullptr, 0));
}

/**
 * @tc.name: AssetAddTest.AssetAddTest009
 * @tc.desc: Add without attr but count is wrong, expect ASSET_INVALID_ARGUMENT
 * @tc.type: FUNC
 * @tc.result:0
 */
HWTEST_F(AssetAddTest, AssetAddTest009, TestSize.Level0)
{
    ASSERT_EQ(ASSET_INVALID_ARGUMENT, OH_Asset_Add(nullptr, 1));
}

/**
 * @tc.name: AssetAddTest.AssetAddTest010
 * @tc.desc: Add with attr but count is wrong, expect ASSET_INVALID_ARGUMENT
 * @tc.type: FUNC
 * @tc.result:0
 */
HWTEST_F(AssetAddTest, AssetAddTest010, TestSize.Level0)
{
    Asset_Blob funcName = { .size = strlen(__func__), .data = reinterpret_cast<uint8_t*>(const_cast<char*>(__func__)) };
    Asset_Attr attr[] = {
        { .tag = ASSET_TAG_ALIAS, .value.blob = funcName },
        { .tag = ASSET_TAG_SECRET, .value.blob = funcName }
    };
    ASSERT_EQ(ASSET_INVALID_ARGUMENT, OH_Asset_Add(attr, 0));
}
}