Puppet: Adding tests for puppet code