zstandard-lua/test.lua at master · BigOyayubi/zstandard-lua · GitHub

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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
require 'busted.runner'()

local zstd = require("zstd")
local readfile = function( path )
  local f = io.open(path)
  local s = f:read("*a")
  f:close()
  return s
end

local read_testfile = function( decompressed_filename )
    local decompressed = readfile(decompressed_filename)
    local decompressed_size = string.len(decompressed)

    local compressed = readfile(decompressed_filename .. '.zstd')
    local compressed_size = string.len(compressed)

    return decompressed, decompressed_size, compressed, compressed_size
end

local read_testfile2 = function( decompressed_file_prefix )
    local decompressed = readfile(decompressed_file_prefix .. ".txt" )
    local decompressed_size = string.len(decompressed)

    local compressed = readfile(decompressed_file_prefix .. ".zstd" )
    local compressed_size = string.len(compressed)

    local dictionary = readfile(decompressed_file_prefix .. ".dict")
    local dictionary_size = string.len(dictionary)

    return decompressed, decompressed_size, compressed, compressed_size, dictionary, dictionary_size
end

describe("decompress method test", function()
  it("return value check", function()
    local decompressed, decompressed_size, compressed, compressed_size = read_testfile( 'testdata/test1/test1.txt' )
    local dst = (" "):rep(decompressed_size)
    local result = zstd.decompress( dst, decompressed_size, compressed, compressed_size )

    assert.are.equal( result, decompressed_size )
  end)

  it("decompressed data check", function()
    local decompressed, decompressed_size, compressed, compressed_size = read_testfile( 'testdata/test1/test1.txt' )
    local dst = (" "):rep(decompressed_size)
    local result = zstd.decompress( dst, decompressed_size, compressed, compressed_size )

    assert.are.equal( dst, decompressed )
  end)

  it("return value isError check", function()
    local decompressed, decompressed_size, compressed, compressed_size = read_testfile( 'testdata/test1/test1.txt' )
    local dst = (" "):rep(decompressed_size)
    local result = zstd.decompress( dst, decompressed_size, compressed, compressed_size )
    local isError = zstd.isError( result )

    assert.are.equal( isError, false )
  end)

  it("decompress fail test", function()
    local src = "aaa"
    local dst = (" "):rep(100)
    local result = zstd.decompress( dst, #dst, src, #src )
    local isError = zstd.isError( result )

    assert.are.equal( isError, true )
  end)
end)

describe("decompress method with dictionary test", function()
  it("load dictionary test", function()
    local dst, dst_size, src, src_size, dict_data, dict_data_size = read_testfile2( 'testdata/test2/test2' )
    local dict = zstd.loadDecompressDictionary( dict_data, dict_data_size )
  end)

  it("decompress with dictionary test", function()
    local dst, dst_size, src, src_size, dict_data, dict_data_size = read_testfile2( 'testdata/test2/test2' )
    local dict = zstd.loadDecompressDictionary( dict_data, dict_data_size )
    local result = dict:decompress( dst, dst_size, src, src_size )
    local isError = zstd.isError( result )
    assert.are.equal( isError, false )
  end)
end)

describe("compress method test", function()
  it("compress and decompress", function()
    local text = ("aaa"):rep(100)
    local compressed = zstd.compress( text, string.len(text) )
    local dst = ("   "):rep(100)
    local result = zstd.decompress( dst, string.len(dst), compressed, string.len(compressed) )
    assert.are.equal( result, string.len(dst) )
    assert.are.equal( text, dst )
  end)
end)

describe("compress method with dictionary test", function()
  it("load dictionary test", function()
    local dst, dst_size, src, src_size, dict_data, dict_data_size = read_testfile2( 'testdata/test2/test2' )
    local dict = zstd.loadCompressDictionary( dict_data, dict_data_size, 1 )
  end)

  it("compress with dictionary test", function()
    local decomp, decomp_size, comp, comp_size, dict_data, dict_data_size = read_testfile2( 'testdata/test2/test2' )
    local compressdict = zstd.loadCompressDictionary( dict_data, dict_data_size, 3 )
    local compressed = compressdict:compress( decomp, decomp_size )
    local decompressdict = zstd.loadDecompressDictionary( dict_data, dict_data_size )
    local buf = (" "):rep(decomp_size)
    local result = decompressdict:decompress( buf, decomp_size, compressed, string.len(compressed) )
    assert.are.Equal( zstd.isError(result), false )
    assert.are.same( buf, decomp )
  end)
end)

describe("stream decompress test", function()
  it("create stream", function()
    local stream = zstd.createDecompressStream()
  end)
  it("set data", function()
    local stream = zstd.createDecompressStream()
    local text = ("aaa"):rep(100)
    stream:set( text, #text )
  end)
  it("decompress", function()
    local stream = zstd.createDecompressStream()
    local text = ("aaa"):rep(100)
    local compressed = zstd.compress( text, #text )
    stream:set( compressed, string.len(compressed) )
    local t = {}
    repeat
      local decomp, done, read_in, read_out = stream:decompress()
      table.insert(t, decomp)
    until done
    local decompressed = table.concat(t)
    assert.are.Equal( text, decompressed )
  end)

end)

describe("stream compress test", function()
  it("create stream", function()
    local stream = zstd.createCompressStream()
  end)
  it("set data", function()
    local stream = zstd.createCompressStream()
    local text = ("aaa"):rep(100)
    stream:set( text, #text )
  end)
  it("compress", function()
    local stream = zstd.createCompressStream()
    local text = ("aaa"):rep(100)
    stream:set( text, #text )
    local t = {}
    repeat
      local comp, done, read_in, read_out = stream:compress()
      table.insert(t, comp)
    until done
    local compressed = table.concat(t)
    local buf = (" "):rep( #text )
    local result = zstd.decompress( buf, #buf, compressed, string.len(compressed) )
    assert.are.Same( text, buf )
  end)
end)