trema 0.1.3

data/spec/trema/dsl/link_spec.rb

@@ -0,0 +1,54 @@
+#
+# 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/link"
+
+
+module Trema
+  module DSL
+    describe Link do
+      context %[when parsing "link peerA peerB"] do
+        it %[recognizes "link peerA peerB" directive] do
+          lambda do
+            Link.new "Host", "Switch"
+          end.should_not raise_error
+        end
+      end
+
+
+      context "when getting attributes of a link" do
+        it "remembers peers" do
+          link = Link.new( "Host", "Switch" )
+          link.peers.size.should == 2
+          link.peers.should include( "Host" )
+          link.peers.should include( "Switch" )
+        end
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/dsl/run_spec.rb

@@ -0,0 +1,78 @@
+#
+# 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/run"
+
+
+module Trema
+  module DSL
+    describe Run do
+      before :each do
+        @run = Run.new
+      end
+
+
+      context %[when parsing "run { ... }"] do
+        it %[recognizes "path COMMAND_PATH" directive] do
+          lambda do
+            @run.path "/usr/bin/tremario"
+          end.should_not raise_error
+        end
+
+
+        it %[recognizes "options OPTIONS..." directive] do
+          lambda do
+            @run.options "--verbose", "--color"
+          end.should_not raise_error
+        end
+      end
+
+
+      context "when getting the attributes of an run" do
+        it "returns its name" do
+          @run.path "/usr/bin/tremario"
+          @run[ :name ].should == "tremario"
+        end
+
+
+        it "returns its path" do
+          @run.path "/usr/bin/tremario"
+          @run[ :path ].should == "/usr/bin/tremario"
+        end
+
+
+        it "returns its options" do
+          @run.options "--verbose", "--color"
+          @run[ :options ].size.should == 2
+          @run[ :options ].should include( "--verbose" )
+          @run[ :options ].should include( "--color" )
+        end
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/dsl/runner_spec.rb

