1 files changed, 65 insertions, 0 deletions
diff --git a/src/cz/crcs/ectester/common/output/TestableWriter.java b/src/cz/crcs/ectester/common/output/TestableWriter.java
new file mode 100644
index 0000000..9876064
--- /dev/null
+++ b/src/cz/crcs/ectester/common/output/TestableWriter.java
@@ -0,0 +1,65 @@
+package cz.crcs.ectester.common.output;
+
+import cz.crcs.ectester.common.test.BaseTestable;
+import cz.crcs.ectester.reader.output.ResponseWriter;
+import cz.crcs.ectester.reader.response.Response;
+import cz.crcs.ectester.reader.test.CommandTestable;
+import cz.crcs.ectester.standalone.test.KeyAgreementTestable;
+import cz.crcs.ectester.standalone.test.KeyGeneratorTestable;
+import cz.crcs.ectester.standalone.test.SignatureTestable;
+
+import java.io.OutputStream;
+import java.io.PrintStream;
+
+/**
+ * @author Jan Jancar johny@neuromancer.sk
+ */
+public class TestableWriter {
+    private PrintStream output;
+    private ResponseWriter respWriter;
+
+    public TestableWriter(PrintStream output) {
+        this.output = output;
+        this.respWriter = new ResponseWriter(output);
+    }
+
+    public TestableWriter(OutputStream output) {
+        this(new PrintStream(output));
+    }
+
+    public String outputTestableSuffix(BaseTestable t) {
+        if (t instanceof CommandTestable) {
+            Response r = ((CommandTestable) t).getResponse();
+            return respWriter.responseSuffix(r);
+        } else if (t instanceof KeyAgreementTestable) {
+
+        } else if (t instanceof KeyGeneratorTestable) {
+
+        } else if (t instanceof SignatureTestable) {
+
+        }
+        return null;
+    }
+
+    public void writeTestableSuffix(BaseTestable t) {
+        output.println(outputTestableSuffix(t));
+    }
+
+    public String outputTestable(BaseTestable t) {
+        if (t instanceof CommandTestable) {
+            CommandTestable testable = (CommandTestable) t;
+            return respWriter.responseString(testable.getResponse());
+        } else if (t instanceof KeyAgreementTestable) {
+
+        } else if (t instanceof KeyGeneratorTestable) {
+
+        } else if (t instanceof SignatureTestable) {
+
+        }
+        return null;
+    }
+
+    public void writeTestable(BaseTestable t) {
+        output.println(outputTestable(t));
+    }
+}