// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you 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.
package org.apache.cloudstack.storage.test;

import java.lang.reflect.Method;

import org.springframework.test.context.testng.AbstractTestNGSpringContextTests;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Parameters;
import org.testng.annotations.Test;

import com.cloud.hypervisor.Hypervisor.HypervisorType;
import com.cloud.utils.db.TransactionLegacy;

public class CloudStackTestNGBase extends AbstractTestNGSpringContextTests {
    private String hostGateway;
    private String hostCidr;
    private String hostIp;
    private String hostGuid;
    private String templateUrl;
    private String localStorageUuid;
    private String primaryStorageUrl;
    private String secondaryStorage;
    private String imageInstallPath;
    private String scriptPath;
    private HypervisorType hypervisor;
    private TransactionLegacy txn;

    private String s3AccessKey;
    private String s3SecretKey;
    private String s3EndPoint;
    private String s3TemplateBucket;
    private String primaryStorageUuid;
    private boolean s3UseHttps;

    protected void injectMockito() {

    }

    @BeforeMethod(alwaysRun = true)
    protected void injectDB(Method testMethod) throws Exception {
        txn = TransactionLegacy.open(testMethod.getName());
    }

    @Test
    protected void injectMockitoTest() {
        injectMockito();
    }

    @AfterMethod(alwaysRun = true)
    protected void closeDB(Method testMethod) throws Exception {
        if (txn != null) {
            txn.close();
        }
    }

    @BeforeMethod(alwaysRun = true)
    @Parameters({"devcloud-host-uuid", "devcloud-host-gateway", "devcloud-host-cidr", "devcloud-host-ip", "template-url", "devcloud-local-storage-uuid",
        "primary-storage-want-to-add", "devcloud-secondary-storage", "s3-accesskey", "s3-secretkey", "s3-endpoint", "s3-template-bucket", "s3-usehttps",
        "image-install-path", "primary-storage-uuid-want-to-add", "script-path", "hypervisor"})
    protected void setup(String hostuuid, String gateway, String cidr, String hostIp, String templateUrl, String localStorageUuid, String primaryStorage,
        String secondaryStorage, String s3AccessKey, String s3SecretKey, String s3Endpoint, String s3TemplateBucket, String s3Usehttps, String imageInstallPath,
        String primaryStorageUuid, String scriptPath, String hypervisor) {
        this.hostGuid = hostuuid;
        this.hostGateway = gateway;
        this.hostCidr = cidr;
        this.hostIp = hostIp;
        this.templateUrl = templateUrl;
        this.localStorageUuid = localStorageUuid;
        this.primaryStorageUrl = primaryStorage;
        this.primaryStorageUuid = primaryStorageUuid;
        this.imageInstallPath = imageInstallPath;
        this.hypervisor = HypervisorType.getType(hypervisor);
        this.setSecondaryStorage(secondaryStorage);
        // set S3 parameters
        this.s3AccessKey = s3AccessKey;
        this.s3SecretKey = s3SecretKey;
        this.s3EndPoint = s3Endpoint;
        this.s3TemplateBucket = s3TemplateBucket;
        this.s3UseHttps = Boolean.parseBoolean(s3Usehttps);
        this.scriptPath = scriptPath;
        if (this.scriptPath != null) {
            System.setProperty("paths.script", this.getScriptPath());
        }
    }

    protected String getHostGuid() {
        return this.hostGuid;
    }

    protected String getHostGateway() {
        return this.hostGateway;
    }

    protected String getHostCidr() {
        return this.hostCidr;
    }

    protected String getHostIp() {
        return this.hostIp;
    }

    protected String getTemplateUrl() {
        return this.templateUrl;
    }

    protected String getLocalStorageUuid() {
        return this.localStorageUuid;
    }

    protected String getPrimaryStorageUrl() {
        return this.primaryStorageUrl;
    }

    public String getSecondaryStorage() {
        return secondaryStorage;
    }

    public void setSecondaryStorage(String secondaryStorage) {
        this.secondaryStorage = secondaryStorage;
    }

    public String getS3AccessKey() {
        return s3AccessKey;
    }

    public String getS3SecretKey() {
        return s3SecretKey;
    }

    public String getS3EndPoint() {
        return s3EndPoint;
    }

    public String getS3TemplateBucket() {
        return s3TemplateBucket;
    }

    public boolean isS3UseHttps() {
        return s3UseHttps;
    }

    public String getImageInstallPath() {
        return imageInstallPath;
    }

    public void setImageInstallPath(String imageInstallPath) {
        this.imageInstallPath = imageInstallPath;
    }

    public String getPrimaryStorageUuid() {
        return primaryStorageUuid;
    }

    public void setPrimaryStorageUuid(String primaryStorageUuid) {
        this.primaryStorageUuid = primaryStorageUuid;
    }

    public String getScriptPath() {
        return scriptPath;
    }

    public void setScriptPath(String scriptPath) {
        this.scriptPath = scriptPath;
    }

    public HypervisorType getHypervisor() {
        return hypervisor;
    }

    public void setHypervisor(HypervisorType hypervisor) {
        this.hypervisor = hypervisor;
    }

}