@@ -0,0 +1,239 @@
+#
+# 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/runner"
+
+
+module Trema
+  module DSL
+    describe Runner do
+      before :each do
+        ::Process.stub!( :fork ).and_yield
+        ::Process.stub!( :waitpid )
+        @switch_manager = mock( "switch manager", :run! => nil )
+        SwitchManager.stub!( :new ).and_return( @switch_manager )
+      end
+
+
+      context "when running" do
+        it "should run tremashark" do
+          tremashark = mock
+          tremashark.should_receive( :run ).once
+
+          context = mock(
+            "context",
+            :tremashark => tremashark,
+            :switch_manager => nil,
+            :packetin_filter => nil,
+            :links => {},
+            :hosts => {},
+            :switches => {},
+            :apps => {},
+            :port => 6633
+          )
+
+          Runner.new( context ).run
+        end
+
+
+        it "should run switch_manager" do
+          @switch_manager.should_receive( :run! ).once
+
+          context = mock(
+            "context",
+            :port => 6633,
+            :tremashark => nil,
+            :switch_manager => nil,
+            :packetin_filter => nil,
+            :links => {},
+            :hosts => {},
+            :switches => {},
+            :apps => {}
+          )
+
+          Runner.new( context ).run
+        end
+
+
+        it "should run packetin_filter" do
+          packetin_filter = mock
+          packetin_filter.should_receive( :run! ).once
+
+          context = mock(
+            "context",
+            :tremashark => nil,
+            :switch_manager => mock( "switch_manager", :run! => nil ),
+            :packetin_filter => packetin_filter,
+            :links => {},
+            :hosts => {},
+            :switches => {},
+            :apps => {},
+            :port => 6633
+          )
+
+          Runner.new( context ).run
+        end
+
+
+        it "should create links" do
+          link0 = mock( "link0" )
+          link0.should_receive( :delete! ).once.ordered
+          link0.should_receive( :enable! ).once.ordered
+
+          link1 = mock( "link1" )
+          link1.should_receive( :delete! ).once.ordered
+          link1.should_receive( :enable! ).once.ordered
+
+          link2 = mock( "link2" )
+          link2.should_receive( :delete! ).once.ordered
+          link2.should_receive( :enable! ).once.ordered
+
+          context = mock(
+            "context",
+            :tremashark => nil,
+            :switch_manager => mock( "switch manager", :run! => nil ),
+            :packetin_filter => nil,
+            :links => { "link0" => link0, "link1" => link1, "link2" => link2 },
+            :hosts => {},
+            :switches => {},
+            :apps => {},
+            :port => 6633
+          )
+
+          Runner.new( context ).run
+        end
+
+
+        it "should run vhosts" do
+          host0 = mock( "host0" )
+          host1 = mock( "host1" )
+          host2 = mock( "host2" )
+
+          host0.should_receive( :run! ).once.ordered
+          host0.should_receive( :add_arp_entry ).with( [ host1, host2 ] ).once.ordered
+
+          host1.should_receive( :run! ).once.ordered
+          host1.should_receive( :add_arp_entry ).with( [ host0, host2 ] ).once.ordered
+
+          host2.should_receive( :run! ).once.ordered
+          host2.should_receive( :add_arp_entry ).with( [ host0, host1 ] ).once.ordered
+
+          context = mock(
+            "context",
+            :tremashark => nil,
+            :switch_manager => mock( "switch manager", :run! => nil ),
+            :packetin_filter => nil,
+            :links => {},
+            :hosts => { "host0" => host0, "host1" => host1, "host2" => host2 },
+            :switches => {},
+            :apps => {},
+            :port => 6633
+          )
+
+          Runner.new( context ).run
+        end
+
+
+        it "should run switches" do
+          switch0 = mock( "switch0" )
+          switch0.should_receive( :run! ).once.ordered
+
+          switch1 = mock( "switch1" )
+          switch1.should_receive( :run! ).once.ordered
+
+          switch2 = mock( "switch2" )
+          switch2.should_receive( :run! ).once.ordered
+
+          context = mock(
+            "context",
+            :tremashark => nil,
+            :switch_manager => mock( "switch manager", :run! => nil ),
+            :packetin_filter => nil,
+            :links => {},
+            :hosts => {},
+            :switches => { "switch0" => switch0, "switch1" => switch1, "switch 2" => switch2 },
+            :apps => {},
+            :port => 6633
+          )
+
+          Runner.new( context ).run
+        end
+
+
+        it "should run apps" do
+          app0 = mock( "app0" )
+          app0.should_receive( :daemonize! ).once.ordered
+
+          app1 = mock( "app1" )
+          app1.should_receive( :daemonize! ).once.ordered
+
+          app2 = mock( "app2", :name => "App2" )
+          app2.should_receive( :run! ).once.ordered
+
+          context = mock(
+            "context",
+            :tremashark => nil,
+            :switch_manager => mock( "switch manager", :run! => nil, :rule => {} ),
+            :packetin_filter => nil,
+            :links => {},
+            :hosts => {},
+            :switches => {},
+            :apps => { "app0" => app0, "app1" => app1, "app2" => app2 }
+          )
+
+          Runner.new( context ).run
+        end
+
+
+        it "should daemonize apps" do
+          app0 = mock( "app0" )
+          app0.should_receive( :daemonize! ).once.ordered
+
+          app1 = mock( "app1" )
+          app1.should_receive( :daemonize! ).once.ordered
+
+          app2 = mock( "app2", :name => "App2" )
+          app2.should_receive( :daemonize! ).once.ordered
+
+          context = mock(
+            "context",
+            :tremashark => nil,
+            :switch_manager => mock( "switch manager", :run! => nil, :rule => {} ),
+            :packetin_filter => nil,
+            :links => {},
+            :hosts => {},
+            :switches => {},
+            :apps => { "app0" => app0, "app1" => app1, "app2" => app2 }
+          )
+
+          Runner.new( context ).daemonize
+        end
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/spec/trema/dsl/switch_spec.rb

@@ -0,0 +1,77 @@
+#
+# 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"
+
+
+module Trema
+  module DSL
+    describe Switch do
+      before :each do
+        @switch = Switch.new
+      end
+
+
+      context %[when parsing "switch { ... }"] do
+        it %[recognizes "dpid DATAPATH_ID" directive] do
+          lambda do
+            @switch.dpid "0xabc"
+          end.should_not raise_error
+        end
+
+
+        it %[recognizes "datapath_id DATAPATH_ID" directive] do
+          lambda do
+            @switch.datapath_id "0xabc"
+          end.should_not raise_error
+        end
+
+
+        it %[recognizes "ports PORT_NUMBERS" directive] do
+          lambda do
+            @switch.ports "0-4"
+          end.should_not raise_error
+        end
+      end
+
+
+      context "when getting the attributes of a switch" do
+        it "returns its dpid in long format" do
+          @switch.dpid "0xabc"
+          @switch[ :dpid_long ].should == "0000000000000abc"
+        end
+
+
+        it "returns its dpid in short format" do
+          @switch.dpid "0xabc"
+          @switch[ :dpid_short ].should == "0xabc"
+        end
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End: