Puppet: Writing tests and fixing methods