Add unit tests on the Puppet plugin