Add unit tests to CLI module.