rbeapi 0.5.1 → 1.0

data/spec/unit/rbeapi/api/interfaces/vxlan_spec.rb

@@ -23,7 +23,7 @@ describe Rbeapi::Api::VxlanInterface do
     let(:resource) { subject.get('Vxlan1') }
 
     let(:keys) do
-      [:type, :shutdown, :description, :name, :source_interface,
+      [:type, :shutdown, :load_interval, :description, :name, :source_interface,
        :multicast_group, :udp_port, :flood_list, :vlans]
     end
 

data/spec/unit/rbeapi/api/prefixlists/default_spec.rb

@@ -0,0 +1,202 @@
+#
+# Copyright (c) 2016, Arista Networks, Inc.
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are
+# met:
+#
+#   Redistributions of source code must retain the above copyright notice,
+#   this list of conditions and the following disclaimer.
+#
+#   Redistributions in binary form must reproduce the above copyright
+#   notice, this list of conditions and the following disclaimer in the
+#   documentation and/or other materials provided with the distribution.
+#
+#   Neither the name of Arista Networks nor the names of its
+#   contributors may be used to endorse or promote products derived from
+#   this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL ARISTA NETWORKS
+# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
+# BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
+# OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
+# IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#
+require 'spec_helper'
+
+require 'rbeapi/api/prefixlists'
+
+include FixtureHelpers
+
+describe Rbeapi::Api::Prefixlists do
+  subject { described_class.new(node) }
+
+  let(:node) { double('node') }
+
+  def prefixlists
+    prefixlists = Fixtures[:prefixlists]
+    return prefixlists if prefixlists
+    fixture('prefixlists', format: :text, dir: File.dirname(__FILE__))
+  end
+
+  before :each do
+    allow(subject.node).to receive(:running_config).and_return(prefixlists)
+  end
+
+  describe '#get' do
+    let(:resource) { subject.get('test1') }
+
+    let(:prefixlist) do
+        [{
+          'seq' => '10',
+          'action' => 'permit',
+          'prefix' => '1.2.3.0/24'
+        },
+        {
+          'seq' => '20',
+          'action' => 'permit',
+          'prefix' => '2.3.4.0/24 le 30'
+        },
+        {
+          'seq' => '30',
+          'action' => 'permit',
+          'prefix' => '3.4.5.0/24 ge 26 le 30'
+        }]
+    end
+
+    let(:keys) { ['seq', 'action', 'prefix'] }
+
+    it 'returns the prefix list for an existing name' do
+      expect(resource).to eq(prefixlist)
+    end
+
+    it 'returns an array of prefixes' do
+      expect(resource).to be_a_kind_of(Array)
+    end
+
+    it 'has three prefixes' do
+      expect(resource.size).to eq(3)
+    end
+
+    it 'has all keys' do
+      resource.each do |prefix|
+        expect(prefix.keys).to match_array(keys)
+      end
+    end
+
+    let(:nonexistent) { subject.get('nonexistent') }
+    it 'returns nil for a non-existing name' do
+      expect(nonexistent).to eq(nil)
+    end
+  end
+
+  describe '#getall' do
+    let(:resource) { subject.getall }
+
+    let(:plists) do
+      { 
+        "test1" => [
+          {
+            "seq" => "10",
+            "action" => "permit",
+            "prefix" => "1.2.3.0/24"
+          },
+          {
+            "seq" => "20",
+            "action" => "permit",
+            "prefix" => "2.3.4.0/24 le 30"
+          },
+          {
+            "seq" => "30",
+            "action" => "permit",
+            "prefix" => "3.4.5.0/24 ge 26 le 30"
+          }
+        ],
+        "test2" => [
+          {
+            "seq" => "10",
+            "action" => "permit",
+            "prefix" => "10.11.0.0/16"
+          },
+          {
+            "seq" => "20",
+            "action" => "permit",
+            "prefix" => "10.12.0.0/16 le 24"
+          }
+        ],
+        "test3" => []
+      }
+    end
+
+    it 'returns all prefix lists' do
+      expect(resource).to eq(plists)
+    end
+
+    it 'returns a hash' do
+      expect(resource).to be_a_kind_of(Hash)
+    end
+
+    it 'has three prefix lists' do
+      expect(resource.size).to eq(3)
+    end
+  end
+
+  describe '#create' do
+    it 'creates a new prefix list' do
+      expect(node).to receive(:config).with('ip prefix-list plist1')
+      expect(subject.create('plist1')).to be_truthy
+    end
+
+    it 'creates an existing prefix list' do
+      expect(node).to receive(:config).with('ip prefix-list test1')
+      expect(subject.create('test1')).to be_truthy
+    end
+  end
+
+  describe '#add_rule' do
+    it 'adds rule to existing prefix list' do
+      expect(node).to receive(:config).with('ip prefix-list test1 seq 25 permit 9.8.7.0/24')
+      expect(subject.add_rule('test1', 'permit','9.8.7.0/24', '25')).to be_truthy
+    end
+
+    it 'adds rule to existing prefix list w/o seq' do
+      expect(node).to receive(:config).with('ip prefix-list test1 permit 8.7.6.0/24')
+      expect(subject.add_rule('test1', 'permit', '8.7.6.0/24')).to be_truthy
+    end
+
+    it 'adds rule to non-existing prefix list' do
+      expect(node).to receive(:config).with('ip prefix-list plist2 seq 10 permit 6.5.4.128/25')
+      expect(subject.add_rule('plist2', 'permit', '6.5.4.128/25', '10')).to be_truthy
+    end
+
+    it 'adds rule to non-existing prefix list w/o seq' do
+      expect(node).to receive(:config).with('ip prefix-list plist2 deny 5.4.3.0/25')
+      expect(subject.add_rule('plist2', 'deny', '5.4.3.0/25')).to be_truthy
+    end
+
+    it 'overwrites existing rule' do
+      expect(node).to receive(:config).with('ip prefix-list test1 seq 20 permit 2.3.5.0/24 le 28')
+      expect(subject.add_rule('test1', 'permit', '2.3.5.0/24 le 28', '20')).to be_truthy
+    end
+  end
+
+  describe '#delete' do
+    it 'deletes a prefix-list' do
+      expect(node).to receive(:config).with('no ip prefix-list test1')
+      expect(subject.delete('test1'))
+    end
+
+    it 'deletes a rule from a prefix-list' do
+      expect(node).to receive(:config).with('no ip prefix-list test2 seq 10')
+      expect(subject.delete('test2', '10'))
+    end
+  end
+
+end

data/spec/unit/rbeapi/api/prefixlists/fixture_prefixlists.text

@@ -0,0 +1,11 @@
+ip prefix-list test1
+   seq 10 permit 1.2.3.0/24
+   seq 20 permit 2.3.4.0/24 le 30
+   seq 30 permit 3.4.5.0/24 ge 26 le 30
+!
+ip prefix-list test2
+   seq 10 permit 10.11.0.0/16
+   seq 20 permit 10.12.0.0/16 le 24
+!
+ip prefix-list test3
+!

data/spec/unit/rbeapi/api/routemaps/default_spec.rb

@@ -244,6 +244,11 @@ describe Rbeapi::Api::Routemaps do
       expect(subject.create('test', 'deny', 20,
                             default: true)).to be_truthy
     end
+
+    it 'raises an ArgumentError if opts match is not an array' do
+      expect { subject.create('test', 'permit', 10, match: '123') }
+        .to raise_error(ArgumentError)
+    end
   end
 
   describe '#delete' do

data/spec/unit/rbeapi/api/switchports/default_spec.rb

@@ -195,11 +195,11 @@ describe Rbeapi::Api::Switchports do
         .to raise_error(ArgumentError)
     end
 
-    it 'sets vlan 8 and 9 to the trunk allowed vlans' do
+    it 'sets vlan 8-10 and 100 to the trunk allowed vlans' do
       expect(node).to receive(:config)
-        .with(['interface Ethernet1', 'switchport trunk allowed vlan none',
-               'switchport trunk allowed vlan 8,9'])
-      expect(subject.set_trunk_allowed_vlans('Ethernet1', value: [8, 9]))
+        .with(['interface Ethernet1', 'switchport trunk allowed vlan 8-10,100'])
+      expect(subject.set_trunk_allowed_vlans('Ethernet1',
+                                             value: ['8-10', '100']))
         .to be_truthy
     end
 

data/spec/unit/rbeapi/api/system/default_spec.rb

@@ -68,6 +68,11 @@ describe Rbeapi::Api::System do
     it 'has four entries' do
       expect(subject.get.size).to eq(4)
     end
+
+    it 'retrieves only global ip routing' do
+      expect(subject.get.size).to eq(4)
+      expect(subject.get[:iprouting]).to eq(true)
+    end
   end
 
   describe '#set_hostname' do

data/spec/unit/rbeapi/api/system/fixture_system.text

@@ -1,5 +1,6 @@
 hostname localhost
 ip routing
+no ip routing vrf foo
 banner login
 Login Banner
 Second Line

data/spec/unit/rbeapi/api/vlans/default_spec.rb

@@ -132,4 +132,34 @@ describe Rbeapi::Api::Vlans do
       expect(subject.remove_trunk_group('1', 'foo')).to be_truthy
     end
   end
+
+  describe '#set_trunk_groups' do
+    it 'raises an ArgumentError if value is not an array' do
+      expect { subject.set_trunk_groups('vlan 1', value: 'foo') }
+        .to raise_error(ArgumentError)
+    end
+
+    it 'sets trunk group to foo bar bang' do
+      expect(node).to receive(:config)
+        .with(['vlan 1', 'trunk group foo',
+               'trunk group bar', 'trunk group bang', 'no trunk group mlag_ctl',
+               'no trunk group test'])
+      expect(subject.set_trunk_groups('1', value: %w(foo bar bang)))
+        .to be_truthy
+    end
+
+    it 'negate switchport trunk group' do
+      expect(node).to receive(:config)
+        .with(['vlan 1', 'no trunk group'])
+      expect(subject.set_trunk_groups('1', enable: false))
+        .to be_truthy
+    end
+
+    it 'default switchport trunk group' do
+      expect(node).to receive(:config)
+        .with(['vlan 1', 'default trunk group'])
+      expect(subject.set_trunk_groups('1', default: true))
+        .to be_truthy
+    end
+  end
 end

data/spec/unit/rbeapi/api/vrrp/default_spec.rb

@@ -169,6 +169,16 @@ describe Rbeapi::Api::Vrrp do
       expect { subject.create('Vlan100', 9) }.to \
         raise_error ArgumentError
     end
+
+    it 'raises an ArgumentError if secondary_ip is not an array' do
+      expect { subject.create('Ethernet1', 9, secondary_ip: '123') }
+        .to raise_error(ArgumentError)
+    end
+
+    it 'raises an ArgumentError if track is not an array' do
+      expect { subject.create('Ethernet1', 9, track: '123') }
+        .to raise_error(ArgumentError)
+    end
   end
 
   describe '#delete' do

data/spec/unit/rbeapi/client_spec.rb

@@ -209,4 +209,46 @@ describe Rbeapi::Client do
                host: 'test2')
     end
   end
