trema 0.1.3

data/spec/trema/barrier-request_spec.rb

@@ -0,0 +1,83 @@
+#
+# 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 BarrierRequest, ".new( OPTIONAL OPTION MISSING )" do
+  it_should_behave_like "any Openflow message with default transaction ID"
+end
+
+
+describe BarrierRequest, ".new( VALID OPTION )" do
+  subject { BarrierRequest.new :transaction_id => transaction_id }
+  it_should_behave_like "any OpenFlow message with transaction_id option"
+
+
+  context "when #barrier_request" do
+    it "should #barrier_reply" do
+      class BarrierController < Controller; end
+      network {
+        vswitch { datapath_id 0xabc }
+      }.run( BarrierController ) {
+        controller( "BarrierController" ).should_receive( :barrier_reply )
+        controller( "BarrierController" ).send_message( 0xabc, BarrierRequest.new )
+        sleep 2 # FIXME: wait to send_message
+      }
+    end
+  end
+end
+
+
+describe BarrierRequest, ".new( OPTIONAL OPTION ) - transaction_id" do
+  context "when #barrier_request" do
+    it "should #barrier_reply with transaction_id == value" do
+      class BarrierController < Controller; end
+      network {
+        vswitch { datapath_id 0xabc }
+      }.run( BarrierController ) {
+        controller( "BarrierController" ).should_receive( :barrier_reply ) do | message |
+          message.datapath_id.should == 0xabc
+          message.transaction_id.should == 1234
+        end
+        barrier_request = BarrierRequest.new( :transaction_id => 1234 )
+        controller( "BarrierController" ).send_message( 0xabc, barrier_request )
+        sleep 2 # FIXME: wait to send_message
+      }
+    end
+  end
+end
+
+
+describe BarrierRequest, ".new( INVALID_OPTION )" do
+  it "should raise TypeError" do
+    expect {
+      BarrierRequest.new "INVALID_OPTION"
+    }.to raise_error( TypeError )
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/cli_spec.rb

