spec/flydata/command/restart_spec.rb in flydata-0.3.24 vs spec/flydata/command/restart_spec.rb in flydata-0.4.0

@@ -4,14 +4,25 @@
 module Flydata
   module Command
     describe Restart do
       subject { described_class.new }
       let(:sender) { double("sender") }
+      let(:helper) { double("helper") }
+      let(:opts) { double("opts") }
 
-      it do
+      it "should restart sender and helper when skip-helper option is not passed" do
         expect(Flydata::Command::Sender).to receive(:new).and_return(sender)
         expect(sender).to receive(:restart)
+        expect(Flydata::Command::Helper).to receive(:new).and_return(helper)
+        expect(helper).to receive(:restart)
         subject.run
+      end
+
+      it "should only restart sender when skip-helper option is passed" do
+        expect(Flydata::Command::Sender).to receive(:new).and_return(sender)
+        expect(sender).to receive(:restart)
+        expect(opts).to receive(:skip_helper?).and_return(true)
+        described_class.new(opts).run
       end
     end
   end
 end