+
+  describe '#get_config' do
+    def startup_config
+      "! Command: show running-config\n! device: jere-debug-agent1 (vEOS, EOS-4.14.9.1M)\n!\n! boot system flash:/vEOS-4.14.9.1M.swi\n!\nip routing vrf MGMT\n!\nmanagement api http-commands\n   no protocol https\n   protocol unix-socket\n   no shutdown\n   vrf MGMT\n      no shutdown\n!\nmanagement ssh\n   vrf MGMT\n      no shutdown\n!\n!\nend\n"
+    end
+
+    def startup_config_response
+      [{"output"=>startup_config}]
+    end
+
+    let(:node) do
+      subject.config.read(fixture_file('dut.conf'))
+      subject.connect_to('dut')
+    end
+
+    before(:each) do
+      allow(node).to receive(:run_commands) { startup_config_response }
+    end
+
+    it 'with no arguments returns the startup-config' do
+      expect(node.get_config()).to eq(startup_config.strip.split("\n"))
+    end
+
+    it 'with no arguments and an empty startup-config returns the startup-config' do
+      allow(node).to receive(:run_commands) { [{"output"=>""}] }
+      expect(node.get_config()).to eq([])
+    end
+
+    it 'with no arguments and no startup-config returns nil' do
+      msg = "CLI command 2 of 2 'show startup-config' failed: could not run command"
+      allow(node).to receive(:run_commands).and_raise(Rbeapi::Eapilib::CommandError.new(msg, 1000))
+      expect(node.get_config()).to be_nil
+    end
+
+    it 'raises invalid command error' do
+      msg = "CLI command 2 of 2 'show startup-configurations' failed: invalid command"
+      allow(node).to receive(:run_commands).and_raise(Rbeapi::Eapilib::CommandError.new(msg, 1000))
+      expect { node.get_config(config: 'running-configurations') }
+        .to raise_error Rbeapi::Eapilib::CommandError
+    end
+
+  end
 end

