3 files changed, 36 insertions, 3 deletions

diff --git a/.gitignore b/.gitignore
index 472e7ee..d9f0e73 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,6 +1,7 @@
 *.swp
 *~
 nimcache/
+tmp/
 
 *
 !*/
diff --git a/src/main.nim b/src/main.nim
index 8f0536c..4fc137f 100644
--- a/src/main.nim
+++ b/src/main.nim
@@ -17,20 +17,20 @@
 import os, streams, sugar
 import bitreader, bitwriter, streamblock
 
-proc transform(operation: (BitReader, BitWriter) -> void, input, output: string) =
+proc transform*(operation: (BitReader, BitWriter) -> void, input, output: string) =
   let inputStream = openFileStream(input, fmRead)
   defer: inputStream.close()
   let outputStream = openFileStream(output, fmWrite)
   defer: outputStream.close()
   operation(inputStream.bitReader(), outputStream.bitWriter())
 
-proc compress(bitReader: BitReader, bitWriter: BitWriter) =
+proc compress*(bitReader: BitReader, bitWriter: BitWriter) =
   while not bitReader.atEnd():
     let streamBlock = streamblock.readRaw(bitReader)
     streamBlock.writeSerialisedTo(bitWriter)
   bitWriter.flush()
 
-proc decompress(bitReader: BitReader, bitWriter: BitWriter) =
+proc decompress*(bitReader: BitReader, bitWriter: BitWriter) =
   var hasMore = true
   while hasMore:
     let streamBlock = streamblock.readSerialised(bitReader)
diff --git a/tests/tmain.nim b/tests/tmain.nim
new file mode 100644
index 0000000..2deb443
--- /dev/null
+++ b/tests/tmain.nim
@@ -0,0 +1,32 @@
+# gzip-like LZSS compressor
+# Copyright (C) 2018  Pacien TRAN-GIRARD
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+import unittest, os, ospaths, osproc
+import main
+
+const tempDir = "tmp"
+
+suite "main":
+  setup: createDir(tempDir)
+  teardown: removeDir(tempDir)
+
+  test "identity":
+    let input = "license.md"
+    let intermediate = tempDir / "compressed"
+    let final = tempDir / "decompressed"
+    compress.transform(input, intermediate)
+    decompress.transform(intermediate, final)
+    check startProcess("cmp", args=[input, final], options={poUsePath}).waitForExit() == 0