From 32364632be7b5008dfcd92731e9a815265938d58 Mon Sep 17 00:00:00 2001
From: Josh Hagins <hagins.josh@gmail.com>
Date: Sat, 5 Sep 2015 13:32:19 -0400
Subject: [PATCH] Include Hbc::Staged module in Hbc::DSL::Preflight

---
 lib/hbc/dsl/preflight.rb        |  2 +
 test/cask/dsl/preflight_test.rb | 76 +++++++++++++++++++++++++++++++++
 2 files changed, 78 insertions(+)
 create mode 100644 test/cask/dsl/preflight_test.rb

diff --git a/lib/hbc/dsl/preflight.rb b/lib/hbc/dsl/preflight.rb
index 7fbdd30adbd..c687fb69658 100644
--- a/lib/hbc/dsl/preflight.rb
+++ b/lib/hbc/dsl/preflight.rb
@@ -1,4 +1,6 @@
 class Hbc::DSL::Preflight < Hbc::DSL::Base
+  include Hbc::Staged
+
   def method_missing(method, *args)
     Hbc::Utils.method_missing_message(method, @cask.to_s, 'preflight')
     return nil
diff --git a/test/cask/dsl/preflight_test.rb b/test/cask/dsl/preflight_test.rb
new file mode 100644
index 00000000000..eae127a2c45
--- /dev/null
+++ b/test/cask/dsl/preflight_test.rb
@@ -0,0 +1,76 @@
+require "test_helper"
+
+describe Hbc::DSL::Preflight do
+  before do
+    cask = Hbc.load('basic-cask')
+    @dsl = Hbc::DSL::Preflight.new(cask, Hbc::FakeSystemCommand)
+  end
+
+  it "can run system commands with list-form arguments" do
+    Hbc::FakeSystemCommand.expects_command(
+      ['echo', 'homebrew-cask', 'rocks!']
+    )
+    @dsl.system_command("echo", :args => ["homebrew-cask", "rocks!"])
+  end
+
+  it "can get the Info.plist file for the primary app" do
+    @dsl.info_plist_file.to_s.must_include 'basic-cask/1.2.3/TestCask.app/Contents/Info.plist'
+  end
+
+  it "can execute commands on the Info.plist file" do
+    @dsl.stubs(:bundle_identifier => 'com.example.BasicCask')
+
+    Hbc::FakeSystemCommand.expects_command(
+      ['/usr/libexec/PlistBuddy', '-c', 'Print CFBundleIdentifier', @dsl.info_plist_file]
+    )
+    @dsl.plist_exec('Print CFBundleIdentifier')
+  end
+
+  it "can set a key in the Info.plist file" do
+    @dsl.stubs(:bundle_identifier => 'com.example.BasicCask')
+
+    Hbc::FakeSystemCommand.expects_command(
+      ['/usr/libexec/PlistBuddy', '-c', 'Set :JVMOptions:JVMVersion 1.6+', @dsl.info_plist_file]
+    )
+    @dsl.plist_set(':JVMOptions:JVMVersion', '1.6+')
+  end
+
+  it "can set the permissions of a file" do
+    Hbc::FakeSystemCommand.expects_command(
+      ['/usr/bin/sudo', '-E', '--', '/bin/chmod', '-R', '--', '777', Pathname('/path/to/file')]
+    )
+    @dsl.set_permissions('/path/to/file', '777')
+  end
+
+  it "can set the permissions of multiple files" do
+    Hbc::FakeSystemCommand.expects_command(
+      ['/usr/bin/sudo', '-E', '--', '/bin/chmod', '-R', '--', '777', Pathname('/path/to/file'), Pathname('/path/to/other-file')]
+    )
+    @dsl.set_permissions(['/path/to/file', '/path/to/other-file'], '777')
+  end
+
+  it "can set the ownership of a file" do
+    @dsl.stubs(:current_user => 'fake_user')
+
+    Hbc::FakeSystemCommand.expects_command(
+      ['/usr/bin/sudo', '-E', '--', '/usr/sbin/chown', '-R', '--', 'fake_user:staff', Pathname('/path/to/file')]
+    )
+    @dsl.set_ownership('/path/to/file')
+  end
+
+  it "can set the ownership of multiple files" do
+    @dsl.stubs(:current_user => 'fake_user')
+
+    Hbc::FakeSystemCommand.expects_command(
+      ['/usr/bin/sudo', '-E', '--', '/usr/sbin/chown', '-R', '--', 'fake_user:staff', Pathname('/path/to/file'), Pathname('/path/to/other-file')]
+    )
+    @dsl.set_ownership(['/path/to/file', '/path/to/other-file'])
+  end
+
+  it "can set the ownership of a file with a different user and group" do
+    Hbc::FakeSystemCommand.expects_command(
+      ['/usr/bin/sudo', '-E', '--', '/usr/sbin/chown', '-R', '--', 'other_user:other_group', Pathname('/path/to/file')]
+    )
+    @dsl.set_ownership('/path/to/file', user: 'other_user', group: 'other_group')
+  end
+end