trema 0.1.3

data/spec/trema/logger_spec.rb

@@ -0,0 +1,48 @@
+#
+# 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"
+
+
+module Trema
+  describe Logger do
+    subject {
+      class LoggingObject
+        include Trema::Logger
+      end
+      LoggingObject.new
+    }
+
+    it { should respond_to :critical }
+    it { should respond_to :error }
+    it { should respond_to :warn }
+    it { should respond_to :notice }
+    it { should respond_to :info }
+    it { should respond_to :debug }
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/mac_spec.rb

@@ -0,0 +1,115 @@
+#
+# 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/mac"
+
+
+module Trema
+  describe Mac do
+    context "when creating from an invalid value" do
+      context %{when "INVALID MAC ADDRESS" } do
+        it "should raise an error" do
+          lambda do
+            Mac.new( "INVALID MAC ADDRESS" )
+          end.should raise_error( %{Invalid MAC address: "INVALID MAC ADDRESS"} )
+        end
+      end
+
+
+      context "when -1" do
+        it "should raise an error" do
+          lambda do
+            Mac.new( -1 )
+          end.should raise_error( "Invalid MAC address: -1" )
+        end
+      end
+
+
+      context "when 0x1000000000000" do
+        it "should raise an error" do
+          lambda do
+            Mac.new( 0x1000000000000 )
+          end.should raise_error( "Invalid MAC address: #{ 0x1000000000000 }" )
+        end
+      end
+
+
+      context %{when "[ 1, 2, 3 ]" } do
+        it "should raise an error" do
+          lambda do
+            Mac.new( [ 1, 2, 3 ] )
+          end.should raise_error( "Invalid MAC address: [1, 2, 3]" )
+        end
+      end
+    end
+
+
+    context "when creating" do
+      subject { Mac.new( mac_address ) }
+
+
+      context %{when "11:22:33:44:55:66"} do
+        let( :mac_address ) { "11:22:33:44:55:66" }
+
+        it { should == Mac.new( "11:22:33:44:55:66" ) }
+        its( :value ) { should == 0x112233445566 }
+        its( :to_s ) { should == "11:22:33:44:55:66" }
+        its( :to_short ) { should == [ 0x11, 0x22, 0x33, 0x44, 0x55, 0x66 ] }
+      end
+
+
+      context "when 0" do
+        let( :mac_address ) { 0 }
+
+        it { should == Mac.new( 0 ) }
+        its( :value ) { should == 0 }
+        its( :to_s ) { should == "00:00:00:00:00:00" }
+        its( :to_short ) { should == [ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 ] }
+      end
+
+
+      context "when 0xffffffffffff" do
+        let( :mac_address ) { 0xffffffffffff }
+
+        it { should == Mac.new( 0xffffffffffff ) }
+        its( :value ) { should == 0xffffffffffff }
+        its( :to_s ) { should == "ff:ff:ff:ff:ff:ff" }
+        its( :to_short ) { should == [ 0xff, 0xff, 0xff, 0xff, 0xff, 0xff ] }
+      end
+    end
+
+
+    context "when querying FDB" do
+      it "should be used for FDB keys" do
+        fdb = {}
+        fdb[ Mac.new( "00:00:00:00:00:01" ) ] = "Port #1"
+        fdb[ Mac.new( "00:00:00:00:00:01" ) ].should == "Port #1"
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/match_spec.rb

@@ -0,0 +1,113 @@
+#
+# Author: SUGYO Kazushi
+#
+# Copyright (C) 2008-2011 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 Match, ".new" do
+  subject {
+    Match.new(
+      :in_port => 1,
+      :dl_src => "00:00:00:00:00:01",
+      :dl_dst => "00:00:00:00:00:02",
+      :dl_vlan => 65535,
+      :dl_vlan_pcp => 0,
+      :dl_type => 0x800,
+      :nw_tos => 0,
+      :nw_proto => 17,
+      :nw_src => "192.168.0.1",
+      :nw_dst => "192.168.0.0/24",
+      :tp_src => 10,
+      :tp_dst => 20
+    )
+  }
+  its( :in_port ) { should == 1 }
+  its( :dl_src ) { subject.to_s.should == "00:00:00:00:00:01" }
+  its( :dl_dst ) { subject.to_s.should == "00:00:00:00:00:02" }
+  its( :dl_vlan ) { should == 65535 }
+  its( :dl_vlan_pcp ) { should == 0 }
+  its( :dl_type ) { should == 0x800 }
+  its( :nw_tos ) { should == 0 }
+  its( :nw_proto ) { should == 17 }
+  its( :nw_src ) { subject.to_s.should == "192.168.0.1" }
+  its( :nw_src ) { subject.prefixlen.should == 32 }
+  its( :nw_dst ) { subject.to_s.should == "192.168.0.0" }
+  its( :nw_dst ) { subject.prefixlen.should == 24 }
+  its( :tp_src ) { should == 10 }
+  its( :tp_dst ) { should == 20 }
+  its( :to_s ) { should == "wildcards = 0x20000, in_port = 1, dl_src = 00:00:00:00:00:01, dl_dst = 00:00:00:00:00:02, dl_vlan = 65535, dl_vlan_pcp = 0, dl_type = 0x800, nw_tos = 0, nw_proto = 17, nw_src = 192.168.0.1/32, nw_dst = 192.168.0.0/24, tp_src = 10, tp_dst = 20" }
+end
+
+
+describe Match, ".compare" do
+  it "Should match" do
+    tester = Match.new(
+      :in_port => 1,
+      :dl_src => "00:00:00:00:00:01",
+      :dl_dst => "00:00:00:00:00:02",
+      :dl_vlan => 65535,
+      :dl_vlan_pcp => 0,
+      :dl_type => 0x800,
+      :nw_tos => 0,
+      :nw_proto => 17,
+      :nw_src => "192.168.0.1",
+      :nw_dst => "192.168.0.2",
+      :tp_src => 10,
+      :tp_dst => 20
+    )
+    pattern = Match.new(
+      :in_port => 1,
+      :nw_src => "192.168.0.0/24",
+      :nw_dst => "192.168.0.0/24"
+    )
+    pattern.compare( tester ).should == true
+  end
+
+  it "Should not match" do
+    tester = Match.new(
+      :in_port => 1,
+      :dl_src => "00:00:00:00:00:01",
+      :dl_dst => "00:00:00:00:00:02",
+      :dl_vlan => 65535,
+      :dl_vlan_pcp => 0,
+      :dl_type => 0x800,
+      :nw_tos => 0,
+      :nw_proto => 17,
+      :nw_src => "192.168.0.1",
+      :nw_dst => "192.168.0.2",
+      :tp_src => 10,
+      :tp_dst => 20
+    )
+    pattern = Match.new(
+      :in_port => 1,
+      :nw_src => "10.0.0.0/8",
+      :nw_dst => "10.0.0.0/8"
+    )
+    pattern.compare( tester ).should == false
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/open-vswitch_spec.rb

@@ -0,0 +1,123 @@
+#
+# 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/open-vswitch"
+
+
+module Trema
+  describe Switch do
+    around do | example |
+      begin
+        Switch.clear
+        example.run
+      ensure
+        Switch.clear
+      end
+    end
+
+    it "should keep a list of vswitches" do
+      OpenVswitch.new mock( "stanza 0", :name => "vswitch 0", :validate => true )
+      OpenVswitch.new mock( "stanza 1", :name => "vswitch 1", :validate => true )
+      OpenVswitch.new mock( "stanza 2", :name => "vswitch 2", :validate => true )
+
+      Switch.should have( 3 ).vswitches
+      Switch[ "vswitch 0" ].should_not be_nil
+      Switch[ "vswitch 1" ].should_not be_nil
+      Switch[ "vswitch 2" ].should_not be_nil
+    end
+  end
+
+
+  describe OpenVswitch, %[dpid = "0xabc"] do
+    subject {
+      stanza = { :dpid_short => "0xabc", :dpid_long => "0000000000000abc", :ip => "127.0.0.1" }
+      stanza.stub!( :validate )
+      stanza.stub!( :name ).and_return( name )
+      OpenVswitch.new stanza
+    }
+
+
+    context "when its name is not set" do
+      let( :name ) { "0xabc" }
+
+      its( :name ) { should == "0xabc" }
+      its( :dpid_short ) { should == "0xabc" }
+      its( :dpid_long ) { should == "0000000000000abc" }
+      its( :network_device ) { should == "vsw_0xabc" }
+    end
+
+
+    context "when its name is set" do
+      let( :name ) { "Otosan Switch" }
+
+      its( :name ) { should == "Otosan Switch" }
+      its( :dpid_short ) { should == "0xabc" }
+      its( :dpid_long ) { should == "0000000000000abc" }
+      its( :network_device ) { should == "vsw_0xabc" }
+    end
+
+
+    context "when getting its flows" do
+      let( :name ) { "0xabc" }
+
+      it "should execute ofctl to get the flows" do
+        ofctl = mock( "ofctl" )
+        Ofctl.stub!( :new ).and_return( ofctl )
+
+        ofctl.should_receive( :users_flows ).with( subject ).once
+
+        subject.flows
+      end
+    end
+
+
+    context "when running it" do
+      let( :name ) { "0xabc" }
+
+      it "should execute ovs openflowd" do
+        subject.should_receive( :sh ).with do | command |
+          command.should include( Executables.ovs_openflowd )
+        end
+
+        subject.run!
+      end
+
+      it "should be connected to virtual ports" do
+        subject << "VirtualInterface0"
+        subject << "VirtualInterface1"
+        subject << "VirtualInterface2"
+
+        subject.should_receive( :sh ).with do | command |
+          command.should include( "--ports=VirtualInterface0,VirtualInterface1,VirtualInterface2" )
+        end
+
+        subject.run!
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/openflow-error_spec.rb

@@ -0,0 +1,141 @@
+#
+# 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 Trema::OpenflowError, "new" do
+  context "when #port_mod with an invalid(port_no) is sent" do
+    it "should receive #error(type=Error::OFPET_PORT_MOD_FAILED,code=Error::OFPPMFC_BAD_PORT)" do
+      class OpenflowErrorController < Controller; end
+      network {
+        vswitch( "error-port") { datapath_id 0xabc }
+      }.run( OpenflowErrorController ) {
+        port_mod = Trema::PortMod.new( :port_no => 2, 
+          :hw_addr => Trema::Mac.new( "11:22:33:44:55:66" ),
+          :config => 1,
+          :mask => 1,
+          :advertise => 0
+        )
+        controller( "OpenflowErrorController" ).should_receive( :openflow_error ) do | arg |
+          arg.datapath_id.should == 0xabc
+          arg.type.should == Error::OFPET_PORT_MOD_FAILED
+          arg.code.should == Error::OFPPMFC_BAD_PORT 
+        end
+        controller( "OpenflowErrorController" ).send_message( 0xabc, port_mod )
+        sleep 2 # FIXME: wait to send_message
+      }
+    end
+  end
+  
+  
+  context "when #port_mod with an invalid(hw_addr) is sent" do
+    it "should receive #error(type=Error::OFPET_PORT_MOD_FAILED,code=Error::OFPPMFC_BAD_HW_ADDR)" do
+      class OpenflowErrorController < Controller; end
+      network {
+        vswitch( "error-hw-addr") { datapath_id 0xabc }
+        vhost "host1"
+        vhost "host2"
+        link "host1", "error-hw-addr"
+        link "host2", "error-hw-addr"
+      }.run( OpenflowErrorController ) {
+        port_mod = Trema::PortMod.new( :port_no => 1, 
+          :hw_addr => Trema::Mac.new( "11:22:33:44:55:66" ),
+          :config => 1,
+          :mask => 1,
+          :advertise => 0
+        )
+        controller( "OpenflowErrorController" ).should_receive( :openflow_error ) do | arg |
+          arg.datapath_id.should == 0xabc
+          arg.type.should == Error::OFPET_PORT_MOD_FAILED
+          arg.code.should == Error::OFPPMFC_BAD_HW_ADDR
+        end
+        controller( "OpenflowErrorController" ).send_message( 0xabc, port_mod )
+        sleep 2 # FIXME: wait to send_message
+      }
+    end
+  end
+  
+  
+  context "when #flow_mod(add) message with an invalid(action_port) is sent" do
+    it "should receive #error(type=Error::OFPET_BAD_ACTION,code=Error::OFPBAC_BAD_OUT_PORT)" do
+      class OpenflowErrorController < Controller; end
+      network {
+        vswitch( "error-out-port") { datapath_id 0xabc }
+        vhost "host1"
+        vhost "host2"
+        link "host1", "error-port"
+        link "host2", "error-port"
+      }.run( OpenflowErrorController ) {
+        controller( "OpenflowErrorController" ).should_receive( :openflow_error ) do | arg |
+          arg.datapath_id.should == 0xabc
+          arg.type.should == Error::OFPET_BAD_ACTION
+          arg.code.should == Error::OFPBAC_BAD_OUT_PORT
+        end
+        controller( "OpenflowErrorController" ).send_flow_mod_add( 0xabc, :actions => ActionOutput.new( :port => 0x5555 ) )
+        sleep 2 # FIXME: wait to send_flow_mod_add
+      }
+    end
+  end
+  
+  
+  context "when an unsupported message is sent" do  
+    it "should receive an openflow error with valid attributes" do
+      class OpenflowController < Controller; end
+      network {
+        vswitch( "error-request") { datapath_id 0xabc }
+      }.run( OpenflowController ) {
+        queue_get_config_request = Trema::QueueGetConfigRequest.new( :port => 1 )
+        controller( "OpenflowController" ).should_receive( :openflow_error ) do | message |
+          message.datapath_id.should == 0xabc
+          message.type.should satisfy { | n |
+            n >= 0 && n <= 5
+          }
+          case message.type
+          when 0,4
+            message.code.should include 0,1
+          when 1,2
+            message.code.should satisfy { | n | 
+              n >= 0 && n <= 8
+            }
+          when 3
+            message.code.should satisfy { | n |
+              n >= 0 && n <= 5
+            }
+          when 5
+            message.code.should satisfy { |n|
+              n >= 0 && n <= 3
+            }
+          end
+        end
+        controller( "OpenflowController" ).send_message( 0xabc, queue_get_config_request )
+        sleep 2 # FIXME: wait to send_message
+      }
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End: