Update unit tests for travis cli