PenTest/nmap
1
0
Fork 0
mirror of https://github.com/nmap/nmap.git synced 2025-12-15 20:29:03 +00:00
Code Issues Projects Releases Wiki Activity

Convert base32 and base64 tests to unittest tests

Browse Source
This commit is contained in:
dmiller
2018-08-28 16:58:02 +00:00
parent 880f883029
commit ed26487ae2
2 changed files with 97 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

89
nselib/base32.lua
View File

@@ -42,6 +42,8 @@ local char = require "string".char
local concat = require "table".concat
local unittest = require "unittest"
_ENV = require "stdnse".module "base32"
local b32standard = {
@@ -185,49 +187,56 @@ function dec (b32, hexExtend)
return concat(out)
end
do
local function test(a, b)
assert(enc(a) == b and dec(b) == a)
end
local function testh(a, b)
assert(enc(a, true) == b and dec(b, true) == a)
end
if not unittest.testing() then
return _ENV
end
test("", "")
test("f", "MY======")
test("fo", "MZXQ====")
test("foo", "MZXW6===")
test("foob", "MZXW6YQ=")
test("fooba", "MZXW6YTB")
test("foobar", "MZXW6YTBOI======")
testh("", "")
testh("f", "CO======")
testh("fo", "CPNG====")
testh("foo", "CPNMU===")
testh("foob", "CPNMUOG=")
testh("foobar", "CPNMUOJ1E8======")
test_suite = unittest.TestSuite:new()
-- extensive tests
if false then
local path = tmpname()
local file = open(path, "w")
local t = {}
for a = 0, 255, random(1, 7) do
for b = 0, 255, random(2, 7) do
for c = 0, 255, random(2, 7) do
t[#t+1] = char(a, b, c, 0xA)
file:write(t[#t])
end
end
end
assert(file:close())
local input = concat(t)
local output = enc(input)
local good = assert(popen("base32 < "..path, "r")):read("a"):gsub("%s", "")
remove(path)
assert(output == good)
assert(dec(output) == input)
local equal = unittest.equal
local function test(a, b)
test_suite:add_test(equal(enc(a), b), "encoding")
test_suite:add_test(equal(dec(b), a), "decoding")
end
local function testh(a, b)
test_suite:add_test(equal(enc(a, true), b), "hex encoding")
test_suite:add_test(equal(dec(b, true), a), "hex decoding")
end
test("", "")
test("f", "MY======")
test("fo", "MZXQ====")
test("foo", "MZXW6===")
test("foob", "MZXW6YQ=")
test("fooba", "MZXW6YTB")
test("foobar", "MZXW6YTBOI======")
testh("", "")
testh("f", "CO======")
testh("fo", "CPNG====")
testh("foo", "CPNMU===")
testh("foob", "CPNMUOG=")
testh("foobar", "CPNMUOJ1E8======")
-- extensive tests
if false then
local path = tmpname()
local file = open(path, "w")
local t = {}
for a = 0, 255, random(1, 7) do
for b = 0, 255, random(2, 7) do
for c = 0, 255, random(2, 7) do
t[#t+1] = char(a, b, c, 0xA)
file:write(t[#t])
end
end
end
assert(file:close())
local input = concat(t)
local output = enc(input)
local good = assert(popen("base32 < "..path, "r")):read("a"):gsub("%s", "")
remove(path)
assert(output == good)
assert(dec(output) == input)
end
return _ENV