@@ -0,0 +1,160 @@
+#
+# 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/cli"
+
+
+module Trema
+  describe Cli do
+    before :each do
+      @source = mock( "source", :interface => "trema-0", :ip => "192.168.0.1" )
+      @dest = mock( "dest", :interface => "trema-1", :ip => "192.168.0.2" )
+      @cli = Cli.new( @source )
+    end
+
+
+    context "when sending packets" do
+      it "should send packets (default)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 1 --length 22$/ )
+        @cli.send_packets( @dest )
+      end
+
+
+      it "should send packets (inc_ip_src)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 1 --length 22 --inc_ip_src$/ )
+        @cli.send_packets( @dest, :inc_ip_src => true )
+      end
+
+
+      it "should send packets (inc_ip_src = 1)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 1 --length 22 --inc_ip_src=1$/ )
+        @cli.send_packets( @dest, :inc_ip_src => 1 )
+      end
+
+
+      it "should send packets (inc_ip_dst)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 1 --length 22 --inc_ip_dst$/ )
+        @cli.send_packets( @dest, :inc_ip_dst => true )
+      end
+
+
+      it "should send packets (inc_ip_dst = 1)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 1 --length 22 --inc_ip_dst=1$/ )
+        @cli.send_packets( @dest, :inc_ip_dst => 1 )
+      end
+
+
+      it "should send_packets (tp_src = 60000)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 60000 --tp_dst 1 --pps 1 --duration 1 --length 22$/ )
+        @cli.send_packets( @dest, :tp_src => 60000 )
+      end
+
+
+      it "should send_packets (tp_dst = 10000)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 10000 --pps 1 --duration 1 --length 22$/ )
+        @cli.send_packets( @dest, :tp_dst => 10000 )
+      end
+
+
+      it "should send packets (inc_tp_src)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 1 --length 22 --inc_tp_src$/ )
+        @cli.send_packets( @dest, :inc_tp_src => true )
+      end
+
+
+      it "should send packets (inc_tp_src = 1)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 1 --length 22 --inc_tp_src=1$/ )
+        @cli.send_packets( @dest, :inc_tp_src => 1 )
+      end
+
+
+      it "should send_packets (pps = 100)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 100 --duration 1 --length 22$/ )
+        @cli.send_packets( @dest, :pps => 100 )
+      end
+
+
+      it "should send packets (duration = 10)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 10 --length 22$/ )
+        @cli.send_packets( @dest, :duration => 10 )
+      end
+
+
+      it "should send packets (length = 1000)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 1 --length 1000$/ )
+        @cli.send_packets( @dest, :length => 1000 )
+      end
+
+
+      it "should send_packets (inc_tp_dst)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 1 --length 22 --inc_tp_dst$/ )
+        @cli.send_packets( @dest, :inc_tp_dst => true )
+      end
+
+
+      it "should send_packets (inc_tp_dst = 65534)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 1 --length 22 --inc_tp_dst=65534$/ )
+        @cli.send_packets( @dest, :inc_tp_dst => 65534 )
+      end
+
+
+      it "should send_packets (inc_payload)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 1 --length 22 --inc_payload$/ )
+        @cli.send_packets( @dest, :inc_payload => true )
+      end
+
+
+      it "should send_packets (inc_payload = 1000)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --duration 1 --length 22 --inc_payload=1000$/ )
+        @cli.send_packets( @dest, :inc_payload => 1000 )
+      end
+
+
+      it "should send_packets (n_pkts = 10)" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 send_packets --ip_src 192.168.0.1 --ip_dst 192.168.0.2 --tp_src 1 --tp_dst 1 --pps 1 --length 22 --n_pkts=10$/ )
+        @cli.send_packets( @dest, :n_pkts => 10 )
+      end
+
+
+      it "should raise if both --duration and --n_pkts are specified" do
+        lambda do
+          @cli.send_packets( @dest, :duration => 10, :n_pkts => 10 )
+        end.should raise_error( "--duration and --n_pkts are exclusive." )
+      end
+    end
+
+
+    context "when resetting stats" do
+      it "should reset_stats" do
+        @cli.should_receive( :sh ).with( /cli -i trema-0 reset_stats --tx$/ )
+        @cli.should_receive( :sh ).with( /cli -i trema-0 reset_stats --rx$/ )
+        @cli.reset_stats
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/controller_spec.rb

