spec/invoker/power/setup_spec.rb in invoker-1.0.0 vs spec/invoker/power/setup_spec.rb in invoker-1.0.1

@@ -1,9 +1,11 @@
 require "spec_helper"
 
 describe "Setup" do
   before {
+    @original_verbosity = $VERBOSE
+    $VERBOSE = nil
     @old_config = Invoker::Power::Config::CONFIG_LOCATION
     Invoker::Power::Config.const_set(:CONFIG_LOCATION, "/tmp/.invoker")
 
     File.exists?(Invoker::Power::Config::CONFIG_LOCATION) &&
       File.delete(Invoker::Power::Config::CONFIG_LOCATION)
@@ -22,10 +24,11 @@
     Invoker::Power::Config.const_set(:CONFIG_LOCATION, @old_config)
 
     File.exists?(Invoker::Power::Setup::RESOLVER_FILE) &&
       File.delete(Invoker::Power::Setup::RESOLVER_FILE)
     Invoker::Power::Setup.const_set(:RESOLVER_FILE, @old_resolver)
+    $VERBOSE = @original_verbosity
   }
 
   describe "When no setup exists" do
     it "should create a config file with port etc" do
       setup = Invoker::Power::Setup.new()
@@ -40,10 +43,18 @@
       config.http_port.should.not == nil
       config.dns_port.should.not == nil
     end
   end
 
+  describe "setup on non osx systems" do
+    it "should not run setup" do
+      Invoker.expects(:ruby_platform).returns("i686-linux")
+      Invoker::Power::Setup.any_instance.expects(:check_if_setup_can_run?).never()
+      Invoker::Power::Setup.install
+    end
+  end
+
   describe "when a setup file exists" do
     it "should throw error about existing file" do
       File.open(Invoker::Power::Config::CONFIG_LOCATION, "w") {|fl|
         fl.write("foo test")
       }
@@ -80,8 +91,22 @@
         @setup.expects(:flush_dns_rules).never
         @setup.expects(:install_firewall).never
 
         @setup.setup_invoker
       end
+    end
+  end
+
+  describe "uninstalling firewall rules" do
+    it "should uninstall firewall rules and remove all files created by setup" do
+      setup = Invoker::Power::Setup.new
+
+      HighLine.any_instance.expects(:agree).returns(true)
+      setup.expects(:remove_resolver_file).once
+      setup.expects(:unload_firewall_rule).with(true).once
+      setup.expects(:flush_dns_rules).once
+      Invoker::Power::Config.expects(:delete).once
+
+      setup.uninstall_invoker
     end
   end
 end