trema 0.1.3

data/spec/trema/stats-request_spec.rb

@@ -0,0 +1,151 @@
+#
+# Author: Nick Karanatsios <nickkaranatsios@gmail.com>
+#
+# Copyright (C) 2008-2012 NEC Corporation
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2, as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+
+
+require File.join( File.dirname( __FILE__ ), "..", "spec_helper" )
+require "trema"
+
+
+shared_examples_for "any stats-request" do
+  it_should_behave_like "any Openflow message with default transaction ID"
+  its( :flags ) { should == 0 }
+end
+
+
+describe StatsRequest do
+  context "when .FlowStatsRequest.new( MANDATORY OPTION MISSING )" do
+    subject { FlowStatsRequest.new }
+    it "should raise ArgumentError" do
+      expect { subject }.to raise_error( ArgumentError ) 
+    end
+  end
+
+
+  context "when .FlowStatsRequest.new( OPTIONAL OPTIONS MISSING )" do
+    subject { FlowStatsRequest.new( :match => Match.new( :dl_type => 0x800, :nw_proto => 17 ) ) }
+    it_should_behave_like "any stats-request"
+    its( :match ) { should be_an_instance_of( Match ) }
+    its( :table_id ) { should == 0xff }
+    its( :out_port ) { should == 0xffff }
+  end
+
+
+  context "when .FlowStatsRequest.new( VALID OPTIONS )" do
+    subject do 
+      FlowStatsRequest.new( 
+        :match => Match.new( :dl_type => 0x800, :nw_proto => 17 ),
+        :table_id => 1,
+        :out_port => 2
+      )
+    end
+    it_should_behave_like "any stats-request"
+    its( :match ) { should be_an_instance_of( Match ) }
+    its( :table_id ) { should == 1 }
+    its( :out_port ) { should == 2 }
+  end
+
+
+  context "when .AggregateStatsRequest.new( MANDATORY OPTION MISSING )" do
+    subject { AggregateStatsRequest.new }
+    it "should raise ArgumentError" do
+      expect { subject }.to raise_error( ArgumentError ) 
+    end
+  end
+
+
+  context "when .AggregateStatsRequest.new( OPTIONAL OPTIONS MISSING )" do
+    subject { AggregateStatsRequest.new( :match => Match.new( :dl_type => 0x800, :nw_proto => 17 ) ) }
+    it_should_behave_like "any stats-request"
+    its( :match ) { should be_an_instance_of( Match ) }
+    its( :table_id ) { should == 0xff }
+    its( :out_port ) { should == 0xffff }
+  end
+
+
+  context "when .AggregateStatsRequest.new( VALID OPTIONS )" do
+    subject do 
+      AggregateStatsRequest.new( 
+        :match => Match.new( :dl_type => 0x800, :nw_proto => 17 ),
+        :table_id => 1,
+        :out_port => 2
+      )
+    end
+    it_should_behave_like "any stats-request"
+    its( :match ) { should be_an_instance_of( Match ) }
+    its( :table_id ) { should == 1 }
+    its( :out_port ) { should == 2 }
+  end
+
+
+  context "when .TableStatsRequest.new( VALID OPTIONS )" do
+    subject { TableStatsRequest.new }
+    it_should_behave_like "any stats-request"
+  end
+
+
+  context "when .PortStasRequest.new( OPTIONAL OPTION MISSING )" do
+    subject { PortStatsRequest.new }
+    it_should_behave_like "any stats-request"
+    its( :port_no ) { should == 0xffff }
+  end
+
+
+  context "when .PortStasRequest.new( VALID OPTIONS )" do
+    subject { PortStatsRequest.new :port_no => 1 }
+    it_should_behave_like "any stats-request"
+    its( :port_no ) { should == 1 }
+  end
+
+
+  context "when .QueueStatsRequest.new( OPTIONAL OPTIONS MISSING )" do
+    subject { QueueStatsRequest.new }
+    it_should_behave_like "any stats-request"
+    its( :port_no ) { should == 0xfffc }
+    its( :queue_id ) { should == 0xffffffff }
+  end
+
+
+  context "when .QueueStatsRequest.new( VALID OPTIONS )" do
+    subject { QueueStatsRequest.new :port_no => 1, :queue_id => 2 }
+    it_should_behave_like "any stats-request"
+    its( :port_no ) { should == 1 }
+    its( :queue_id ) { should == 2 }
+  end
+
+
+  context "when .VendorStatsRequest.new( OPTIONAL OPTION MISSING )" do
+    subject { VendorStatsRequest.new }
+    it_should_behave_like "any stats-request"
+    its( :vendor_id ) { should == 0x00004cff }
+  end
+
+
+  context "when .VendorStatsRequest.new( VALID OPTION )" do
+    subject { VendorStatsRequest.new :vendor_id => 123 }
+    it_should_behave_like "any stats-request"
+    its( :vendor_id ) { should == 123 }
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/switch-disconnected_spec.rb

