Update tests to work with rspec-puppet 2.x