forked from eklitzke/lua-bz2
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathltn12-test.lua
65 lines (45 loc) · 1.78 KB
/
ltn12-test.lua
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
local ltn12 = require("ltn12")
local bz2_ltn12 = require("bz2.ltn12")
math.randomseed(42)
local data_stream = {}
for i = 1, 1000 do
data_stream[#data_stream + 1] = tostring(math.random()) .. "\n"
end
local data = table.concat(data_stream)
local filename = os.tmpname()
local source = ltn12.source.string(data)
local sink = bz2_ltn12.sink.file(filename)
assert(ltn12.pump.all(source, sink))
local source = bz2_ltn12.source.file(filename)
local sink, result_stream = ltn12.sink.table()
assert(ltn12.pump.all(source, sink))
local result = table.concat(result_stream)
assert(#data == #result, "Length does not match, expected:" .. #data .. " got " .. #result)
assert(data == result, "Data does not match")
os.remove(filename)
-- Test round-trim equality
local source = ltn12.source.string(data)
local sink, result_stream = ltn12.sink.table()
local filter = ltn12.filter.chain(
bz2_ltn12.filter.compress(),
bz2_ltn12.filter.decompress()
)
sink = ltn12.sink.chain(filter, sink)
assert(ltn12.pump.all(source, sink))
assert(#data == #result, "Length does not match, expected:" .. #data .. " got " .. #result)
assert(data == result, "Data does not match")
-- Tests mixed handling
local filename = os.tmpname()
local source = ltn12.source.string(data)
local sink = ltn12.sink.file(io.open(filename, "wb"))
local filter = bz2_ltn12.filter.compress()
sink = ltn12.sink.chain(filter, sink)
assert(ltn12.pump.all(source, sink))
local source = ltn12.source.file(io.open(filename, "rb"))
local sink, result_stream = ltn12.sink.table()
local filter = bz2_ltn12.filter.decompress()
sink = ltn12.sink.chain(filter, sink)
assert(ltn12.pump.all(source, sink))
assert(#data == #result, "Length does not match, expected:" .. #data .. " got " .. #result)
assert(data == result, "Data does not match")
os.remove(filename)