Convert Pre/PostflightBlock test to spec.

2 files changed, 82 insertions, 0 deletions

diff --git a/Library/Homebrew/cask/spec/cask/artifact/postflight_block_spec.rb b/Library/Homebrew/cask/spec/cask/artifact/postflight_block_spec.rb
new file mode 100644
index 000000000..6f58afb2a
--- /dev/null
+++ b/Library/Homebrew/cask/spec/cask/artifact/postflight_block_spec.rb
@@ -0,0 +1,41 @@
+require "spec_helper"
+
+describe Hbc::Artifact::PostflightBlock do
+  describe "install_phase" do
+    it "calls the specified block after installing, passing a Cask mini-dsl" do
+      called = false
+      yielded_arg = nil
+
+      cask = Hbc::Cask.new("with-postflight") do
+        postflight do |c|
+          called = true
+          yielded_arg = c
+        end
+      end
+
+      described_class.new(cask).install_phase
+
+      expect(called).to be true
+      expect(yielded_arg).to be_kind_of(Hbc::DSL::Postflight)
+    end
+  end
+
+  describe "uninstall_phase" do
+    it "calls the specified block after uninstalling, passing a Cask mini-dsl" do
+      called = false
+      yielded_arg = nil
+
+      cask = Hbc::Cask.new("with-uninstall-postflight") do
+        uninstall_postflight do |c|
+          called = true
+          yielded_arg = c
+        end
+      end
+
+      described_class.new(cask).uninstall_phase
+
+      expect(called).to be true
+      expect(yielded_arg).to be_kind_of(Hbc::DSL::UninstallPostflight)
+    end
+  end
+end
diff --git a/Library/Homebrew/cask/spec/cask/artifact/preflight_block_spec.rb b/Library/Homebrew/cask/spec/cask/artifact/preflight_block_spec.rb
new file mode 100644
index 000000000..ac5313bca
--- /dev/null
+++ b/Library/Homebrew/cask/spec/cask/artifact/preflight_block_spec.rb
@@ -0,0 +1,41 @@
+require "spec_helper"
+
+describe Hbc::Artifact::PreflightBlock do
+  describe "install_phase" do
+    it "calls the specified block before installing, passing a Cask mini-dsl" do
+      called = false
+      yielded_arg = nil
+
+      cask = Hbc::Cask.new("with-preflight") do
+        preflight do |c|
+          called = true
+          yielded_arg = c
+        end
+      end
+
+      described_class.new(cask).install_phase
+
+      expect(called).to be true
+      expect(yielded_arg).to be_kind_of Hbc::DSL::Preflight
+    end
+  end
+
+  describe "uninstall_phase" do
+    it "calls the specified block before uninstalling, passing a Cask mini-dsl" do
+      called = false
+      yielded_arg = nil
+
+      cask = Hbc::Cask.new("with-uninstall-preflight") do
+        uninstall_preflight do |c|
+          called = true
+          yielded_arg = c
+        end
+      end
+
+      described_class.new(cask).uninstall_phase
+
+      expect(called).to be true
+      expect(yielded_arg).to be_kind_of Hbc::DSL::UninstallPreflight
+    end
+  end
+end