test_utils.lua

local pretty = require"pl.pretty"
local pl_path = require"pl.path"
local pl_tmpname = pl_path.tmpname
local pl_file = require"pl.file"

local _M = {}

_M.sh = function(...)
    local command = table.concat({...})
    print(command)
    local ret = os.execute(command)
    assert(ret == 0, "sh [" .. command .. "] exit with code: " .. ret)
end

local utils = require"pl.utils"

_M.stderr = function(...)
    local command = table.concat({...})
    print(command)
    local ok, status, _, stderr = utils.executeex(command, true)
    assert(ok, "stderr [" .. command .. "] fail")
    assert(status == 0, "stderr [" .. command .. "] exit with code: " .. status)
    stderr = stderr:gsub('^%s+', '')
    stderr = stderr:gsub('%s+$', '')
    print(stderr)
    return stderr
end

_M.stdout = function(...)
    local command = table.concat({...})
    print(command)
    local ok, status, stdout, stderr = utils.executeex(command)
    if not ok then
        print(stderr)
    end
    assert(ok, "stdout [" .. command .. "] fail")
    assert(status == 0, "stdout [" .. command .. "] exit with code: " .. status)
    stdout = stdout:gsub('^%s+', '')
    stdout = stdout:gsub('%s+$', '')
    print(stdout)
    return stdout
end

_M.sh_ex = function(...)
    local command = table.concat({...})
    print(command)
    local ok, status, stdout, stderr = utils.executeex(command, true)
    print(stderr)
    assert(ok, "sh_ex [" .. command .. "] fail")
    assert(status == 0, "stdout [" .. command .. "] exit with code: " .. status)
    stdout = stdout:gsub('^%s+', '')
    stdout = stdout:gsub('%s+$', '')
    print(stdout)
    stderr = stderr:gsub('^%s+', '')
    stderr = stderr:gsub('%s+$', '')
    return stdout, stderr
end

local sleep = function(n, quiet)
    if not quiet then print("Waiting " .. n .. " seconds ...") end
    os.execute("sleep " .. tonumber(n))
end

_M.sleep = sleep

_M.sh_until_ok = function(tries, ...)
    assert(tries and type(tries) == "number" and tries > 0)
    local command = table.concat({...})
    print(command)
    local ok, status, stdout, stderr
    for i = 1, tries  do
        ok, status, stdout, stderr = utils.executeex(command, true)
        if ok then
            break
        end
        sleep(1)
    end
    print(stderr)
    assert(ok, "sh_ex [" .. command .. "] fail")
    assert(status == 0, "stdout [" .. command .. "] exit with code: " .. status)
    stdout = stdout:gsub('^%s+', '')
    stdout = stdout:gsub('%s+$', '')
    print(stdout)
    stderr = stderr:gsub('^%s+', '')
    stderr = stderr:gsub('%s+$', '')
    return stdout, stderr
end

_M.dump_table_to_tmp_file = function(t)
    local tdump = "return " .. pretty.write(t) .. "\n"
    local tmp = pl_tmpname()
    pl_file.write(tmp, tdump)
    _M.sh("chmod 666 ", tmp)
    return tmp
end


return _M