@@ -0,0 +1,100 @@
+#
+# 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 Controller do
+    context "when using OpenFlow constants" do
+      subject { Controller.constants }
+
+      it { should include "OFPP_MAX" }
+      it { should include "OFPP_IN_PORT" }
+      it { should include "OFPP_TABLE" }
+      it { should include "OFPP_NORMAL" }
+      it { should include "OFPP_FLOOD" }
+      it { should include "OFPP_ALL" }
+      it { should include "OFPP_CONTROLLER" }
+      it { should include "OFPP_LOCAL" }
+      it { should include "OFPP_NONE" }
+
+      it { should include "OFPPR_ADD" }
+      it { should include "OFPPR_DELETE" }
+      it { should include "OFPPR_MODIFY" }
+
+      it { should include "OFPC_FLOW_STATS" }
+      it { should include "OFPC_TABLE_STATS" }
+      it { should include "OFPC_PORT_STATS" }
+      it { should include "OFPC_STP" }
+      it { should include "OFPC_RESERVED" }
+      it { should include "OFPC_IP_REASM" }
+      it { should include "OFPC_QUEUE_STATS" }
+      it { should include "OFPC_ARP_MATCH_IP" }
+
+      it { should include "OFPAT_OUTPUT" }
+      it { should include "OFPAT_SET_VLAN_VID" }
+      it { should include "OFPAT_SET_VLAN_PCP" }
+      it { should include "OFPAT_STRIP_VLAN" }
+      it { should include "OFPAT_SET_DL_SRC" }
+      it { should include "OFPAT_SET_DL_DST" }
+      it { should include "OFPAT_SET_NW_SRC" }
+      it { should include "OFPAT_SET_NW_DST" }
+      it { should include "OFPAT_SET_NW_TOS" }
+      it { should include "OFPAT_SET_TP_DST" }
+      it { should include "OFPAT_ENQUEUE" }
+      it { should include "OFPAT_VENDOR" }
+    end
+
+
+    context "when logging" do
+      subject { Controller.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
+
+
+    context "when sending flow_mod messages" do
+      it "should send a flow_mod_add message" do
+        class FlowModAddController < Controller; end
+
+        network {
+          vswitch { datapath_id 0xabc }
+        }.run( FlowModAddController ) {
+          controller( "FlowModAddController" ).send_flow_mod_add( 0xabc )
+          vswitch( "0xabc" ).should have( 1 ).flows
+        }
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/dsl/configuration_spec.rb

@@ -0,0 +1,122 @@
+#
+# 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"
+
+
+module Trema
+  module DSL
+    describe Configuration do
+      before :each do
+        @context = Configuration.new
+      end
+
+
+      context "when parsing trema configurations" do
+        it "should remember apps" do
+          @context.should have( 0 ).apps
+
+          Trema::App.add mock( "app #0", :name => "app #0" )
+          Trema::App.add mock( "app #1", :name => "app #1" )
+          Trema::App.add mock( "app #2", :name => "app #2" )
+
+          @context.should have( 3 ).apps
+
+          @context.apps[ "app #0" ].name.should == "app #0"
+          @context.apps[ "app #1" ].name.should == "app #1"
+          @context.apps[ "app #2" ].name.should == "app #2"
+        end
+
+
+        it "should remember hosts" do
+          @context.should have( 0 ).hosts
+
+          Trema::Host.add mock( "host #0", :name => "host #0" )
+          Trema::Host.add mock( "host #1", :name => "host #1" )
+          Trema::Host.add mock( "host #2", :name => "host #2" )
+
+          @context.should have( 3 ).hosts
+
+          @context.hosts[ "host #0" ].name.should == "host #0"
+          @context.hosts[ "host #1" ].name.should == "host #1"
+          @context.hosts[ "host #2" ].name.should == "host #2"
+        end
+
+
+        it "should remember links" do
+          @context.should have( 0 ).links
+
+          Trema::Link.add mock( "link #0", :name => "link #0" )
+          Trema::Link.add mock( "link #1", :name => "link #1" )
+          Trema::Link.add mock( "link #2", :name => "link #2" )
+
+          @context.should have( 3 ).links
+
+          @context.links[ "link #0" ].name.should == "link #0"
+          @context.links[ "link #1" ].name.should == "link #1"
+          @context.links[ "link #2" ].name.should == "link #2"
+        end
+
+
+        it "should remember filter settings" do
+          @context.packetin_filter.should be_nil
+
+          packetin_filter = mock( "filter", :name => "filter" )
+          Trema::PacketinFilter.add packetin_filter
+
+          @context.packetin_filter.should == packetin_filter
+        end
+
+
+        it "should remember switch manager" do
+          @context.switch_manager.should be_nil
+
+          switch_manager = mock( "switch manager", :name => "switch manager" )
+          Trema::SwitchManager.add switch_manager
+
+          @context.switch_manager.should == switch_manager
+        end
+
+
+        it "should remember switches" do
+          @context.should have( 0 ).switches
+
+          Trema::Switch.add mock( "switch #0", :name => "switch #0" )
+          Trema::Switch.add mock( "switch #1", :name => "switch #1" )
+          Trema::Switch.add mock( "switch #2", :name => "switch #2" )
+
+          @context.should have( 3 ).switches
+
+          @context.switches[ "switch #0" ].name.should == "switch #0"
+          @context.switches[ "switch #1" ].name.should == "switch #1"
+          @context.switches[ "switch #2" ].name.should == "switch #2"
+        end
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End: