trema 0.1.3

data/spec/trema/openflow-switch_spec.rb

@@ -0,0 +1,56 @@
+#
+# 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/switch"
+require "trema/openflow-switch"
+
+
+module Trema
+  describe OpenflowSwitch, %[dpid = "0xabc"] do
+    before :each do
+      stanza = DSL::Switch.new
+      stanza.dpid "0xabc"
+      @switch = OpenflowSwitch.new( stanza )
+    end
+
+
+    it "should return its name" do
+      @switch.name.should == "0xabc"
+    end
+
+
+    it "should return dpid in long format" do
+      @switch.dpid_long.should == "0000000000000abc"
+    end
+
+
+    it "should return dpid in short format" do
+      @switch.dpid_short.should == "0xabc"
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/packet-in_spec.rb

@@ -0,0 +1,168 @@
+#
+# 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::PacketIn do
+  def send_and_wait
+    send_packets "host1", "host2"
+    sleep 2
+  end
+  
+  
+  class PacketInController < Controller; end
+
+  
+  context "when instance is created" do
+    it "should have valid datapath_id" do
+      network {
+        vswitch( "test" ) { datapath_id 0xabc }
+        vhost "host1"
+        vhost "host2"
+        link "test", "host1"
+        link "test", "host2"
+      }.run( PacketInController ) {
+        controller( "PacketInController" ).should_receive( :packet_in ) do | datapath_id, message | 
+          datapath_id.should == 0xabc
+          message.datapath_id.should == 0xabc
+        end
+        send_and_wait
+      }
+    end
+    
+    
+    it "should have user data" do
+      network {
+        vswitch( "test" ) { datapath_id 0xabc }
+        vhost "host1"
+        vhost "host2"
+        link "test", "host1"
+        link "test", "host2"
+      }.run( PacketInController ) {
+        controller( "PacketInController" ).should_receive( :packet_in ) do | datapath_id, message | 
+           # packet_in expected to have data portion.
+          message.total_len.should > 20
+          message.data.should be_instance_of( String )
+          message.buffered?.should be_false
+        end
+        send_and_wait
+      }
+    end
+
+    
+    it "should have user L2 information (macsa)" do
+      network {
+        vswitch( "test" ) { datapath_id 0xabc }
+        vhost( "host1" ) { mac "00:00:00:00:00:01" }
+        vhost( "host2" ) { mac "00:00:00:00:00:02" }
+        link "test", "host1"
+        link "test", "host2"
+      }.run( PacketInController ) {
+        controller( "PacketInController" ).should_receive( :packet_in ) do | datapath_id, message | 
+          message.macsa.should be_instance_of( Trema::Mac )
+          message.macsa.to_s.should == "00:00:00:00:00:01"
+        end
+        send_and_wait
+      }
+    end
+    
+
+    it "should have user L2 information (macda)" do
+      network {
+        vswitch( "test" ) { datapath_id 0xabc }
+        vhost( "host1" ) { mac "00:00:00:00:00:01" }
+        vhost( "host2" ) { mac "00:00:00:00:00:02" }
+        link "test", "host1"
+        link "test", "host2"
+      }.run( PacketInController ) {
+        controller( "PacketInController" ).should_receive( :packet_in ) do | datapath_id, message | 
+          message.macda.should be_instance_of( Trema::Mac )
+          message.macda.to_s.should == "00:00:00:00:00:02"
+        end
+        send_and_wait
+      }
+    end
+
+
+    it "should have user L3 information" do
+      network {
+        vswitch( "test" ) { datapath_id 0xabc }
+        vhost( "host1" ) { ip "192.168.1.1" }
+        vhost( "host2" ) { ip "192.168.1.2" }
+        link "test", "host1"
+        link "test", "host2"
+      }.run( PacketInController ) {
+        controller( "PacketInController" ).should_receive( :packet_in ) do | datapath_id, message | 
+          message.eth_type.should == 0x0800
+          message.ipv4?.should == true
+          message.ipv4_saddr.should be_instance_of( Trema::IP )
+          message.ipv4_saddr.to_s.should == "192.168.1.1"
+          message.ipv4_daddr.should be_instance_of( Trema::IP )
+          message.ipv4_daddr.to_s.should == "192.168.1.2"
+        end
+        send_and_wait
+      }
+    end
+
+
+    it "should have user L4 information (udp)" do
+      network {
+        vswitch( "test" ) { datapath_id 0xabc }
+        vhost( "host1" )
+        vhost( "host2" )
+        link "test", "host1"
+        link "test", "host2"
+      }.run( PacketInController ) {
+        controller( "PacketInController" ).should_receive( :packet_in ) do | datapath_id, message | 
+          message.udp?.should == true
+          message.udp_src_port.should == 9000
+          message.udp_dst_port.should == 9001
+        end
+        send_packets "host1", "host2", :tp_src => 9000, :tp_dst => 9001
+        sleep 2
+      }
+    end
+
+    
+    it "should have valid input port" do
+      network {
+        vswitch( "test" ) { datapath_id 0xabc }
+        vhost "host1"
+        vhost "host2"
+        link "test", "host1"
+        link "test", "host2"
+      }.run( PacketInController ) {
+        controller( "PacketInController" ).should_receive( :packet_in ) do | datapath_id, message | 
+          message.in_port.should > 0
+        end
+        send_and_wait
+      }
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/packet-out_spec.rb

