/*
 * 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.
 */
import {describe, beforeAll, beforeEach, afterEach, afterAll, it, expect} from '@ohos/hypium'
import data_rdb from '@ohos.data.rdb'
import ability_featureAbility from '@ohos.ability.featureAbility'


const TAG = "[RDB_JSKITS_TEST]"
const CREATE_TABLE_TEST = "CREATE TABLE IF NOT EXISTS test (" + "id INTEGER PRIMARY KEY AUTOINCREMENT, "
    + "name TEXT NOT NULL, " + "age INTEGER, " + "salary REAL, " + "blobType BLOB)"

var rdbStore
var context
const STORE_CONFIG_ENCRYPT = {
    name: "Encrypt.db",
    encrypt: true,
}
const STORE_CONFIG_UNENCRYPT = {
    name: "Unencrypt.db",
    encrypt: false,
}
const STORE_CONFIG_WRONG = {
    name: "Encrypt.db",
    encrypt: false,
}

export default function rdbStoreEncryptionTest() {
async function CreatRdbStore(context, STORE_CONFIG) {
    let rdbStore = await data_rdb.getRdbStore(context, STORE_CONFIG, 1)
    await rdbStore.executeSql(CREATE_TABLE_TEST, null)
    let u8 = new Uint8Array([1, 2, 3])
    {
        const valueBucket = {
            "name": "zhangsan",
            "age": 18,
            "salary": 100.5,
            "blobType": u8,
        }
        await rdbStore.insert("test", valueBucket)
    }
    {
        const valueBucket = {
            "name": "lisi",
            "age": 28,
            "salary": 100.5,
            "blobType": u8,
        }
        await rdbStore.insert("test", valueBucket)
    }
    {
        const valueBucket = {
            "name": "wangwu",
            "age": 38,
            "salary": 90.0,
            "blobType": u8,
        }
        await rdbStore.insert("test", valueBucket)
    }
    return rdbStore
}

describe('rdbEncryptTest', function () {
        beforeAll(async function () {
            console.info(TAG + 'beforeAll')

        })

        beforeEach(async function () {
            console.info(TAG + 'beforeEach')

        })

        afterEach(async function () {
            console.info(TAG + 'afterEach')
            await data_rdb.deleteRdbStore(context, STORE_CONFIG_ENCRYPT.name)
            await data_rdb.deleteRdbStore(context, STORE_CONFIG_UNENCRYPT.name)
            await data_rdb.deleteRdbStore(context, STORE_CONFIG_WRONG.name)
            rdbStore = null
        })

        afterAll(async function () {
            console.info(TAG + 'afterAll')
        })

        console.log(TAG + "*************Unit Test Begin*************")

        /**
         * @tc.name RDB encrypted test
         * @tc.number SUB_DDM_RDB_JS_RdbEncryptTest_0010
         * @tc.desc RDB create encrypt db test
         */
        it('RdbEncryptTest_0010', 0, async function (done) {
            await console.log(TAG + "************* RdbEncryptTest_0010 start *************")
            context = ability_featureAbility.getContext()
            let storePromise = data_rdb.getRdbStore(context, STORE_CONFIG_ENCRYPT, 1);
            storePromise.then(async (store) => {
                try {
                    await console.log(TAG + "getRdbStore done: " + store);
                } catch (err) {
                    expect(null).assertFail();
                }
            }).catch((err) => {
                expect(null).assertFail();
            })
            await storePromise
            storePromise = null

            done()
            await console.log(TAG + "************* RdbEncryptTest_0010 end *************")
        })

        /**
         * @tc.name RDB unencrypted test
         * @tc.number SUB_DDM_RDB_JS_RdbEncryptTest_0020
         * @tc.desc RDB create unencrypted db test
         */
        it('RdbEncryptTest_0020', 0, async function (done) {
            await console.log(TAG + "************* RdbEncryptTest_0020 start *************")
            context = ability_featureAbility.getContext()
            let storePromise = data_rdb.getRdbStore(context, STORE_CONFIG_UNENCRYPT, 1);
            storePromise.then(async (store) => {
                try {
                    await console.log(TAG + "getRdbStore done: " + store);
                } catch (err) {
                    expect(null).assertFail();
                }
            }).catch((err) => {
                expect(null).assertFail();
            })
            await storePromise
            storePromise = null

            done()
            await console.log(TAG + "************* RdbEncryptTest_0020 end *************")
        })


        /**
         * @tc.name RDB Encrypt test
         * @tc.number SUB_DDM_RDB_JS_RdbEncryptTest_0030
         * @tc.desc RDB Encrypt function test
         */
        it('RdbEncryptTest_0030', 0, async function (done) {
            await console.log(TAG + "************* RdbEncryptTest_0030 start *************")
            context = ability_featureAbility.getContext()
            rdbStore = await CreatRdbStore(context, STORE_CONFIG_ENCRYPT)
            let predicates = new data_rdb.RdbPredicates("test")
            predicates.equalTo("name", "zhangsan")
            let resultSet = await rdbStore.query(predicates)
            try {
                console.log(TAG + "After restore resultSet query done")
                expect(true).assertEqual(resultSet.goToFirstRow())
                const id = resultSet.getLong(resultSet.getColumnIndex("id"))
                const name = resultSet.getString(resultSet.getColumnIndex("name"))
                const blobType = resultSet.getBlob(resultSet.getColumnIndex("blobType"))
                expect(1).assertEqual(id)
                expect("zhangsan").assertEqual(name)
                expect(1).assertEqual(blobType[0])
            } catch (err) {
                expect(false).assertTrue()
            }
            resultSet = null
            rdbStore = null
            done()
            await console.log(TAG + "************* RdbEncryptTest_0030 end *************")
        })

        /**
         * @tc.name RDB Encrypt test
         * @tc.number SUB_DDM_RDB_JS_RdbEncryptTest_0040
         * @tc.desc RDB Encrypt function test
         */
        it('RdbEncryptTest_0040', 0, async function (done) {
            await console.log(TAG + "************* RdbEncryptTest_0040 start *************")
            context = ability_featureAbility.getContext()
            rdbStore = await CreatRdbStore(context, STORE_CONFIG_ENCRYPT)
            rdbStore = null
            rdbStore = await CreatRdbStore(context, STORE_CONFIG_WRONG)
            expect(rdbStore).assertNull

            done()
            await console.log(TAG + "************* RdbEncryptTest_0040 end *************")
        })
        console.log(TAG + "*************Unit Test End*************")
    }
)

}