Add tests for the Puppet plugin