mirror of https://github.com/xzeldon/htop.git
496 lines
12 KiB
Lua
Executable File
496 lines
12 KiB
Lua
Executable File
#!/usr/bin/env lua
|
|
|
|
local VISUALDELAY = os.getenv("VISUALDELAY")
|
|
|
|
local visual = VISUALDELAY or false
|
|
local visual_delay = VISUALDELAY and (tonumber(VISUALDELAY)) or 0.1
|
|
|
|
local unistd = require("posix.unistd")
|
|
local time = require("posix.time")
|
|
local curses = require("posix.curses")
|
|
local rote = require("rote")
|
|
|
|
local rt = rote.RoteTerm(24, 80)
|
|
|
|
os.execute("make coverage")
|
|
os.execute("rm -f *.gcda */*.gcda")
|
|
os.execute("rm -f coverage.info test.htoprc")
|
|
os.execute("rm -rf lcov")
|
|
os.execute("killall htop")
|
|
os.execute("ps aux | grep '[s]leep 12345' | awk '{print $2}' | xargs kill 2> /dev/null")
|
|
|
|
os.execute("cp ./default.htoprc ./test.htoprc")
|
|
rt:forkPty("LC_ALL=C HTOPRC=./test.htoprc ./htop")
|
|
|
|
local stdscr, term_win
|
|
-- Curses initalization needed even when not in visual mode
|
|
-- because luaposix only initializes KEY_* constants after initscr().
|
|
stdscr = curses.initscr()
|
|
if visual then
|
|
curses.echo(false)
|
|
curses.start_color()
|
|
curses.raw(true)
|
|
curses.halfdelay(1)
|
|
stdscr:keypad(true)
|
|
term_win = curses.newwin(24, 80, 0, 0)
|
|
local function makePair(foreground, background)
|
|
return background * 8 + 7 - foreground
|
|
end
|
|
-- initialize the color pairs the way rt:draw() expects it
|
|
for foreground = 0, 7 do
|
|
for background = 0, 7 do
|
|
if foreground ~= 7 or background ~= 0 then
|
|
local pair = makePair(foreground, background)
|
|
curses.init_pair(pair, foreground, background)
|
|
end
|
|
end
|
|
end
|
|
else
|
|
curses.endwin()
|
|
end
|
|
|
|
local function delay(t)
|
|
time.nanosleep({ tv_sec = math.floor(t), tv_nsec = (t - math.floor(t)) * 1000000000 })
|
|
end
|
|
|
|
local function show(key)
|
|
rt:update()
|
|
if visual then
|
|
rt:draw(term_win, 0, 0)
|
|
if key then
|
|
term_win:mvaddstr(0, 0, tostring(key))
|
|
end
|
|
term_win:refresh()
|
|
|
|
delay(visual_delay)
|
|
end
|
|
end
|
|
|
|
local function send(key, times)
|
|
if times == 0 then return end
|
|
for _ = 1, times or 1 do
|
|
delay(0.003) -- 30ms delay to avoid clobbering Esc sequences
|
|
if type(key) == "string" then
|
|
for c in key:gmatch('.') do
|
|
rt:keyPress(string.byte(c))
|
|
end
|
|
else
|
|
rt:keyPress(key)
|
|
end
|
|
show(key)
|
|
end
|
|
end
|
|
|
|
local function string_at(x, y, len)
|
|
rt:update()
|
|
local out = {}
|
|
for i = 1, len do
|
|
out[#out+1] = rt:cellChar(y-1, x+i-2)
|
|
end
|
|
return table.concat(out)
|
|
end
|
|
|
|
local function is_string_at(x, y, str)
|
|
return string_at(x, y, #str) == str
|
|
end
|
|
|
|
local function check_string_at(x, y, str)
|
|
return { str, string_at(x, y, #str) }
|
|
end
|
|
|
|
local ESC = "\27\27"
|
|
|
|
time.nanosleep({ tv_sec = 0, tv_nsec = 150000000 }) -- give some time for htop to initialize.
|
|
|
|
local y_panelhdr = (function()
|
|
for y = 1, 24 do
|
|
if is_string_at(3, y, "PID") then
|
|
return y
|
|
end
|
|
end
|
|
end)() or 1
|
|
|
|
local x_metercol2 = 43
|
|
|
|
show()
|
|
|
|
os.execute("sleep 12345 &")
|
|
|
|
local function terminated()
|
|
return not os.execute("ps aux | grep -q '\\./[h]top'")
|
|
end
|
|
|
|
local function running_it(desc, fn)
|
|
it(desc, function()
|
|
assert(not terminated())
|
|
show()
|
|
fn()
|
|
assert(not terminated())
|
|
end)
|
|
end
|
|
|
|
local function check(t)
|
|
return t[1], t[2]
|
|
end
|
|
|
|
local attrs = {
|
|
black_on_cyan = 6,
|
|
red_on_cyan = 22,
|
|
}
|
|
|
|
local function find_selected_y(from)
|
|
rt:update()
|
|
for y = from or (y_panelhdr + 1), rt:rows() - 1 do
|
|
local attr = rt:cellAttr(y-1, 1)
|
|
if attr == attrs.black_on_cyan then
|
|
return y
|
|
end
|
|
end
|
|
return y_panelhdr + 1
|
|
end
|
|
|
|
local function find_command_x()
|
|
for x = 1, 80 do
|
|
if is_string_at(x, y_panelhdr, "Command") then
|
|
return x
|
|
end
|
|
end
|
|
return 64
|
|
end
|
|
|
|
describe("htop test suite", function()
|
|
|
|
running_it("performs incremental filter", function()
|
|
send("\\")
|
|
send("x\127bux\127sted") -- test backspace
|
|
send("\n")
|
|
delay(0.2)
|
|
rt:update()
|
|
local pid = (" "..tostring(unistd.getpid())):sub(-5)
|
|
local ourpid = check_string_at(1, y_panelhdr + 1, pid)
|
|
send("\\")
|
|
send(ESC)
|
|
send(curses.KEY_F5)
|
|
send(curses.KEY_HOME)
|
|
delay(0.15)
|
|
rt:update()
|
|
local initpid = check_string_at(1, y_panelhdr + 1, " 1")
|
|
delay(0.15)
|
|
rt:update()
|
|
send(curses.KEY_F5)
|
|
assert.equal(check(ourpid))
|
|
assert.equal(check(initpid))
|
|
end)
|
|
|
|
running_it("performs incremental search", function()
|
|
send(curses.KEY_HOME)
|
|
send("/")
|
|
send("busted")
|
|
local attr = rt:cellAttr(rt:rows() - 1, 30)
|
|
send("\n")
|
|
delay(0.4)
|
|
local line = find_selected_y()
|
|
local pid = (" "..tostring(unistd.getpid())):sub(-5)
|
|
assert.equal(attr, attrs.black_on_cyan)
|
|
local ourpid = check_string_at(1, line, pid)
|
|
send(curses.KEY_HOME)
|
|
assert.equal(check(ourpid))
|
|
end)
|
|
|
|
running_it("kills a process", function()
|
|
send(curses.KEY_HOME)
|
|
send("\\")
|
|
send("sleep 12345")
|
|
local attr = rt:cellAttr(rt:rows() - 1, 30)
|
|
assert.equal(attr, attrs.black_on_cyan)
|
|
send("\n")
|
|
delay(0.3)
|
|
rt:update()
|
|
local col = find_command_x()
|
|
local procname = check_string_at(col, y_panelhdr + 1, "sleep 12345")
|
|
send("k")
|
|
send("\n")
|
|
send("\\")
|
|
send(ESC)
|
|
delay(0.3)
|
|
assert.equal(check(procname))
|
|
assert.not_equal((os.execute("ps aux | grep -q '[s]leep 12345'")), true)
|
|
end)
|
|
|
|
running_it("runs strace", function()
|
|
send(curses.KEY_HOME)
|
|
send("/")
|
|
send("busted")
|
|
send("\n")
|
|
send("s")
|
|
delay(1)
|
|
send(ESC)
|
|
end)
|
|
|
|
running_it("runs lsof", function()
|
|
send(curses.KEY_HOME)
|
|
send("/")
|
|
send("busted")
|
|
send("\n")
|
|
send("l")
|
|
delay(1)
|
|
send(ESC)
|
|
end)
|
|
|
|
running_it("performs filtering in lsof", function()
|
|
send(curses.KEY_HOME)
|
|
send("/")
|
|
send("htop")
|
|
send("\n")
|
|
send("l")
|
|
send(curses.KEY_F4)
|
|
send("pipe")
|
|
delay(1)
|
|
local pipefd = check_string_at(1, 3, " 3")
|
|
send(ESC)
|
|
assert.equal(check(pipefd))
|
|
end)
|
|
|
|
running_it("performs search in lsof", function()
|
|
send(curses.KEY_HOME)
|
|
send("/")
|
|
send("htop")
|
|
send("\n")
|
|
send("l")
|
|
send(curses.KEY_F3)
|
|
send("pipe")
|
|
delay(1)
|
|
local line = find_selected_y(3)
|
|
local pipefd = check_string_at(1, line, " 3")
|
|
send(ESC)
|
|
assert.equal(check(pipefd))
|
|
end)
|
|
|
|
|
|
running_it("cycles through meter modes in the default meters", function()
|
|
send("S")
|
|
for _ = 1, 2 do
|
|
send(curses.KEY_RIGHT)
|
|
for _ = 1, 3 do
|
|
send("\n", 4)
|
|
send(curses.KEY_DOWN)
|
|
end
|
|
end
|
|
send(ESC)
|
|
end)
|
|
|
|
running_it("show process of a user", function()
|
|
send(curses.KEY_F5)
|
|
send("u")
|
|
send(curses.KEY_DOWN)
|
|
delay(0.3)
|
|
rt:update()
|
|
local chosen = string_at(1, y_panelhdr + 2, 9)
|
|
send("\n")
|
|
send(curses.KEY_HOME)
|
|
delay(0.3)
|
|
rt:update()
|
|
local shown = string_at(7, y_panelhdr + 1, 9)
|
|
send("u")
|
|
send("\n")
|
|
send(curses.KEY_HOME)
|
|
delay(0.3)
|
|
rt:update()
|
|
local inituser = string_at(7, y_panelhdr + 1, 9)
|
|
send(curses.KEY_F5)
|
|
assert.equal(shown, chosen)
|
|
assert.equal(inituser, "root ")
|
|
end)
|
|
|
|
running_it("performs failing search", function()
|
|
send(curses.KEY_HOME)
|
|
send("/")
|
|
send("xxxxxxxxxx")
|
|
delay(0.3)
|
|
rt:update()
|
|
local attr = rt:cellAttr(rt:rows() - 1, 30)
|
|
assert.equal(attr, attrs.red_on_cyan)
|
|
send("\n")
|
|
end)
|
|
|
|
running_it("cycles through search", function()
|
|
send(curses.KEY_HOME)
|
|
send("/")
|
|
send("sh")
|
|
local lastpid
|
|
local pidpairs = {}
|
|
for _ = 1, 3 do
|
|
send(curses.KEY_F3)
|
|
local line = find_selected_y()
|
|
local pid = string_at(1, line, 5)
|
|
if lastpid then
|
|
pidpairs[#pidpairs + 1] = { lastpid, pid }
|
|
lastpid = pid
|
|
end
|
|
end
|
|
send(curses.KEY_HOME)
|
|
for _, pair in pairs(pidpairs) do
|
|
assert.not_equal(pair[1], pair[2])
|
|
end
|
|
end)
|
|
|
|
running_it("visits each setup screen", function()
|
|
send("S")
|
|
send(curses.KEY_DOWN, 3)
|
|
send(curses.KEY_F10)
|
|
end)
|
|
|
|
running_it("adds and removes PPID column", function()
|
|
send("S")
|
|
send(curses.KEY_DOWN, 3)
|
|
send(curses.KEY_RIGHT, 2)
|
|
send(curses.KEY_DOWN, 2)
|
|
send("\n")
|
|
send(curses.KEY_F10)
|
|
delay(0.2)
|
|
local ppid = check_string_at(2, y_panelhdr, "PPID")
|
|
send("S")
|
|
send(curses.KEY_DOWN, 3)
|
|
send(curses.KEY_RIGHT, 1)
|
|
send(curses.KEY_DC)
|
|
send(curses.KEY_F10)
|
|
delay(0.2)
|
|
local not_ppid = check_string_at(2, y_panelhdr, "PPID")
|
|
assert.equal(check(ppid))
|
|
assert.not_equal(check(not_ppid))
|
|
end)
|
|
|
|
running_it("changes CPU affinity for a process", function()
|
|
send("a")
|
|
send(" \n")
|
|
send(ESC)
|
|
end)
|
|
|
|
running_it("changes IO priority for a process", function()
|
|
send("/")
|
|
send("htop")
|
|
send("\n")
|
|
send("i")
|
|
send(curses.KEY_END)
|
|
send("\n")
|
|
send(ESC)
|
|
end)
|
|
|
|
local meters = {
|
|
{ name = "clock", down = 0, string = "Time" },
|
|
{ name = "load", down = 2, string = "Load" },
|
|
{ name = "battery", down = 7, string = "Battery" },
|
|
{ name = "hostname", down = 8, string = "Hostname" },
|
|
}
|
|
|
|
for _, item in ipairs(meters) do
|
|
running_it("adds and removes a "..item.name.." widget", function()
|
|
send("S")
|
|
send(curses.KEY_RIGHT, 3)
|
|
send(curses.KEY_DOWN, item.down)
|
|
if branch == "wip" then
|
|
send("\n")
|
|
send(curses.KEY_UP, 4)
|
|
send("\n")
|
|
else
|
|
send(curses.KEY_F6)
|
|
send(curses.KEY_LEFT)
|
|
send(curses.KEY_DOWN, 4)
|
|
send(curses.KEY_F7, 4)
|
|
end
|
|
send(curses.KEY_F4, 4) -- cycle through meter modes
|
|
delay(0.15)
|
|
rt:update()
|
|
local with = check_string_at(x_metercol2, 2, item.string)
|
|
send(curses.KEY_DC)
|
|
delay(0.15)
|
|
local without = check_string_at(x_metercol2, 2, item.string)
|
|
send(curses.KEY_F10)
|
|
assert.equal(check(with))
|
|
assert.not_equal(check(without))
|
|
end)
|
|
end
|
|
|
|
running_it("goes through themes", function()
|
|
send(curses.KEY_F2)
|
|
send(curses.KEY_DOWN, 2)
|
|
send(curses.KEY_RIGHT)
|
|
for _ = 1, 6 do
|
|
send("\n")
|
|
send(curses.KEY_DOWN)
|
|
end
|
|
send(curses.KEY_UP, 6)
|
|
send("\n")
|
|
send(curses.KEY_F10)
|
|
end)
|
|
|
|
local display_options = {
|
|
{ name = "tree view", down = 0 },
|
|
{ name = "shadow other user's process", down = 1 },
|
|
{ name = "hide kernel threads", down = 2 },
|
|
{ name = "hide userland threads", down = 3 },
|
|
{ name = "display threads in different color", down = 4 },
|
|
{ name = "show custom thread names", down = 5 },
|
|
{ name = "highlight basename", down = 6 },
|
|
{ name = "highlight large numbers", down = 7 },
|
|
{ name = "leave margin around header", down = 8 },
|
|
{ name = "use detailed CPU time", down = 9 },
|
|
{ name = "count from zero", down = 10 },
|
|
{ name = "update process names", down = 11 },
|
|
{ name = "guest time in CPU%", down = 12 },
|
|
}
|
|
|
|
for _, item in ipairs(display_options) do
|
|
running_it("checks display option to "..item.name, function()
|
|
for _ = 1, 2 do
|
|
send("S")
|
|
send(curses.KEY_DOWN)
|
|
send(curses.KEY_RIGHT)
|
|
send(curses.KEY_DOWN, item.down)
|
|
send("\n")
|
|
send(curses.KEY_F10)
|
|
delay(0.1)
|
|
end
|
|
end)
|
|
end
|
|
|
|
running_it("shows detailed CPU with guest time", function()
|
|
for _ = 1, 2 do
|
|
send("S")
|
|
send(curses.KEY_DOWN)
|
|
send(curses.KEY_RIGHT)
|
|
send(curses.KEY_DOWN, 9)
|
|
send("\n")
|
|
send(curses.KEY_DOWN, 3)
|
|
send("\n")
|
|
send(curses.KEY_F10)
|
|
delay(0.1)
|
|
end
|
|
end)
|
|
|
|
for i = 1, 62 do
|
|
running_it("show column "..i, function()
|
|
send("S")
|
|
send(curses.KEY_END)
|
|
send(curses.KEY_RIGHT, 2)
|
|
send(curses.KEY_DOWN, i)
|
|
send("\n")
|
|
send(curses.KEY_F10)
|
|
end)
|
|
end
|
|
|
|
it("finally quits", function()
|
|
assert(not terminated())
|
|
send("q")
|
|
while not terminated() do
|
|
unistd.sleep(1)
|
|
end
|
|
assert(terminated())
|
|
if visual then
|
|
curses.endwin()
|
|
end
|
|
os.execute("make lcov && xdg-open lcov/index.html")
|
|
end)
|
|
end)
|
|
|