@@ -0,0 +1,58 @@
+#
+# Author: Nick Karanatsios <nickkaranatsios@gmail.com>
+#
+# Copyright (C) 2008-2012 NEC Corporation
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2, as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+
+
+require File.join( File.dirname( __FILE__ ), "..", "spec_helper" )
+require "trema"
+
+
+class SwitchDisconnected < Controller; end
+
+
+describe SwitchDisconnected do
+  context "when switch is killed" do
+    it "should receive a switch_disconnected" do
+      network {
+        vswitch( "switch-disconnect" ) { datapath_id 0xabc }
+      }.run( SwitchDisconnected ) {
+        controller( "SwitchDisconnected" ).should_receive( :switch_disconnected )
+        vswitch( "switch-disconnect" ).shutdown!
+        sleep 3 # FIXME: wait to shutdown
+      }
+    end
+
+
+    it "should receive a switch_disconnected with valid datapath_id" do
+      network {
+        vswitch( "switch-disconnect" ) { datapath_id 0xabc }
+      }.run( SwitchDisconnected ) {
+        controller( "SwitchDisconnected" ).should_receive( :switch_disconnected ).with( 0xabc )
+        vswitch( "switch-disconnect" ).shutdown!
+        sleep 3 # FIXME: wait to shutdown
+      }
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/switch-manager_spec.rb

@@ -0,0 +1,43 @@
+#
+# Author: Yasuhito Takamiya <yasuhito@gmail.com>
+#
+# Copyright (C) 2008-2012 NEC Corporation
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2, as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+
+
+require File.join( File.dirname( __FILE__ ), "..", "spec_helper" )
+require "trema/switch-manager"
+
+
+module Trema
+  describe SwitchManager do
+    it "should run switch_manager command with proper options" do
+      rule = { :port_status => "topology", :packet_in => "controller", :state_notify => "topology", :vendor => "controller" }
+      switch_manager = SwitchManager.new( rule )
+
+      switch_manager.should_receive( :sh ).once.with( /port_status::topology packet_in::controller state_notify::topology vendor::controller$/ )
+
+      switch_manager.run!
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/tremashark_spec.rb

@@ -0,0 +1,41 @@
+#
+# Author: Yasuhito Takamiya <yasuhito@gmail.com>
+#
+# Copyright (C) 2008-2012 NEC Corporation
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2, as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+
+
+require File.join( File.dirname( __FILE__ ), "..", "spec_helper" )
+require "trema/tremashark"
+
+
+module Trema
+  describe Tremashark do
+    it "should run tremashark command with proper options" do
+      tremashark = Tremashark.new
+      tremashark.should_receive( :sh ).once.with( /tremashark \-\-daemonize$/ )
+
+      tremashark.run
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/util_spec.rb

