package com.sequenceiq.it.cloudbreak;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.junit.Assert;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Optional;
import org.testng.annotations.Parameters;
import org.testng.annotations.Test;

import com.amazonaws.regions.Region;
import com.amazonaws.regions.Regions;
import com.amazonaws.services.ec2.AmazonEC2Client;
import com.amazonaws.services.ec2.model.DescribeSpotInstanceRequestsResult;
import com.amazonaws.services.ec2.model.SpotInstanceRequest;
import com.sequenceiq.cloudbreak.api.endpoint.StackEndpoint;
import com.sequenceiq.cloudbreak.api.model.InstanceGroupResponse;
import com.sequenceiq.cloudbreak.api.model.InstanceMetaDataJson;
import com.sequenceiq.cloudbreak.api.model.StackResponse;
import com.sequenceiq.it.IntegrationTestContext;

public class AwsCheckSpotInstance extends AbstractCloudbreakIntegrationTest {

    @BeforeMethod
    public void setContextParams() {
        IntegrationTestContext itContext = getItContext();
        Assert.assertNotNull(itContext.getContextParam(CloudbreakITContextConstants.STACK_ID), "Stack id is mandatory.");
    }

    @Parameters({"region", "hostGroupToCheck", "scalingAdjustment"})
    @Test
    public void checkSpotInstance(Regions region, String hostGroupToCheck, @Optional Integer scalingAdjustment) {
        //GIVEN
        Integer spotInstanceCount = 0;
        IntegrationTestContext itContext = getItContext();

        String stackId = itContext.getContextParam(CloudbreakITContextConstants.STACK_ID);

        StackEndpoint stackEndpoint = getCloudbreakClient().stackEndpoint();
        StackResponse stackResponse = stackEndpoint.get(Long.valueOf(stackId));

        List<InstanceGroupResponse> instanceGroups = stackResponse.getInstanceGroups();

        List<String> instanceIdList = new ArrayList<>();
        List<String> hostGroupList = Arrays.asList(hostGroupToCheck.split(","));

        for (InstanceGroupResponse instanceGroup : instanceGroups) {
            if (hostGroupList.contains(instanceGroup.getGroup().toString())) {
                Set<InstanceMetaDataJson> instanceMetaData = instanceGroup.getMetadata();
                for (InstanceMetaDataJson metaData : instanceMetaData) {
                    instanceIdList.add(metaData.getInstanceId());
                }
            }
        }
        //WHEN
        AmazonEC2Client ec2 = new AmazonEC2Client();
        ec2.setRegion(Region.getRegion(region));
        DescribeSpotInstanceRequestsResult describeSpotInstanceRequestsResult = ec2.describeSpotInstanceRequests();
        List<SpotInstanceRequest> spotInstanceRequests = describeSpotInstanceRequestsResult.getSpotInstanceRequests();
        //THEN
        Assert.assertFalse(spotInstanceRequests.isEmpty());

        List<String> spotInstanceIdList = new ArrayList<>();

        for (SpotInstanceRequest request : spotInstanceRequests) {
            spotInstanceIdList.add(request.getInstanceId());
        }

        for (String id : instanceIdList) {
            Assert.assertTrue(spotInstanceIdList.contains(id));
            if (spotInstanceIdList.contains(id)) {
                spotInstanceCount += 1;
            }
        }

        if (scalingAdjustment != null) {
            Assert.assertNotNull(itContext.getContextParam(CloudbreakITContextConstants.INSTANCE_COUNT, List.class));
            Integer instanceCountPrev = 0;
            for (String hostGroup : hostGroupList) {
                List<Map<String, Integer>> instanceList = itContext.getContextParam(CloudbreakITContextConstants.INSTANCE_COUNT, List.class);
                Assert.assertTrue(instanceList.size() >= 2);
                instanceCountPrev += instanceList.get(instanceList.size() - 2).get(hostGroup);
            }
            Assert.assertEquals(Integer.valueOf(instanceCountPrev + scalingAdjustment), spotInstanceCount);
        }
    }
}