/*
 * 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
 *
 *   https://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.
 *
 *   SPDX-License-Identifier: Apache-2.0
 */

package dboe;

import java.util.List;

import org.apache.jena.atlas.lib.Lib;
import org.apache.jena.atlas.lib.RandomLib;
import org.apache.jena.cmd.CmdException;
import org.apache.jena.cmd.CmdMain;

public abstract class BaseSoakTest extends CmdMain {

    protected final int MinOrder = 2;
    protected final int MinSize  = 1;
    protected int       MaxOrder = -1;
    protected int       MaxSize  = -1;
    protected int       NumTest  = -1;

    protected BaseSoakTest(String[] argv) {
        super(argv);
    }

    protected abstract void before();
    protected abstract void after();

    @Override
    protected String getSummary() {
        return "Usage: "+Lib.className(this)+" maxOrder maxSize NumTests";
    }

    @Override
    protected void processModulesAndArgs() {
        List<String> args = super.getPositional();
        if ( args.size() != 3 )
            throw new CmdException("Usage: maxOrder maxSize NumTests");

        try { MaxOrder = Integer.parseInt(args.get(0)); }
        catch (NumberFormatException ex)
        { throw new CmdException("Bad number for MaxOrder"); }

        try { MaxSize = Integer.parseInt(args.get(1)); }
        catch (NumberFormatException ex)
        { throw new CmdException("Bad number for MaxSize"); }

        try { NumTest = Integer.parseInt(args.get(2)); }
        catch (NumberFormatException ex)
        { throw new CmdException("Bad number for NumTest"); }
    }

    @Override
    protected void exec() {
        int successes   = 0;
        int failures    = 0;

        // Number of dots.
        int numOnLine = 50;
        int testsPerTick;
        if ( NumTest < 20 )
            testsPerTick = 5;
        else if ( NumTest < 200 )
            testsPerTick = 50;
        else
            testsPerTick = 500;


        // ---- Format for line counter.
        int numLines = (int)Math.ceil( ((double)NumTest) / (testsPerTick * numOnLine) );
        // Start of last line.
        int z = (numLines-1)*(testsPerTick * numOnLine);
        int digits = 1;
        if ( z > 0 )
            digits = 1+(int)Math.floor(Math.log10(z));
        String format = "[%"+digits+"d] ";

        System.out.printf("TEST : %,d tests : Max Order=%d  Max Items=%,d [tests per tick=%d]\n", NumTest, MaxOrder, MaxSize, testsPerTick);

        before();

        int testCount = 1;

        for ( testCount = 1; testCount <= NumTest; testCount++ ) {
            if ( testCount % testsPerTick == 0 )
                System.out.print(".");
            if ( testCount % (testsPerTick * numOnLine) == 0 )
                System.out.println("");
            if ( testCount % (testsPerTick * numOnLine) == 1 )
                System.out.printf(format, testCount-1);

            int idx = testCount - 1;
            int order = ( MinOrder == MaxOrder ) ? MinOrder : MinOrder + RandomLib.random.nextInt(MaxOrder-MinOrder);
            int size =  ( MinSize  == MaxSize  ) ? MinSize :  MinSize  + RandomLib.random.nextInt(MaxSize-MinSize);            try {
                //System.out.printf("TEST : %,d : Order=%-2d : Size=%d\n", testCount, order, size);
                runOneTest(testCount, order, size);
                successes++;
            }
            catch (AssertionError | RuntimeException ex) {
                System.err.printf("-- Fail: (order=%d, size=%d)\n", order, size);
                ex.printStackTrace(System.err);
                System.err.printf("--------------------------\n");
                failures++;
            }
        }

        // Did the last loop print a new line?
        if ( (testCount-1) % (testsPerTick*numOnLine) != 0 )
            System.out.println();

        after();
        System.err.flush();
        System.out.flush();
        System.out.printf("DONE : %,d tests : Success=%,d; Failures=%,d\n", NumTest, successes, failures);
    }

    protected abstract void runOneTest(int testCount, int order, int size, boolean debug);

    protected abstract void runOneTest(int testCount, int order, int size);

    @Override
    protected String getCommandName() { return Lib.className(this); }
}