@@ -0,0 +1,93 @@
+#
+# Author: Yasuhito Takamiya <yasuhito@gmail.com>
+#
+# Copyright (C) 2008-2012 NEC Corporation
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2, as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+
+
+require File.join( File.dirname( __FILE__ ), "..", "spec_helper" )
+require "trema/dsl/configuration"
+require "trema/dsl/runner"
+require "trema/util"
+
+
+include Trema::Util
+
+
+describe Trema::Util do
+  it "should check the sanity of trema" do
+    Trema::Executables.stub!( :compiled? ).and_return( false )
+    $stderr.should_receive( :puts ).with( /^ERROR/ )
+    lambda do
+      sanity_check
+    end.should raise_error( SystemExit )
+  end
+
+
+  it "should execute and check the results of a command" do
+    lambda do
+      sh "NO SUCH COMMAND"
+    end.should raise_error( "Command 'NO SUCH COMMAND' failed!" )
+  end
+
+
+  it "should cleanup current session" do
+    apps = { "app 1" => mock( "app 1" ), "app 2" => mock( "app 2" ), "app 3" => mock( "app 3" ) }
+    apps.each do | name, app |
+      app.should_receive( :shutdown! )
+    end
+
+    switches = { "switch 1" => mock( "switch 1" ), "switch 2" => mock( "switch 2" ), "switch 3" => mock( "switch 3" ) }
+    switches.each do | name, switch |
+      switch.should_receive( :shutdown! )
+    end
+
+    hosts = { "host 1" => mock( "host 1" ), "host 2" => mock( "host 2" ), "host 3" => mock( "host 3" ) }
+    hosts.each do | name, host |
+      host.should_receive( :shutdown! )
+    end
+
+    links = { "link 1" => mock( "link 1" ), "link 2" => mock( "link 2" ), "link 3" => mock( "link 3" ) }
+    links.each do | name, link |
+      link.should_receive( :delete! )
+    end
+
+    last_session = mock( "last session" )
+    last_session.stub!( :apps ).and_return( apps )
+    last_session.stub!( :switches ).and_return( switches )
+    last_session.stub!( :hosts ).and_return( hosts )
+    last_session.stub!( :links ).and_return( links )
+    Trema::DSL::Context.stub!( :load_current ).and_return( last_session )
+
+    pid_files = [ mock( "PID file #0" ), mock( "PID file #1" ), mock( "PID file #2" ) ]
+    Dir.stub!( :glob ).and_return( pid_files )
+
+    process = mock( "process" )
+    process.should_receive( :kill! ).exactly( 3 ).times
+    Trema::Process.should_receive( :read ).with( pid_files[ 0 ] ).once.ordered.and_return( process )
+    Trema::Process.should_receive( :read ).with( pid_files[ 1 ] ).once.ordered.and_return( process )
+    Trema::Process.should_receive( :read ).with( pid_files[ 2 ] ).once.ordered.and_return( process )
+
+    cleanup_current_session
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/vendor-request_spec.rb

@@ -0,0 +1,79 @@
+#
+# Author: Nick Karanatsios <nickkaranatsios@gmail.com>
+#
+# Copyright (C) 2008-2012 NEC Corporation
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2, as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+
+
+require File.join( File.dirname( __FILE__ ), "..", "spec_helper" )
+require "trema"
+
+
+describe VendorRequest, ".new( OPTIONAL OPTION MISSING )" do
+  its( :vendor ) { should == 0xccddeeff }
+  its( :data ) { should have( 16 ).items  }
+  it_should_behave_like "any Openflow message with default transaction ID"
+end
+
+
+describe VendorRequest, ".new( VALID OPTIONS )" do
+  subject {
+    @vendor_data = "this is a test".unpack( "C*" )
+    VendorRequest.new :vendor_id => 0x5555, :vendor_data => @vendor_data, :transaction_id => transaction_id
+  }
+  let( :transaction_id ) { 1234 }
+  its( :vendor ) { should == 21845 }
+  let( :vendor_data ) { subject.data[0...@vendor_data.length]}
+  describe "data" do
+    it %{ should == [ 116, 104, 105, 115, 32, 105, 115, 32, 97, 32, 116, 101, 115, 116 ] } do
+      vendor_data.should == @vendor_data
+    end
+  end
+  it_should_behave_like "any OpenFlow message with transaction_id option"
+end
+
+
+describe VendorRequest, ".new( INVALID OPTION ) - vendor_data as String" do
+  subject { VendorRequest.new :vendor_data => "test" }
+  it "should raise TypeError" do
+    expect { subject }.to raise_error( TypeError )
+  end
+end
+
+
+describe VendorRequest, ".new( INVALID OPTIONS )" do
+  it "should raise TypeError" do
+    expect {
+      VendorRequest.new "INVALID OPTIONS"
+    }.to raise_error( TypeError )
+  end
+end
+
+
+describe VendorRequest, ".new( VALID OPTIONS ) - vendor_data greater than 16 bytes" do
+  vendor_data = "this is a long message....".unpack( "C*" )
+  subject { VendorRequest.new :vendor_data => vendor_data }
+  it "should assign the first 16 bytes only" do
+    subject.data.should == vendor_data[ 0..15 ]
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End: