aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--.gitignore1
-rw-r--r--src/main.nim6
-rw-r--r--tests/tmain.nim32
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 @@
1*.swp 1*.swp
2*~ 2*~
3nimcache/ 3nimcache/
4tmp/
4 5
5* 6*
6!*/ 7!*/
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 @@
17import os, streams, sugar 17import os, streams, sugar
18import bitreader, bitwriter, streamblock 18import bitreader, bitwriter, streamblock
19 19
20proc transform(operation: (BitReader, BitWriter) -> void, input, output: string) = 20proc transform*(operation: (BitReader, BitWriter) -> void, input, output: string) =
21 let inputStream = openFileStream(input, fmRead) 21 let inputStream = openFileStream(input, fmRead)
22 defer: inputStream.close() 22 defer: inputStream.close()
23 let outputStream = openFileStream(output, fmWrite) 23 let outputStream = openFileStream(output, fmWrite)
24 defer: outputStream.close() 24 defer: outputStream.close()
25 operation(inputStream.bitReader(), outputStream.bitWriter()) 25 operation(inputStream.bitReader(), outputStream.bitWriter())
26 26
27proc compress(bitReader: BitReader, bitWriter: BitWriter) = 27proc compress*(bitReader: BitReader, bitWriter: BitWriter) =
28 while not bitReader.atEnd(): 28 while not bitReader.atEnd():
29 let streamBlock = streamblock.readRaw(bitReader) 29 let streamBlock = streamblock.readRaw(bitReader)
30 streamBlock.writeSerialisedTo(bitWriter) 30 streamBlock.writeSerialisedTo(bitWriter)
31 bitWriter.flush() 31 bitWriter.flush()
32 32
33proc decompress(bitReader: BitReader, bitWriter: BitWriter) = 33proc decompress*(bitReader: BitReader, bitWriter: BitWriter) =
34 var hasMore = true 34 var hasMore = true
35 while hasMore: 35 while hasMore:
36 let streamBlock = streamblock.readSerialised(bitReader) 36 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 @@
1# gzip-like LZSS compressor
2# Copyright (C) 2018 Pacien TRAN-GIRARD
3#
4# This program is free software: you can redistribute it and/or modify
5# it under the terms of the GNU Affero General Public License as
6# published by the Free Software Foundation, either version 3 of the
7# License, or (at your option) any later version.
8#
9# This program is distributed in the hope that it will be useful,
10# but WITHOUT ANY WARRANTY; without even the implied warranty of
11# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12# GNU Affero General Public License for more details.
13#
14# You should have received a copy of the GNU Affero General Public License
15# along with this program. If not, see <https://www.gnu.org/licenses/>.
16
17import unittest, os, ospaths, osproc
18import main
19
20const tempDir = "tmp"
21
22suite "main":
23 setup: createDir(tempDir)
24 teardown: removeDir(tempDir)
25
26 test "identity":
27 let input = "license.md"
28 let intermediate = tempDir / "compressed"
29 let final = tempDir / "decompressed"
30 compress.transform(input, intermediate)
31 decompress.transform(intermediate, final)
32 check startProcess("cmp", args=[input, final], options={poUsePath}).waitForExit() == 0