lib/commands/network.rb in testlab-0.6.6 vs lib/commands/network.rb in testlab-0.6.7

@@ -185,11 +185,12 @@
         help_now!('a name is required') if options[:name].nil?
 
         network = @testlab.networks.select{ |c| c.id.to_sym == options[:name].to_sym }.first
         network.nil? and raise TestLab::TestLabError, "We could not find the network you supplied!"
 
-        network.manage_route(:add)
+        p = TestLab::Provisioner::Route.new({}, @ui)
+        p.on_network_setup(network)
         @testlab.ui.stdout.puts("Added routes successfully!".green.bold)
         @testlab.ui.stdout.puts %x(netstat -nr | grep '#{network.node.ip}').strip
       end
     end
 
@@ -201,10 +202,11 @@
         help_now!('a name is required') if options[:name].nil?
 
         network = @testlab.networks.select{ |c| c.id.to_sym == options[:name].to_sym }.first
         network.nil? and raise TestLab::TestLabError, "We could not find the network you supplied!"
 
-        network.manage_route(:del)
+        p = TestLab::Provisioner::Route.new({}, @ui)
+        p.on_network_teardown(network)
         @testlab.ui.stdout.puts("Deleted routes successfully!".red.bold)
         @testlab.ui.stdout.puts %x(netstat -nr | grep '#{network.node.ip}').strip
       end
     end