/*
 * 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 <cstdio>
#include <iostream>
#include <atomic>
#include <fstream>
#include <mutex>
#include <queue>
#include <string>
#include <thread>

#include "gtest/gtest.h"
#include "videoenc_sample.h"
#include "avcodec_codec_name.h"
#include "native_avcapability.h"
using namespace std;
using namespace OHOS;
using namespace OHOS::Media;
using namespace testing::ext;
namespace OHOS {
namespace Media {
class HwEncStateNdkTest : public testing::Test {
public:
    // SetUpTestCase: Called before all test cases
    static void SetUpTestCase(void);
    // TearDownTestCase: Called after all test case
    static void TearDownTestCase(void);
    // SetUp: Called before each test cases
    void SetUp(void);
    // TearDown: Called after each test cases
    void TearDown(void);

protected:
    const ::testing::TestInfo *testInfo_ = nullptr;
    bool createCodecSuccess_ = false;
    OH_AVCapability *cap = nullptr;
};

void HwEncStateNdkTest::SetUpTestCase(void) {}

void HwEncStateNdkTest::TearDownTestCase(void) {}

VEncNdkSample *vEncSample = NULL;

void HwEncStateNdkTest::SetUp(void)
{
    vEncSample = new VEncNdkSample();
    cap = OH_AVCodec_GetCapabilityByCategory(OH_AVCODEC_MIMETYPE_VIDEO_AVC, true, HARDWARE);
    const char *codeName = OH_AVCapability_GetName(cap);
    cout << "codecname: " << codeName << endl;
    int32_t ret = vEncSample->CreateVideoEncoder(codeName);
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->SetVideoEncoderCallback();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->ConfigureVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
    vEncSample->INP_DIR = "/data/test/media/1280_720_nv.yuv";
}

void HwEncStateNdkTest::TearDown(void)
{
    vEncSample->Release();
    delete vEncSample;
    vEncSample = nullptr;
}
} // namespace Media
} // namespace OHOS

namespace {
/**
 * @tc.number    : VIDEO_ENCODE_STATE_0100
 * @tc.name      : create-configure-error
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_0100, TestSize.Level2)
{
    int32_t ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->SetVideoEncoderCallback();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_0101
 * @tc.name      : create-configure-stop
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_0101, TestSize.Level2)
{
    int32_t ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_0200
 * @tc.name      : create-configure-start-stop-start
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_0200, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_0300
 * @tc.name      : create-configure-start-stop-release
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_0300, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Release();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_0400
 * @tc.name      : create-configure-start-stop-reset
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_0400, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_0500CODEC_NAME_AVC
 * @tc.name      : create-configure-start-stop-error
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_0500, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->SetVideoEncoderCallback();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_0600
 * @tc.name      : create-configure-start-EOS-stop-start
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_0600, TestSize.Level2)
{
    int32_t ret = vEncSample->StartVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
    vEncSample->WaitForEOS();
    ASSERT_EQ(AV_ERR_OK, ret);
    ASSERT_EQ(0, vEncSample->errCount);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_0700
 * @tc.name      : create-configure-start-EOS-stop-release
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_0700, TestSize.Level2)
{
    int32_t ret = vEncSample->StartVideoEncoder();
    vEncSample->WaitForEOS();
    ASSERT_EQ(AV_ERR_OK, ret);
    ASSERT_EQ(0, vEncSample->errCount);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Release();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_0800
 * @tc.name      : create-configure-start-EOS-stop-reset
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_0800, TestSize.Level2)
{
    int32_t ret = vEncSample->StartVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
    vEncSample->WaitForEOS();
    ASSERT_EQ(AV_ERR_OK, ret);
    ASSERT_EQ(0, vEncSample->errCount);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_0900
 * @tc.name      : create-configure-start-EOS-flush
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_0900, TestSize.Level2)
{
    int32_t ret = vEncSample->StartVideoEncoder();
    vEncSample->WaitForEOS();
    ASSERT_EQ(AV_ERR_OK, ret);
    ASSERT_EQ(0, vEncSample->errCount);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_1000
 * @tc.name      : create-configure-start-EOS-flush-start
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_1000, TestSize.Level2)
{
    int32_t ret = vEncSample->StartVideoEncoder();
    vEncSample->WaitForEOS();
    ASSERT_EQ(AV_ERR_OK, ret);
    ASSERT_EQ(0, vEncSample->errCount);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_1100
 * @tc.name      : create-configure-start-EOS-flush-stop
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_1100, TestSize.Level2)
{
    int32_t ret = vEncSample->StartVideoEncoder();
    vEncSample->WaitForEOS();
    ASSERT_EQ(AV_ERR_OK, ret);
    ASSERT_EQ(0, vEncSample->errCount);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_1200
 * @tc.name      : create-configure-start-EOS-flush-reset
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_1200, TestSize.Level2)
{
    int32_t ret = vEncSample->StartVideoEncoder();
    vEncSample->WaitForEOS();
    ASSERT_EQ(AV_ERR_OK, ret);
    ASSERT_EQ(0, vEncSample->errCount);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_1300
 * @tc.name      : create-configure-start-EOS-flush-error
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_1300, TestSize.Level2)
{
    int32_t ret = vEncSample->StartVideoEncoder();
    vEncSample->WaitForEOS();
    ASSERT_EQ(AV_ERR_OK, ret);
    ASSERT_EQ(0, vEncSample->errCount);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Release();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->ConfigureVideoEncoder();
    ASSERT_EQ(AV_ERR_INVALID_VAL, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_1400
 * @tc.name      : create-configure-start-EOS-reset-configure
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_1400, TestSize.Level2)
{
    int32_t ret = vEncSample->StartVideoEncoder();
    vEncSample->WaitForEOS();
    ASSERT_EQ(AV_ERR_OK, ret);
    ASSERT_EQ(0, vEncSample->errCount);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->ConfigureVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_1500
 * @tc.name      : create-configure-start-EOS-reset-release
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_1500, TestSize.Level2)
{
    int32_t ret = vEncSample->StartVideoEncoder();
    vEncSample->WaitForEOS();
    ASSERT_EQ(AV_ERR_OK, ret);
    ASSERT_EQ(0, vEncSample->errCount);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Release();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_1600
 * @tc.name      : create-configure-start-EOS-reset-error
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_1600, TestSize.Level2)
{
    int32_t ret = vEncSample->StartVideoEncoder();
    vEncSample->WaitForEOS();
    ASSERT_EQ(AV_ERR_OK, ret);
    ASSERT_EQ(0, vEncSample->errCount);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_1700
 * @tc.name      : create-configure-start-flush-start-flush
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_1700, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_1800
 * @tc.name      : create-configure-start-flush-start-eos
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_1800, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    cout << "flush ok" << endl;
    ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    cout << "start ok" << endl;
    ret = vEncSample->state_EOS();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_1900
 * @tc.name      : create-configure-start-flush-start-stop
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_1900, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_2000
 * @tc.name      : create-configure-start-flush-start-reset
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_2000, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_2100
 * @tc.name      : create-configure-start-flush-start-error
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_2100, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->ConfigureVideoEncoder();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->SetVideoEncoderCallback();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Release();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_2200
 * @tc.name      : create-configure-start-flush-start-error
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_2200, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    cout << "set callback" << endl;
    ret = vEncSample->SetVideoEncoderCallback();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_2300
 * @tc.name      : create-configure-start-flush-stop-start
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_2300, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Release();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_2400
 * @tc.name      : create-configure-start-flush-stop-reset
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_2400, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_2500
 * @tc.name      : create-configure-start-flush-stop-error
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_2500, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->SetVideoEncoderCallback();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_2600
 * @tc.name      : create-configure-start-flush-reset-configure
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_2600, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->ConfigureVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_2700
 * @tc.name      : create-configure-start-flush-reset-release
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_2700, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Release();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_2800
 * @tc.name      : create-configure-start-flush-reset-error
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_2800, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->SetVideoEncoderCallback();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_2900
 * @tc.name      : create-configure-start-reset-configure
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_2900, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->ConfigureVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_3000
 * @tc.name      : create-configure-start-reset-release
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_3000, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Release();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_3100
 * @tc.name      : create-configure-start-reset-error
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_3100, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->Stop();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->Flush();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->SetVideoEncoderCallback();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_3200
 * @tc.name      : create-configure-start-error
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_3200, TestSize.Level2)
{
    int32_t ret = vEncSample->Start();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->ConfigureVideoEncoder();
    ASSERT_EQ(AV_ERR_INVALID_STATE, ret);
    ret = vEncSample->SetVideoEncoderCallback();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->Release();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_3300
 * @tc.name      : create-configure-reset-configure
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_3300, TestSize.Level2)
{
    int32_t ret = vEncSample->Reset();
    ASSERT_EQ(AV_ERR_OK, ret);
    ret = vEncSample->ConfigureVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_3400
 * @tc.name      : create-configure-release
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_3400, TestSize.Level2)
{
    int32_t ret = vEncSample->Release();
    ASSERT_EQ(AV_ERR_OK, ret);
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_3500
 * @tc.name      : Flush in surf encode callback function
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_3500, TestSize.Level1)
{
    vEncSample->outputCallbackFlush = true;
    vEncSample->SURF_INPUT = true;
    int32_t ret = vEncSample->StartVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
    vEncSample->WaitForEOS();
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_3600
 * @tc.name      : Stop in surf encode callback function
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_3600, TestSize.Level1)
{
    vEncSample->outputCallbackStop = true;
    vEncSample->SURF_INPUT = true;
    int32_t ret = vEncSample->StartVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
    vEncSample->WaitForEOS();
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_3700
 * @tc.name      : Flush or stop in buffer encode callback function
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_3700, TestSize.Level1)
{
    vEncSample->inputCallbackFlush = true;
    int32_t ret = vEncSample->StartVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
    vEncSample->WaitForEOS();
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_3800
 * @tc.name      : Flush or stop in buffer encode callback function
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_3800, TestSize.Level1)
{
    vEncSample->inputCallbackStop = true;
    int32_t ret = vEncSample->StartVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
    vEncSample->WaitForEOS();
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_3900
 * @tc.name      : Flush or stop in buffer encode callback function
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_3900, TestSize.Level1)
{
    vEncSample->outputCallbackFlush = true;
    int32_t ret = vEncSample->StartVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
    vEncSample->WaitForEOS();
}

/**
 * @tc.number    : VIDEO_ENCODE_STATE_4000
 * @tc.name      : Flush or stop in buffer encode callback function
 * @tc.desc      : function test
 */
HWTEST_F(HwEncStateNdkTest, VIDEO_ENCODE_STATE_4000, TestSize.Level1)
{
    vEncSample->outputCallbackStop = true;
    int32_t ret = vEncSample->StartVideoEncoder();
    ASSERT_EQ(AV_ERR_OK, ret);
    vEncSample->WaitForEOS();
}
} // namespace