@@ -0,0 +1,128 @@
+#
+# 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 PacketOutController < Controller 
+  def packet_in datapath_id, message
+    send_flow_mod_add(
+      datapath_id,
+      :match => Match.from( message ),
+      :actions => Trema::ActionOutput.new( :port => 2 )
+    )
+    send_packet_out(
+      datapath_id,
+      :packet_in => message,
+      :actions => Trema::ActionOutput.new( :port => 2 )
+    )
+  end
+end
+
+
+describe "packet-out" do
+  context "a controller instance" do
+    it "should respond to #send_packet_out" do
+      PacketOutController.new.should respond_to(:send_packet_out)
+    end
+  end
+    
+  
+  context "when invoked with no datapath_id" do
+    it "should raise an error" do
+      lambda do
+        PacketOutController.new.send_packet_out
+      end.should raise_error("wrong number of arguments (0 for 1)")
+    end
+  end
+  
+  
+  context "when #packet_in" do
+    it "should #packet_out" do
+      network {
+        vswitch( "packet-out" ) { datapath_id 0xabc }
+        vhost "host1"
+        vhost "host2"
+        link "host1", "packet-out"
+        link "host2", "packet-out"
+      }.run( PacketOutController ) {
+        send_packets "host2", "host1"
+        sleep 2
+        vhost( "host1" ).rx_stats.n_pkts.should == 1
+      }
+    end
+  end
+
+
+  context "when data argument is string type" do
+    it "should #packet_out" do
+      network {
+        vswitch( "packet-out" ) { datapath_id 0xabc }
+        vhost "host1"
+        vhost ( "host2" ) {
+          ip "192.168.0.2"
+          netmask "255.255.0.0"
+          mac "00:00:00:00:00:02"
+        }
+        link "host1", "packet-out"
+        link "host2", "packet-out"
+      }.run( PacketOutController ) {
+        data = [
+          0x00, 0x00, 0x00, 0x00, 0x00, 0x02, # dst
+          0x00, 0x00, 0x00, 0x00, 0x00, 0x01, # src
+          0x08, 0x00, # ether type
+          # ipv4
+          0x45, 0x00, # version
+          0x00, 0x32, # length
+          0x00, 0x00,
+          0x00, 0x00,
+          0x40,       # ttl
+          0x11,       # protocol
+          0xf9, 0x67, # checksum
+          0xc0, 0xa8, 0x00, 0x01, # src
+          0xc0, 0xa8, 0x00, 0x02, # dst
+          # udp
+          0x00, 0x01, # src port
+          0x00, 0x01, # dst port
+          0x00, 0x1e, # length
+          0x00, 0x00, # checksum
+          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+          0x00, 0x00, 0x00, 0x00, 0x00, 0x00
+        ].pack( "C*" )
+        controller( "PacketOutController" ).send_packet_out(
+          0xabc,
+          :data => data,
+          :actions => Trema::ActionOutput.new( :port => 1 )
+	)
+        sleep 2
+        vhost( "host2" ).rx_stats.n_pkts.should == 1
+      }
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/packetin-filter_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/packetin-filter"
+
+
+module Trema
+  describe PacketinFilter do
+    it "should run packetin_filter with proper options" do
+      packetin_filter = PacketinFilter.new( :lldp => "TopologyManager", :packet_in => "OpenFlowPingPong" )
+      packetin_filter.should_receive( :sh ).once.with( /packetin_filter \-\-daemonize \-\-name=filter lldp::TopologyManager packet_in::OpenFlowPingPong$/ )
+
+      packetin_filter.run!
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/port-mod_spec.rb

@@ -0,0 +1,101 @@
+#
+# 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 PortMod, ".new( VALID OPTIONS )" do
+  subject {
+    PortMod.new(
+      :port_no => 2,
+      :hw_addr => Mac::new( "11:22:33:44:55:66" ),
+      :config => 1,
+      :mask => 1,
+      :advertise => 0
+    )
+  }
+  its( :port_no ) { should == 2 }
+  its( :hw_addr ) { subject.to_s.should eq "11:22:33:44:55:66" }
+  its( :config ) { should == 1 }
+  its( :mask ) { should == 1 }
+  its( :advertise ) { should == 0 }
+  it_should_behave_like "any Openflow message with default transaction ID"
+
+
+  describe "hw_addr" do
+    it "should be a Trema::Mac object" do
+      PortMod.new(
+        :port_no => 2,
+        :hw_addr => Mac::new( "11:22:33:44:55:66" ),
+        :config => 1,
+        :mask => 1,
+        :advertise => 0
+      ).hw_addr.to_s.should eq( "11:22:33:44:55:66" )
+    end
+
+
+    it "should be a string('11:22:33:44:55')" do
+      PortMod.new(
+        :port_no => 2,
+        :hw_addr => "11:22:33:44:55:66",
+        :config => 1,
+        :mask => 1,
+        :advertise => 0 ).hw_addr.to_s.should eq( "11:22:33:44:55:66" )
+    end
+
+
+    it "should be a number(281474976710655)" do
+      PortMod.new(
+        :port_no => 2,
+        :hw_addr => 281474976710655,
+        :config => 1,
+        :mask => 1,
+        :advertise => 0 ).hw_addr.to_s.should eq( "ff:ff:ff:ff:ff:ff" )
+    end
+
+
+    it "should otherwise raise ArgumentError" do
+      lambda do
+        PortMod.new(
+          :port_no => 2,
+          :hw_addr => Array.new( 1234 ),
+          :config => 1,
+          :mask => 1,
+          :advertise => 0
+        )
+      end.should raise_error ArgumentError
+    end
+  end
+end
+
+
+describe PortMod, "new( MANDATORY OPTIONS MISSING )" do
+  it "should raise ArgumentError" do
+    expect { subject }.to raise_error( ArgumentError, "Port no, hw_addr, config, mask, advertise are mandatory options" )
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End: