test/jdk/tools/jpackage/helpers/jdk/jpackage/test/TestBuilder.java
author herrick
Wed, 16 Oct 2019 10:32:08 -0400
branchJDK-8200758-branch
changeset 58648 3bf53ffa9ae7
parent 58416 f09bf58c1f17
permissions -rw-r--r--
8232279 : Improve test helpers #2 Submitted-by: asemenyuk Reviewed-by: aherrick, almatvee
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
     1
/*
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
     2
 * Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved.
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
     4
 *
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
     7
 * published by the Free Software Foundation.
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
     8
 *
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
     9
 * This code is distributed in the hope that it will be useful, but WITHOUT
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    10
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    11
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    12
 * version 2 for more details (a copy is included in the LICENSE file that
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    13
 * accompanied this code).
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    14
 *
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    15
 * You should have received a copy of the GNU General Public License version
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    16
 * 2 along with this work; if not, write to the Free Software Foundation,
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    17
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    18
 *
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    19
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    20
 * or visit www.oracle.com if you need additional information or have any
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    21
 * questions.
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    22
 */
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    23
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    24
package jdk.jpackage.test;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    25
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    26
import java.lang.reflect.Array;
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    27
import java.lang.reflect.InvocationTargetException;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    28
import java.lang.reflect.Method;
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    29
import java.lang.reflect.Modifier;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    30
import java.util.*;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    31
import java.util.concurrent.atomic.AtomicInteger;
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    32
import java.util.function.Consumer;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    33
import java.util.function.Function;
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    34
import java.util.function.UnaryOperator;
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    35
import java.util.stream.Collectors;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    36
import java.util.stream.Stream;
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    37
import jdk.jpackage.test.Annotations.AfterEach;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    38
import jdk.jpackage.test.Annotations.BeforeEach;
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    39
import jdk.jpackage.test.Annotations.Parameter;
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    40
import jdk.jpackage.test.Annotations.ParameterGroup;
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    41
import jdk.jpackage.test.Annotations.Parameters;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    42
import jdk.jpackage.test.Annotations.Test;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    43
import jdk.jpackage.test.Functional.ThrowingConsumer;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    44
import jdk.jpackage.test.Functional.ThrowingFunction;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    45
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    46
final class TestBuilder implements AutoCloseable {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    47
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    48
    @Override
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    49
    public void close() throws Exception {
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    50
        flushTestGroup();
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    51
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    52
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    53
    TestBuilder(Consumer<TestInstance> testConsumer) {
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    54
        argProcessors = Map.of(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    55
                CMDLINE_ARG_PREFIX + "after-run",
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    56
                arg -> getJavaMethodsFromArg(arg).map(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    57
                        this::wrap).forEachOrdered(afterActions::add),
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    58
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    59
                CMDLINE_ARG_PREFIX + "before-run",
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    60
                arg -> getJavaMethodsFromArg(arg).map(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    61
                        this::wrap).forEachOrdered(beforeActions::add),
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    62
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    63
                CMDLINE_ARG_PREFIX + "run",
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    64
                arg -> addTestGroup(getJavaMethodsFromArg(arg).map(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    65
                        ThrowingFunction.toFunction(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    66
                                TestBuilder::toMethodCalls)).flatMap(s -> s).collect(
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    67
                        Collectors.toList())),
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    68
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    69
                CMDLINE_ARG_PREFIX + "exclude",
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    70
                arg -> (excludedTests = Optional.ofNullable(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    71
                        excludedTests).orElse(new HashSet<String>())).add(arg),
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    72
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    73
                CMDLINE_ARG_PREFIX + "include",
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    74
                arg -> (includedTests = Optional.ofNullable(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    75
                        includedTests).orElse(new HashSet<String>())).add(arg),
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    76
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    77
                CMDLINE_ARG_PREFIX + "space-subst",
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    78
                arg -> spaceSubstitute = arg,
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    79
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    80
                CMDLINE_ARG_PREFIX + "group",
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    81
                arg -> flushTestGroup(),
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    82
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    83
                CMDLINE_ARG_PREFIX + "dry-run",
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    84
                arg -> dryRun = true
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
    85
        );
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    86
        this.testConsumer = testConsumer;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    87
        clear();
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    88
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    89
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    90
    void processCmdLineArg(String arg) throws Throwable {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    91
        int separatorIdx = arg.indexOf('=');
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    92
        final String argName;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    93
        final String argValue;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    94
        if (separatorIdx != -1) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    95
            argName = arg.substring(0, separatorIdx);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    96
            argValue = arg.substring(separatorIdx + 1);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    97
        } else {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    98
            argName = arg;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
    99
            argValue = null;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   100
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   101
        try {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   102
            ThrowingConsumer<String> argProcessor = argProcessors.get(argName);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   103
            if (argProcessor == null) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   104
                throw new ParseException("Unrecognized");
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   105
            }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   106
            argProcessor.accept(argValue);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   107
        } catch (ParseException ex) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   108
            ex.setContext(arg);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   109
            throw ex;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   110
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   111
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   112
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   113
    private void addTestGroup(List<MethodCall> newTestGroup) {
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   114
        if (testGroup != null) {
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   115
            testGroup.addAll(newTestGroup);
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   116
        } else {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   117
            testGroup = newTestGroup;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   118
        }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   119
    }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   120
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   121
    private static Stream<MethodCall> filterTests(Stream<MethodCall> tests,
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   122
            Set<String> filters, UnaryOperator<Boolean> pred, String logMsg) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   123
        if (filters == null) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   124
            return tests;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   125
        }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   126
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   127
        // Log all matches before returning from the function
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   128
        return tests.filter(test -> {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   129
            String testDescription = test.createDescription().testFullName();
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   130
            boolean match = filters.stream().anyMatch(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   131
                    v -> testDescription.contains(v));
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   132
            if (match) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   133
                trace(String.format(logMsg + ": %s", testDescription));
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   134
            }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   135
            return pred.apply(match);
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   136
        }).collect(Collectors.toList()).stream();
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   137
    }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   138
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   139
    private Stream<MethodCall> filterTestGroup() {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   140
        Objects.requireNonNull(testGroup);
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   141
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   142
        UnaryOperator<Set<String>> restoreSpaces = filters -> {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   143
            if (spaceSubstitute == null || filters == null) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   144
                return filters;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   145
            }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   146
            return filters.stream().map(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   147
                    filter -> filter.replace(spaceSubstitute, " ")).collect(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   148
                            Collectors.toSet());
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   149
        };
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   150
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   151
        if (includedTests != null) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   152
            return filterTests(testGroup.stream(), restoreSpaces.apply(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   153
                    includedTests), x -> x, "Include");
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   154
        }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   155
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   156
        return filterTests(testGroup.stream(),
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   157
                restoreSpaces.apply(excludedTests), x -> !x, "Exclude");
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   158
    }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   159
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   160
    private void flushTestGroup() {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   161
        if (testGroup != null) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   162
            filterTestGroup().forEach(testBody -> createTestInstance(testBody));
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   163
            clear();
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   164
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   165
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   166
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   167
    private void createTestInstance(MethodCall testBody) {
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   168
        final List<ThrowingConsumer> curBeforeActions;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   169
        final List<ThrowingConsumer> curAfterActions;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   170
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   171
        Method testMethod = testBody.getMethod();
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   172
        if (Stream.of(BeforeEach.class, AfterEach.class).anyMatch(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   173
                type -> testMethod.isAnnotationPresent(type))) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   174
            curBeforeActions = beforeActions;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   175
            curAfterActions = afterActions;
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   176
        } else {
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   177
            curBeforeActions = new ArrayList<>(beforeActions);
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   178
            curAfterActions = new ArrayList<>(afterActions);
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   179
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   180
            selectFrameMethods(testMethod.getDeclaringClass(), BeforeEach.class).map(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   181
                    this::wrap).forEachOrdered(curBeforeActions::add);
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   182
            selectFrameMethods(testMethod.getDeclaringClass(), AfterEach.class).map(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   183
                    this::wrap).forEachOrdered(curAfterActions::add);
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   184
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   185
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   186
        TestInstance test = new TestInstance(testBody, curBeforeActions,
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   187
                curAfterActions, dryRun);
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   188
        if (includedTests == null) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   189
            trace(String.format("Create: %s", test.fullName()));
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   190
        }
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   191
        testConsumer.accept(test);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   192
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   193
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   194
    private void clear() {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   195
        beforeActions = new ArrayList<>();
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   196
        afterActions = new ArrayList<>();
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   197
        excludedTests = null;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   198
        includedTests = null;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   199
        spaceSubstitute = null;
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   200
        testGroup = null;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   201
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   202
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   203
    private static Class probeClass(String name) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   204
        try {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   205
            return Class.forName(name);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   206
        } catch (ClassNotFoundException ex) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   207
            return null;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   208
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   209
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   210
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   211
    private static Stream<Method> selectFrameMethods(Class type, Class annotationType) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   212
        return Stream.of(type.getMethods())
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   213
                .filter(m -> m.getParameterCount() == 0)
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   214
                .filter(m -> !m.isAnnotationPresent(Test.class))
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   215
                .filter(m -> m.isAnnotationPresent(annotationType))
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   216
                .sorted((a, b) -> a.getName().compareTo(b.getName()));
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   217
    }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   218
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   219
    private static Stream<String> cmdLineArgValueToMethodNames(String v) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   220
        List<String> result = new ArrayList<>();
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   221
        String defaultClassName = null;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   222
        for (String token : v.split(",")) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   223
            Class testSet = probeClass(token);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   224
            if (testSet != null) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   225
                // Test set class specified. Pull in all public methods
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   226
                // from the class with @Test annotation removing name duplicates.
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   227
                // Overloads will be handled at the next phase of processing.
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   228
                defaultClassName = token;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   229
                Stream.of(testSet.getMethods()).filter(
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   230
                        m -> m.isAnnotationPresent(Test.class)).map(
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   231
                                Method::getName).distinct().forEach(
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   232
                                name -> result.add(String.join(".", token, name)));
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   233
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   234
                continue;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   235
            }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   236
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   237
            final String qualifiedMethodName;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   238
            final int lastDotIdx = token.lastIndexOf('.');
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   239
            if (lastDotIdx != -1) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   240
                qualifiedMethodName = token;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   241
                defaultClassName = token.substring(0, lastDotIdx);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   242
            } else if (defaultClassName == null) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   243
                throw new ParseException("Default class name not found in");
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   244
            } else {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   245
                qualifiedMethodName = String.join(".", defaultClassName, token);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   246
            }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   247
            result.add(qualifiedMethodName);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   248
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   249
        return result.stream();
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   250
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   251
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   252
    private static boolean filterMethod(String expectedMethodName, Method method) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   253
        if (!method.getName().equals(expectedMethodName)) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   254
            return false;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   255
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   256
        switch (method.getParameterCount()) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   257
            case 0:
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   258
                return !isParametrized(method);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   259
            case 1:
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   260
                return isParametrized(method);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   261
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   262
        return false;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   263
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   264
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   265
    private static boolean isParametrized(Method method) {
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   266
        return method.isAnnotationPresent(ParameterGroup.class) || method.isAnnotationPresent(
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   267
                Parameter.class);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   268
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   269
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   270
    private static List<Method> getJavaMethodFromString(
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   271
            String qualifiedMethodName) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   272
        int lastDotIdx = qualifiedMethodName.lastIndexOf('.');
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   273
        if (lastDotIdx == -1) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   274
            throw new ParseException("Class name not found in");
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   275
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   276
        String className = qualifiedMethodName.substring(0, lastDotIdx);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   277
        String methodName = qualifiedMethodName.substring(lastDotIdx + 1);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   278
        Class methodClass;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   279
        try {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   280
            methodClass = Class.forName(className);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   281
        } catch (ClassNotFoundException ex) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   282
            throw new ParseException(String.format("Class [%s] not found;",
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   283
                    className));
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   284
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   285
        // Get the list of all public methods as need to deal with overloads.
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   286
        List<Method> methods = Stream.of(methodClass.getMethods()).filter(
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   287
                (m) -> filterMethod(methodName, m)).collect(Collectors.toList());
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   288
        if (methods.isEmpty()) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   289
            new ParseException(String.format(
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   290
                    "Method [%s] not found in [%s] class;",
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   291
                    methodName, className));
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   292
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   293
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   294
        trace(String.format("%s -> %s", qualifiedMethodName, methods));
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   295
        return methods;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   296
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   297
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   298
    private static Stream<Method> getJavaMethodsFromArg(String argValue) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   299
        return cmdLineArgValueToMethodNames(argValue).map(
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   300
                ThrowingFunction.toFunction(
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   301
                        TestBuilder::getJavaMethodFromString)).flatMap(
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   302
                        List::stream).sequential();
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   303
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   304
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   305
    private static Parameter[] getMethodParameters(Method method) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   306
        if (method.isAnnotationPresent(ParameterGroup.class)) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   307
            return ((ParameterGroup) method.getAnnotation(ParameterGroup.class)).value();
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   308
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   309
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   310
        if (method.isAnnotationPresent(Parameter.class)) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   311
            return new Parameter[]{(Parameter) method.getAnnotation(
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   312
                Parameter.class)};
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   313
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   314
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   315
        // Unexpected
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   316
        return null;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   317
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   318
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   319
    private static Stream<Object[]> toCtorArgs(Method method) throws
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   320
            IllegalAccessException, InvocationTargetException {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   321
        Class type = method.getDeclaringClass();
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   322
        List<Method> paremetersProviders = Stream.of(type.getMethods())
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   323
                .filter(m -> m.getParameterCount() == 0)
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   324
                .filter(m -> (m.getModifiers() & Modifier.STATIC) != 0)
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   325
                .filter(m -> m.isAnnotationPresent(Parameters.class))
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   326
                .sorted()
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   327
                .collect(Collectors.toList());
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   328
        if (paremetersProviders.isEmpty()) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   329
            // Single instance using the default constructor.
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   330
            return Stream.ofNullable(MethodCall.DEFAULT_CTOR_ARGS);
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   331
        }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   332
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   333
        // Pick the first method from the list.
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   334
        Method paremetersProvider = paremetersProviders.iterator().next();
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   335
        if (paremetersProviders.size() > 1) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   336
            trace(String.format(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   337
                    "Found %d public static methods without arguments with %s annotation. Will use %s",
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   338
                    paremetersProviders.size(), Parameters.class,
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   339
                    paremetersProvider));
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   340
            paremetersProviders.stream().map(Method::toString).forEach(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   341
                    TestBuilder::trace);
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   342
        }
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   343
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   344
        // Construct collection of arguments for test class instances.
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   345
        return ((Collection) paremetersProvider.invoke(null)).stream();
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   346
    }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   347
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   348
    private static Stream<MethodCall> toMethodCalls(Method method) throws
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   349
            IllegalAccessException, InvocationTargetException {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   350
        return toCtorArgs(method).map(v -> toMethodCalls(v, method)).flatMap(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   351
                s -> s).peek(methodCall -> {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   352
                    // Make sure required constructor is accessible if the one is needed.
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   353
                    // Need to probe all methods as some of them might be static
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   354
                    // and some class members.
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   355
                    // Only class members require ctors.
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   356
                    try {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   357
                        methodCall.checkRequiredConstructor();
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   358
                    } catch (NoSuchMethodException ex) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   359
                        throw new ParseException(ex.getMessage() + ".");
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   360
                    }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   361
                });
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   362
    }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   363
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   364
    private static Stream<MethodCall> toMethodCalls(Object[] ctorArgs, Method method) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   365
        if (!isParametrized(method)) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   366
            return Stream.of(new MethodCall(ctorArgs, method));
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   367
        }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   368
        Parameter[] annotations = getMethodParameters(method);
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   369
        if (annotations.length == 0) {
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   370
            return Stream.of(new MethodCall(ctorArgs, method));
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   371
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   372
        return Stream.of(annotations).map((a) -> {
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   373
            Class paramType = method.getParameterTypes()[0];
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   374
            final Object annotationValue;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   375
            if (!paramType.isArray()) {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   376
                annotationValue = fromString(a.value()[0], paramType);
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   377
            } else {
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   378
                Class paramComponentType = paramType.getComponentType();
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   379
                annotationValue = Array.newInstance(paramComponentType, a.value().length);
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   380
                var idx = new AtomicInteger(-1);
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   381
                Stream.of(a.value()).map(v -> fromString(v, paramComponentType)).sequential().forEach(
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   382
                        v -> Array.set(annotationValue, idx.incrementAndGet(), v));
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   383
            }
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   384
            return new MethodCall(ctorArgs, method, annotationValue);
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   385
        });
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   386
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   387
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   388
    private static Object fromString(String value, Class toType) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   389
        Function<String, Object> converter = conv.get(toType);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   390
        if (converter == null) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   391
            throw new RuntimeException(String.format(
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   392
                    "Failed to find a conversion of [%s] string to %s type",
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   393
                    value, toType));
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   394
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   395
        return converter.apply(value);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   396
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   397
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   398
    // Wraps Method.invike() into ThrowingRunnable.run()
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   399
    private ThrowingConsumer wrap(Method method) {
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   400
        return (test) -> {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   401
            Class methodClass = method.getDeclaringClass();
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   402
            String methodName = String.join(".", methodClass.getName(),
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   403
                    method.getName());
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   404
            TKit.log(String.format("[ CALL     ] %s()", methodName));
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   405
            if (!dryRun) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   406
                if (methodClass.isInstance(test)) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   407
                    method.invoke(test);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   408
                } else {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   409
                    method.invoke(null);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   410
                }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   411
            }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   412
        };
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   413
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   414
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   415
    private static class ParseException extends IllegalArgumentException {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   417
        ParseException(String msg) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   418
            super(msg);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   419
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   420
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   421
        void setContext(String badCmdLineArg) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   422
            this.badCmdLineArg = badCmdLineArg;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   423
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   424
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   425
        @Override
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   426
        public String getMessage() {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   427
            String msg = super.getMessage();
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   428
            if (badCmdLineArg != null) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   429
                msg = String.format("%s parameter=[%s]", msg, badCmdLineArg);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   430
            }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   431
            return msg;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   432
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   433
        private String badCmdLineArg;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   434
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   435
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   436
    static void trace(String msg) {
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   437
        if (TKit.VERBOSE_TEST_SETUP) {
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   438
            TKit.log(msg);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   439
        }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   440
    }
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   441
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   442
    private final Map<String, ThrowingConsumer<String>> argProcessors;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   443
    private Consumer<TestInstance> testConsumer;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   444
    private List<MethodCall> testGroup;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   445
    private List<ThrowingConsumer> beforeActions;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   446
    private List<ThrowingConsumer> afterActions;
58648
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   447
    private Set<String> excludedTests;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   448
    private Set<String> includedTests;
3bf53ffa9ae7 8232279 : Improve test helpers #2
herrick
parents: 58416
diff changeset
   449
    private String spaceSubstitute;
58416
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   450
    private boolean dryRun;
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   451
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   452
    private final static Map<Class, Function<String, Object>> conv = Map.of(
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   453
            boolean.class, Boolean::valueOf,
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   454
            Boolean.class, Boolean::valueOf,
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   455
            int.class, Integer::valueOf,
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   456
            Integer.class, Integer::valueOf,
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   457
            long.class, Long::valueOf,
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   458
            Long.class, Long::valueOf,
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   459
            String.class, String::valueOf);
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   460
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   461
    final static String CMDLINE_ARG_PREFIX = "--jpt-";
f09bf58c1f17 8231605: Improve test helpers
herrick
parents:
diff changeset
   462
}