1 files changed, 43 insertions, 0 deletions
diff --git a/acceptance/.shared/kitchen_acceptance/libraries/kitchen.rb b/acceptance/.shared/kitchen_acceptance/libraries/kitchen.rb
new file mode 100644
index 0000000000..5380b7002c
--- /dev/null
+++ b/acceptance/.shared/kitchen_acceptance/libraries/kitchen.rb
@@ -0,0 +1,43 @@
+class Kitchen < Chef::Resource
+  resource_name :kitchen
+
+  property :command, String, name_property: true
+  property :driver, %w(ec2 vagrant), coerce: proc { |v| v.to_s }, default: lazy { ENV["KITCHEN_DRIVER"] || :ec2 }
+  property :instances, String, default: lazy { ENV["KITCHEN_INSTANCES"] }
+  property :kitchen_dir, String, default: Chef.node['chef-acceptance']['suite-dir']
+  property :chef_product, String, default: lazy {
+    ENV["KITCHEN_CHEF_PRODUCT"] ||
+    # If we're running the chef or chefdk projects in jenkins, pick up the project name.
+    (%w(chef chefdk).include?(ENV["PROJECT_NAME"]) ? ENV["PROJECT_NAME"] : "chef")
+  }
+  property :chef_channel, String, default: lazy {
+    ENV["KITCHEN_CHEF_CHANNEL"] ||
+    # Pick up current if we can't connect to artifactory
+    (ENV["ARTIFACTORY_USERNAME"] ? "unstable" : "current")
+  }
+  property :chef_version, String, default: lazy {
+    ENV["KITCHEN_CHEF_VERSION"] ||
+    # If we're running the chef or chefdk projects in jenkins, pick up the project name.
+    (ENV["PROJECT_NAME"] == chef_product ? ENV["OMNIBUS_BUILD_VERSION"] : nil) ||
+    "latest"
+  }
+  property :artifactory_username, String, default: lazy { ENV["ARTIFACTORY_USERNAME"] }
+  property :artifactory_password, String, default: lazy { ENV["ARTIFACTORY_PASSWORD"] }
+  property :env, Hash, default: {}
+
+  action :run do
+    execute "bundle exec kitchen #{command}#{instances ? " #{instances}" : ""}" do
+      cwd kitchen_dir
+      env({
+        "KITCHEN_DRIVER" => driver,
+        "KITCHEN_INSTANCES" => instances,
+        "KITCHEN_LOCAL_YAML" => ::File.expand_path("../../.kitchen.#{driver}.yml", __FILE__),
+        "KITCHEN_CHEF_PRODUCT" => chef_product,
+        "KITCHEN_CHEF_CHANNEL" => chef_channel,
+        "KITCHEN_CHEF_VERSION" => chef_version,
+        "ARTIFACTORY_USERNAME" => artifactory_username,
+        "ARTIFACTORY_PASSWORD" => artifactory_password
+      }.merge(new_resource.env))
+    end
+  end
+end