Fix shell output unittests

Since fix for 0ed431a went in we revert our unittests to the old (working) behavior.

Signed-off-by: Adam Stokes <adam.stokes@ubuntu.com>