gpeng-throttler 0.0.3 → 0.1.0

data/VERSION

@@ -1 +1 @@
-0.0.3
+0.1.0

data/bin/throttler

@@ -12,14 +12,17 @@ require 'throttler'
 
 def help_text(exit_code = 0)
   #TODO add help text
-  "Need to add help text"
+  puts "Need to add help text"
 end
 
-if ARGV.size.zero? || ['-h', '--help', 'help']
+if ARGV.size.zero? || ['-h', '--help', 'help'].include?(ARGV.first)
   help_text
 else
   case ARGV[0]
-  when 'test'
-    puts Throttler.test
+  when 'throttle'
+    ARGV.shift
+    Throttler.throttle(*ARGV)
+  when 'restore'
+    Throttler.restore
   end
 end

data/lib/throttler.rb

@@ -2,8 +2,28 @@ $: << File.dirname(__FILE__)
 
 class Throttler
   class << self
-    def test
-      "The test is working..."
+    def throttle(kb_speed)
+      #TODO check the argument is a positive number
+      #TODO check that there are no exisiting ipfw rules that would be deleted by the restore command
+      #TODO delete pipe 1 in case this is called more than once without restoring
+
+      exec_command "sudo ipfw add 1 pipe 1 ip from any to any out src-port 80"
+      exec_command "sudo ipfw add 2 pipe 1 ip from any to any in src-port 80"
+      exec_command "sudo ipfw pipe 1 config bw #{kb_speed}Kbit/s"
+      exec_command "sudo ipfw pipe 2 config bw #{kb_speed}Kbit/s"
+      puts "Connection speed set to #{kb_speed}Kbit/s on port 80"
+    end
+    
+    def restore
+      #TODO add error message if this is called when no throttling is in place
+      exec_command "sudo ipfw delete 1"
+      exec_command "sudo ipfw delete 2"
+      puts "Connection speed restored"
+    end
+    
+    private
+    def exec_command(command)
+      `#{command}`
     end
   end
 end

data/spec/throttler_spec.rb

@@ -1,7 +1,47 @@
 require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
 
 describe "Throttler" do
-  it "fails" do
-    fail "hey buddy, you should probably rename this file and start specing for real"
+  before(:each) do
+    Throttler.stub!(:exec_command)
+    Throttler.stub!(:puts)
   end
+  
+  describe "throttle" do
+    
+    it "should add the pipes" do
+      Throttler.should_receive(:exec_command).with("sudo ipfw add 1 pipe 1 ip from any to any out src-port 80")
+      Throttler.should_receive(:exec_command).with("sudo ipfw add 2 pipe 1 ip from any to any in src-port 80")
+      Throttler.throttle 10
+    end
+    
+    it "should set the ipfw speed to 10Kbit/s" do
+      Throttler.should_receive(:exec_command).with("sudo ipfw pipe 1 config bw 10Kbit/s")
+      Throttler.should_receive(:exec_command).with("sudo ipfw pipe 2 config bw 10Kbit/s")
+      Throttler.throttle 10
+    end
+    
+
+    it "should return the message 'Connection speed set to *Kbit/s on port 80'" do
+      Throttler.should_receive(:puts).with("Connection speed set to 10Kbit/s on port 80")
+      Throttler.throttle 10
+    end
+  end
+  
+  describe "restore" do
+    it "should delete the ipfw with id 1" do
+      Throttler.should_receive(:exec_command).with("sudo ipfw delete 1")
+      Throttler.restore
+    end
+    
+    it "should delete the ipfw with id 2" do
+      Throttler.should_receive(:exec_command).with("sudo ipfw delete 2")
+      Throttler.restore
+    end
+    
+    it "should return the message 'Connection speed restored" do
+      Throttler.should_receive(:puts).with("Connection speed restored")
+      Throttler.restore
+    end
+  end
+  
 end

data/throttler.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{throttler}
-  s.version = "0.0.3"
+  s.version = "0.1.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Graham Pengelly"]
@@ -29,6 +29,7 @@ Gem::Specification.new do |s|
      "lib/throttler.rb",
      "spec/spec_helper.rb",
      "spec/throttler_spec.rb",
+     "throttler-0.0.3.gem",
      "throttler.gemspec"
   ]
   s.homepage = %q{http://github.com/gpeng/throttler}

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: gpeng-throttler
 version: !ruby/object:Gem::Version 
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors: 
 - Graham Pengelly
@@ -42,6 +42,7 @@ files:
 - lib/throttler.rb
 - spec/spec_helper.rb
 - spec/throttler_spec.rb
+- throttler-0.0.3.gem
 - throttler.gemspec
 has_rdoc: false
 homepage: http://github.com/gpeng/throttler