/*
 * Copyright The WildFly Authors
 * SPDX-License-Identifier: Apache-2.0
 */

package org.jboss.as.test.integration.batch.batchlet;

import java.util.Arrays;
import java.util.List;
import java.util.Properties;
import java.util.Set;
import jakarta.batch.operations.JobOperator;
import jakarta.batch.runtime.BatchRuntime;
import jakarta.batch.runtime.BatchStatus;
import jakarta.batch.runtime.JobExecution;
import jakarta.batch.runtime.JobInstance;
import jakarta.batch.runtime.StepExecution;

import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.as.test.integration.batch.common.AbstractBatchTestCase;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;

/**
 * Tests that a {@code @RequestScoped} bean will work correctly in a batch job.
 * <p>
 * The order of the two tests should not matter. The reason it's tested twice however is a {@code @RequestScoped} bean
 * should create a new instance for each job executed. Executing twice ensures the scoping for batch is correct.
 * </p>
 *
 * @author <a href="mailto:jperkins@redhat.com">James R. Perkins</a>
 */
@RunWith(Arquillian.class)
public class SimpleCounterBatchletTestCase extends AbstractBatchTestCase {

    @Deployment
    public static WebArchive createDeployment() {
        return createDefaultWar("simple-counter-batchlet.war",
                        SimpleCounterBatchletTestCase.class.getPackage(),
                        "simple-counter-batchlet.xml",
                        "inactive-job.xml")
                .addPackage(SimpleCounterBatchletTestCase.class.getPackage());
    }

    /**
     * Tests that a request scope bean will be injected and the counter should start with 1 and end at 10.
     *
     * @throws Exception if an error occurs
     */
    @Test
    public void testRequestScope10() throws Exception {
        final Properties jobProperties = new Properties();
        jobProperties.setProperty("count", "10");
        final JobOperator jobOperator = BatchRuntime.getJobOperator();

        // check available job names
        // jobOperator.getJobNames() should return all available jobs in the
        // current deployment, whether a job has been started or not.
        final String inactiveJobName = "inactive-job";
        final List<String> expectedJobNames = Arrays.asList("simple-counter-batchlet", inactiveJobName);
        final Set<String> jobNames = jobOperator.getJobNames();
        Assert.assertTrue(String.format("Expecting job names: %s, but got %s", expectedJobNames, jobNames),
                jobNames.containsAll(expectedJobNames));
        Assert.assertEquals(2, jobNames.size());

        //getJobInstanceCount, getJobInstances & getRunningExecutions on an inactive job name should not cause exception
        final int jobInstanceCount = jobOperator.getJobInstanceCount(inactiveJobName);
        Assert.assertEquals("getJobInstanceCount() should return 0 for job name: " + inactiveJobName, 0, jobInstanceCount);
        final List<JobInstance> jobInstances = jobOperator.getJobInstances(inactiveJobName, 0, Integer.MAX_VALUE);
        Assert.assertEquals("getJobInstances() should return empty list for job name: " + inactiveJobName, 0, jobInstances.size());
        final List<Long> runningExecutions = jobOperator.getRunningExecutions(inactiveJobName);
        Assert.assertEquals("getRunningExecutions() should return empty list for job name: " + inactiveJobName, 0, runningExecutions.size());

        // Start the first job
        long executionId = jobOperator.start("simple-counter-batchlet", jobProperties);

        JobExecution jobExecution = jobOperator.getJobExecution(executionId);

        // Wait until the job is complete for a maximum of 5 seconds
        waitForTermination(jobExecution, 5);
        Assert.assertEquals(BatchStatus.COMPLETED, jobExecution.getBatchStatus());

        // Get the first step, should only be one, and get the exit status
        Assert.assertEquals("1,2,3,4,5,6,7,8,9,10", getFirst(jobOperator, executionId).getExitStatus());
    }


    /**
     * Tests that a request scope bean will be injected and the counter should start with 1 and end at 8.
     *
     * @throws Exception if an error occurs
     */
    @Test
    public void testRequestScope8() throws Exception {
        final Properties jobProperties = new Properties();
        jobProperties.setProperty("count", "8");
        final JobOperator jobOperator = BatchRuntime.getJobOperator();

        // Start the first job
        long executionId = jobOperator.start("simple-counter-batchlet", jobProperties);

        JobExecution jobExecution = jobOperator.getJobExecution(executionId);

        // Wait until the job is complete for a maximum of 5 seconds
        waitForTermination(jobExecution, 5);
        Assert.assertEquals(BatchStatus.COMPLETED, jobExecution.getBatchStatus());

        // Get the first step, should only be one, and get the exit status
        Assert.assertEquals("1,2,3,4,5,6,7,8", getFirst(jobOperator, executionId).getExitStatus());
    }

    private StepExecution getFirst(final JobOperator jobOperator, final long executionId) {
        final List<StepExecution> steps = jobOperator.getStepExecutions(executionId);
        Assert.assertFalse(steps.isEmpty());
        return steps.get(0);
    }
}