data/spec/unit/rbeapi/switchconfig2_spec.rb

@@ -0,0 +1,119 @@
+#
+# Copyright (c) 2016, Arista Networks, Inc.
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are
+# met:
+#
+#   Redistributions of source code must retain the above copyright notice,
+#   this list of conditions and the following disclaimer.
+#
+#   Redistributions in binary form must reproduce the above copyright
+#   notice, this list of conditions and the following disclaimer in the
+#   documentation and/or other materials provided with the distribution.
+#
+#   Neither the name of Arista Networks nor the names of its
+#   contributors may be used to endorse or promote products derived from
+#   this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL ARISTA NETWORKS
+# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
+# BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
+# OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
+# IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#
+require 'spec_helper'
+
+require 'rbeapi/switchconfig'
+
+include FixtureHelpers
+include Rbeapi::SwitchConfig
+
+describe Rbeapi::SwitchConfig::SwitchConfig do
+  test_config = <<-EOS
+! Test on multi-level children sections
+management api http-commands
+   no shutdown
+   vrf cloud-mgmt
+      no shutdown
+EOS
+  test_config_global = ['management api http-commands']
+  cmds = [['   no shutdown',
+           '   vrf cloud-mgmt'],
+          ['      no shutdown']]
+
+  subject { described_class.new(test_config) }
+
+  # SwitchConfig class methods
+  describe '#initialize' do
+    it 'returns the processed configuration' do
+      sc = subject.global
+      # Validate the global section
+      expect(sc.line).to eq('')
+      expect(sc.parent).to eq(nil)
+      expect(sc.cmds).to eq(test_config_global)
+      expect(sc.children.length).to eq(1)
+
+      # Validate the child of global
+      expect(sc.children[0].line).to eq(test_config_global[0])
+      expect(sc.children[0].parent).to eq(sc)
+      expect(sc.children[0].cmds).to eq(cmds[0])
+      expect(sc.children[0].children.length).to eq(1)
+
+      # Validate the child of global
+      child = sc.children[0].children
+      expect(child[0].line).to eq(cmds[0][1])
+      expect(child[0].parent).to eq(sc.children[0])
+      expect(child[0].cmds).to eq(cmds[1])
+      expect(child[0].children.length).to eq(0)
+    end
+  end
+
+  describe '#compare' do
+    it 'Verify compare returns array of 2 Sections' do
+      expect(subject.compare(subject)).to be_instance_of(Array)
+      expect(subject.compare(subject)[0]).to be_instance_of(Section)
+      expect(subject.compare(subject)[1]).to be_instance_of(Section)
+    end
+
+    it 'Verify compare of same switch configs' do
+      expect(subject.compare(subject)[0].line).to eq('')
+      expect(subject.compare(subject)[0].cmds).to eq([])
+      expect(subject.compare(subject)[0].children).to eq([])
+      expect(subject.compare(subject)[1].line).to eq('')
+      expect(subject.compare(subject)[1].cmds).to eq([])
+      expect(subject.compare(subject)[1].children).to eq([])
+    end
+
+    it 'Verify compare of shutdown management vrf' do
+      new_conf = <<-EOS
+management api http-commands
+   no shutdown
+   vrf cloud-mgmt
+      shutdown
+EOS
+      org_new_diff = <<-EOS
+management api http-commands
+   vrf cloud-mgmt
+      no shutdown
+EOS
+      new_org_diff = <<-EOS
+management api http-commands
+   vrf cloud-mgmt
+      shutdown
+EOS
+      swc_new = Rbeapi::SwitchConfig::SwitchConfig.new(new_conf)
+      swc_org_new = Rbeapi::SwitchConfig::SwitchConfig.new(org_new_diff)
+      swc_new_org = Rbeapi::SwitchConfig::SwitchConfig.new(new_org_diff)
+      expect(subject.compare(swc_new)[0]).to section_equal(swc_org_new.global)
+      expect(subject.compare(swc_new)[1]).to section_equal(swc_new_org.global)
+    end
+  end
+end