spec/action_spec.rb in vagrant-shell-commander-0.2.0 vs spec/action_spec.rb in vagrant-shell-commander-0.3.0

@@ -1,28 +1,40 @@
 require 'spec_helper'
 
-describe VagrantShellCommander::Action do
+describe VagrantPlugins::ShellCommander::Action do
   let(:app) {double(call: true)}
   let(:machine) {double(action: true)}
   let(:env) {double(:[] => machine)}
   let(:cmd) {'cmd'}
   let(:sh) {double(after_share_folders: cmd)}
   let(:global_config) {double(sh: sh)}
   let(:action_env) {{global_config: global_config}}
   let(:subject) {described_class.new(app, env)}
 
   describe "#call" do
-    after(:each) do
-      subject.call(action_env)
-    end
-
     it "should call the next middleware" do
       expect(app).to receive(:call).with(action_env)
+
+      subject.call(action_env)
     end
 
-    it "should call SSHRun action of the current machine with the after_boot option as command" do
-      allow(env).to receive(:[]).with(:machine).and_return(machine)
+    describe "SSHRun call" do
+      it "should call SSHRun action of the current machine with the after_boot option as command" do
+        allow(env).to receive(:[]).with(:machine).and_return(machine)
+        
+        expect(machine).to receive(:action).with(:ssh_run, ssh_run_command: cmd, ssh_opts: {:extra_args=>[]})
+
+        subject.call(action_env)
+      end
       
-      expect(machine).to receive(:action).with(:ssh_run, ssh_run_command: cmd)
+      it "should not call SSHRun action if after_boot option is nil" do
+        sh = double(after_share_folders: nil)
+        global_config = double(sh: sh)
+        action_env = {global_config: global_config}
+        
+        expect(machine).not_to receive(:action)
+
+        subject.call(action_env)
+      end
     end
   end
 end