Add tests for arg

Signed-off-by: onevcat <onevcat@gmail.com>