Introduce SimpleTest, implement other tests usign it.

6 files changed, 61 insertions, 67 deletions

diff --git a/src/cz/crcs/ectester/standalone/test/KeyAgreementTest.java b/src/cz/crcs/ectester/standalone/test/KeyAgreementTest.java
index a24346e..9604e75 100644
--- a/src/cz/crcs/ectester/standalone/test/KeyAgreementTest.java
+++ b/src/cz/crcs/ectester/standalone/test/KeyAgreementTest.java
@@ -1,7 +1,7 @@
 package cz.crcs.ectester.standalone.test;
 
 import cz.crcs.ectester.common.test.Result;
-import cz.crcs.ectester.common.test.Test;
+import cz.crcs.ectester.common.test.SimpleTest;
 import cz.crcs.ectester.common.test.TestCallback;
 import cz.crcs.ectester.common.test.TestException;
 
@@ -10,13 +10,9 @@ import java.util.Arrays;
 /**
  * @author Jan Jancar johny@neuromancer.sk
  */
-public class KeyAgreementTest extends Test {
-    private KeyAgreementTestable ka;
-    private TestCallback<KeyAgreementTestable> callback;
-
+public class KeyAgreementTest extends SimpleTest<KeyAgreementTestable> {
     private KeyAgreementTest(KeyAgreementTestable ka, TestCallback<KeyAgreementTestable> callback) {
-        this.ka = ka;
-        this.callback = callback;
+        super(ka, callback);
     }
 
     public static KeyAgreementTest match(KeyAgreementTestable ka, byte[] expectedSecret) {
@@ -52,8 +48,10 @@ public class KeyAgreementTest extends Test {
 
     @Override
     public void run() throws TestException {
-        ka.run();
-        result = callback.apply(ka);
+        if (hasRun)
+            return;
+        testable.run();
+        result = callback.apply(testable);
         hasRun = true;
     }
 }
diff --git a/src/cz/crcs/ectester/standalone/test/KeyAgreementTestable.java b/src/cz/crcs/ectester/standalone/test/KeyAgreementTestable.java
index fedf519..fc0f007 100644
--- a/src/cz/crcs/ectester/standalone/test/KeyAgreementTestable.java
+++ b/src/cz/crcs/ectester/standalone/test/KeyAgreementTestable.java
@@ -14,7 +14,7 @@ import java.security.interfaces.ECPublicKey;
  */
 public class KeyAgreementTestable extends BaseTestable {
     private KeyAgreement ka;
-    private ECPrivateKey privateKey;
+    private ECPrivateKey privateKey;a
     private ECPublicKey publicKey;
     private byte[] secret;
 
diff --git a/src/cz/crcs/ectester/standalone/test/KeyGenerationTest.java b/src/cz/crcs/ectester/standalone/test/KeyGenerationTest.java
deleted file mode 100644
index 0ec4741..0000000
--- a/src/cz/crcs/ectester/standalone/test/KeyGenerationTest.java
+++ /dev/null
@@ -1,44 +0,0 @@
-package cz.crcs.ectester.standalone.test;
-
-import cz.crcs.ectester.common.test.Result;
-import cz.crcs.ectester.common.test.Test;
-import cz.crcs.ectester.common.test.TestCallback;
-import cz.crcs.ectester.common.test.TestException;
-
-/**
- * @author Jan Jancar johny@neuromancer.sk
- */
-public class KeyGenerationTest extends Test {
-    private KeyGenerationTestable kg;
-    private TestCallback<KeyGenerationTestable> callback;
-
-    private KeyGenerationTest(KeyGenerationTestable kg, TestCallback<KeyGenerationTestable> callback) {
-        this.kg = kg;
-        this.callback = callback;
-    }
-
-    public static KeyGenerationTest expect(KeyGenerationTestable kg, Result.ExpectedValue expected) {
-        return new KeyGenerationTest(kg, new TestCallback<KeyGenerationTestable>() {
-            @Override
-            public Result apply(KeyGenerationTestable keyGenerationTestable) {
-                return new Result(Result.Value.fromExpected(expected, keyGenerationTestable.ok(), keyGenerationTestable.error()));
-            }
-        });
-    }
-
-    public static KeyGenerationTest function(KeyGenerationTestable ka, TestCallback<KeyGenerationTestable> callback) {
-        return new KeyGenerationTest(ka, callback);
-    }
-
-    @Override
-    public String getDescription() {
-        return null;
-    }
-
-    @Override
-    public void run() throws TestException {
-        kg.run();
-        result = callback.apply(kg);
-        hasRun = true;
-    }
-}
diff --git a/src/cz/crcs/ectester/standalone/test/KeyGeneratorTest.java b/src/cz/crcs/ectester/standalone/test/KeyGeneratorTest.java
new file mode 100644
index 0000000..9032415
--- /dev/null
+++ b/src/cz/crcs/ectester/standalone/test/KeyGeneratorTest.java
@@ -0,0 +1,42 @@
+package cz.crcs.ectester.standalone.test;
+
+import cz.crcs.ectester.common.test.Result;
+import cz.crcs.ectester.common.test.SimpleTest;
+import cz.crcs.ectester.common.test.TestCallback;
+import cz.crcs.ectester.common.test.TestException;
+
+/**
+ * @author Jan Jancar johny@neuromancer.sk
+ */
+public class KeyGeneratorTest extends SimpleTest<KeyGeneratorTestable> {
+    private KeyGeneratorTest(KeyGeneratorTestable kg, TestCallback<KeyGeneratorTestable> callback) {
+        super(kg, callback);
+    }
+
+    public static KeyGeneratorTest expect(KeyGeneratorTestable kg, Result.ExpectedValue expected) {
+        return new KeyGeneratorTest(kg, new TestCallback<KeyGeneratorTestable>() {
+            @Override
+            public Result apply(KeyGeneratorTestable keyGenerationTestable) {
+                return new Result(Result.Value.fromExpected(expected, keyGenerationTestable.ok(), keyGenerationTestable.error()));
+            }
+        });
+    }
+
+    public static KeyGeneratorTest function(KeyGeneratorTestable ka, TestCallback<KeyGeneratorTestable> callback) {
+        return new KeyGeneratorTest(ka, callback);
+    }
+
+    @Override
+    public String getDescription() {
+        return null;
+    }
+
+    @Override
+    public void run() throws TestException {
+        if (hasRun)
+            return;
+        testable.run();
+        result = callback.apply(testable);
+        hasRun = true;
+    }
+}
diff --git a/src/cz/crcs/ectester/standalone/test/KeyGenerationTestable.java b/src/cz/crcs/ectester/standalone/test/KeyGeneratorTestable.java
index 5a891b7..ca7f1e7 100644
--- a/src/cz/crcs/ectester/standalone/test/KeyGenerationTestable.java
+++ b/src/cz/crcs/ectester/standalone/test/KeyGeneratorTestable.java
@@ -8,22 +8,22 @@ import java.security.KeyPair;
 import java.security.KeyPairGenerator;
 import java.security.spec.ECParameterSpec;
 
-public class KeyGenerationTestable extends BaseTestable {
+public class KeyGeneratorTestable extends BaseTestable {
     private KeyPair kp;
     private KeyPairGenerator kpg;
     private int keysize = 0;
     private ECParameterSpec spec = null;
 
-    public KeyGenerationTestable(KeyPairGenerator kpg) {
+    public KeyGeneratorTestable(KeyPairGenerator kpg) {
         this.kpg = kpg;
     }
 
-    public KeyGenerationTestable(KeyPairGenerator kpg, int keysize) {
+    public KeyGeneratorTestable(KeyPairGenerator kpg, int keysize) {
         this.kpg = kpg;
         this.keysize = keysize;
     }
 
-    public KeyGenerationTestable(KeyPairGenerator kpg, ECParameterSpec spec) {
+    public KeyGeneratorTestable(KeyPairGenerator kpg, ECParameterSpec spec) {
         this.kpg = kpg;
         this.spec = spec;
     }
diff --git a/src/cz/crcs/ectester/standalone/test/SignatureTest.java b/src/cz/crcs/ectester/standalone/test/SignatureTest.java
index 828b84f..272a3f7 100644
--- a/src/cz/crcs/ectester/standalone/test/SignatureTest.java
+++ b/src/cz/crcs/ectester/standalone/test/SignatureTest.java
@@ -1,19 +1,15 @@
 package cz.crcs.ectester.standalone.test;
 
-import cz.crcs.ectester.common.test.Test;
+import cz.crcs.ectester.common.test.SimpleTest;
 import cz.crcs.ectester.common.test.TestCallback;
 import cz.crcs.ectester.common.test.TestException;
 
 /**
  * @author Jan Jancar johny@neuromancer.sk
  */
-public class SignatureTest extends Test {
-    private SignatureTestable sig;
-    private TestCallback<SignatureTestable> callback;
-
+public class SignatureTest extends SimpleTest<SignatureTestable> {
     private SignatureTest(SignatureTestable sig, TestCallback<SignatureTestable> callback) {
-        this.sig = sig;
-        this.callback = callback;
+        super(sig, callback);
     }
 
     @Override
@@ -23,8 +19,10 @@ public class SignatureTest extends Test {
 
     @Override
     public void run() throws TestException {
-        sig.run();
-        result = callback.apply(sig);
+        if (hasRun)
+            return;
+        testable.run();
+        result = callback.apply(testable);
         hasRun = true;
     }
 }