Okay you are missing this

    @puppetssh.stubs(:shell_command).returns(true)

before the assert in this test. Thats it.

Looks good to me.