aboutsummaryrefslogtreecommitdiff
path: root/src/cz/crcs/ectester/reader/test/CommandTest.java
blob: b7728b6b7d25f2ce860920055b8d4ed97013cbe7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
package cz.crcs.ectester.reader.test;

import cz.crcs.ectester.common.test.Result;
import cz.crcs.ectester.common.test.Test;
import cz.crcs.ectester.common.test.TestException;
import cz.crcs.ectester.reader.command.Command;
import cz.crcs.ectester.reader.response.Response;

import javax.smartcardio.CardException;
import java.util.function.BiFunction;

/**
 * A simple test that runs one Command to get and evaluate one Response
 * to get a Result and compare it with the expected one.
 */
public class CommandTest extends Test {
    private BiFunction<Command, Response, Result> callback;
    private Command command;
    private Response response;

    public CommandTest(Command command, BiFunction<Command, Response, Result> callback) {
        this.command = command;
        this.callback = callback;
    }

    public CommandTest(Command command, Result.ExpectedValue expected, String ok, String nok) {
        this(command, (cmd, resp) -> {
            Result.Value resultValue = Result.Value.fromExpected(expected, resp.successful(), resp.error());
            return new Result(resultValue, resultValue.ok() ? ok : nok);
        });
    }

    public CommandTest(Command command, Result.ExpectedValue expected) {
        this(command, expected, null, null);
    }

    public Command getCommand() {
        return command;
    }

    public Response getResponse() {
        return response;
    }

    @Override
    public void run() throws TestException {
        if (hasRun)
            return;

        try {
            response = command.send();
        } catch (CardException e) {
            throw new TestException(e);
        }
        if (callback != null) {
            result = callback.apply(command, response);
        } else {
            if (response.successful()) {
                result = new Result(Result.Value.SUCCESS);
            } else {
                result = new Result(Result.Value.FAILURE);
            }
        }
        hasRun = true;
    }

    @Override
    public String getDescription() {
        return response.getDescription();
    }
}