cisco_node_utils_mgx 2.1.0.1

data/tests/test_pim.rb

@@ -0,0 +1,203 @@
+# Copyright (c) 2013-2016 Cisco and/or its affiliates.
+#
+# Smitha Gopalan, November 2015
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#-------------------------------------------------------------------------------
+# CLI: <afi> pim ssm-range <range>  (under different VRFs)
+#-------------------------------------------------------------------------------
+# Testcases: All Tests create all instances within a test
+# 1. test_single_ssm_range_single_vrf:
+#         vrf default, ip pim ssm-range 229.0.0.0/8
+#
+# 2. test_single_ssm_range_none_single_vrf:
+#         vrf default, ip pim ssm-range none
+#
+# 3. test_multiple_ssm_range_multiple_vrfs:
+#         vrf default, ip pim ssm-range 229.0.0.0/8 225.0.0.0/8 224.0.0.0/8
+#         vrf red, ip pim ssm-range 230.0.0.0/8 228.0.0.0/8 224.0.0.0/8
+#         vrf black, ip pim ssm-range none
+#
+# 4. test_multiple_ssm_range_overwrite_multiple_vrfs:
+#         vrf default, ip pim ssm-range 229.0.0.0/8 225.0.0.0/8 224.0.0.0/8
+#         vrf red, ip pim ssm-range 230.0.0.0/8 228.0.0.0/8 224.0.0.0/8 ->
+#             gets overwritten to
+#         vrf red, ip pim ssm-range none
+#
+# 5. test_single_invalid_ssm_range_single_vrf:
+#         vrf default, ip pim ssm-range 1.1.1.1/8
+#-------------------------------------------------------------------------------
+
+require_relative 'ciscotest'
+require_relative '../lib/cisco_node_utils/pim'
+
+include Cisco
+
+# TestPim - Minitest for Pim Feature
+class TestPim < CiscoTestCase
+  @skip_unless_supported = 'pim'
+
+  @@pre_clean_needed = true # rubocop:disable Style/ClassVars
+
+  # Enables feature pim
+  #---------------------
+  def setup
+    super
+    remove_all_pims if @@pre_clean_needed
+    @@pre_clean_needed = false # rubocop:disable Style/ClassVars
+  end
+
+  def teardown
+    super
+    remove_all_pims
+  end
+
+  def remove_all_pims
+    Pim.pims.each do |_afi, vrfs|
+      vrfs.each do |_vrf, obj|
+        obj.destroy
+      end
+    end
+  end
+
+  # Creates single ssm range under default vrf
+  #-----------------------------------------------
+  def create_single_ssm_range_single_vrf(afi)
+    range = '229.0.0.0/8'
+    p1 = Pim.new(afi, 'default')
+    p1.ssm_range = (range)
+    assert_equal(p1.ssm_range.split.sort.join(' '), range.split.sort.join(' '))
+    p1.ssm_range = 'none'
+    assert_equal('none', p1.ssm_range)
+  end
+
+  # Tests single ssm range none under default vrf
+  #-----------------------------------------------
+  def test_ssm_range
+    %w(ipv4).each do |afi|
+      create_single_ssm_range_single_vrf(afi)
+    end
+  end
+
+  # Creates single ssm range none under default vrf
+  #-----------------------------------------------
+  def create_single_ssm_range_none_single_vrf(afi)
+    range = 'none'
+    p1 = Pim.new(afi, 'default')
+    p1.ssm_range = (range)
+    assert_equal(p1.ssm_range.split.sort.join(' '), range.split.sort.join(' '))
+    p1.ssm_range = 'none'
+    assert_equal('none', p1.ssm_range)
+  end
+
+  # Tests single ssm range none under default vrf
+  #-----------------------------------------------
+  def test_ssm_range_none
+    %w(ipv4).each do |afi|
+      create_single_ssm_range_none_single_vrf(afi)
+    end
+  end
+
+  # Creates multiple ssm ranges under different vrfs
+  #-----------------------------------------------
+  def create_multiple_ssm_range_multiple_vrfs(afi)
+    range1 = '229.0.0.0/8 225.0.0.0/8 224.0.0.0/8'
+    range2 = '230.0.0.0/8 228.0.0.0/8 224.0.0.0/8'
+    range3 = 'none'
+    p1 = Pim.new(afi, 'default')
+    p2 = Pim.new(afi, 'red')
+    p3 = Pim.new(afi, 'black')
+    p1.ssm_range = (range1)
+    p2.ssm_range = (range2)
+    p3.ssm_range = (range3)
+    assert_equal(p1.ssm_range.split.sort.join(' '), range1.split.sort.join(' '))
+    assert_equal(p2.ssm_range.split.sort.join(' '), range2.split.sort.join(' '))
+    assert_equal(p3.ssm_range.split.sort.join(' '), range3.split.sort.join(' '))
+    p1.ssm_range = 'none'
+    assert_equal('none', p1.ssm_range)
+    p2.ssm_range = 'none'
+    assert_equal('none', p2.ssm_range)
+  end
+
+  # Tests multiple ssm ranges under different vrfs
+  #-----------------------------------------------
+  def test_ssm_range_multiple
+    %w(ipv4).each do |afi|
+      create_multiple_ssm_range_multiple_vrfs(afi)
+    end
+  end
+
+  # Creates multiple ssm ranges overwrite under different vrfs
+  #-----------------------------------------------
+  def create_multiple_ssm_range_overwrite_multiple_vrfs(afi)
+    range1 = '229.0.0.0/8 225.0.0.0/8 224.0.0.0/8'
+    range2 = '230.0.0.0/8 228.0.0.0/8 224.0.0.0/8'
+    range3 = 'none'
+    p1 = Pim.new(afi, 'default')
+    p2 = Pim.new(afi, 'red')
+    p1.ssm_range = (range1)
+    p2.ssm_range = (range2)
+    assert_equal(p1.ssm_range.split.sort.join(' '), range1.split.sort.join(' '))
+    assert_equal(p2.ssm_range.split.sort.join(' '), range2.split.sort.join(' '))
+    p2.ssm_range = (range3)
+    assert_equal(p2.ssm_range.split.sort.join(' '), range3.split.sort.join(' '))
+
+    p1.ssm_range = 'none'
+    assert_equal('none', p1.ssm_range)
+    p1.destroy
+    p2.destroy
+  end
+
+  # Tests multiple ssm ranges overwrite under different vrfs
+  #-----------------------------------------------
+  def test_ssm_range_multiple_overwrite
+    %w(ipv4).each do |afi|
+      create_multiple_ssm_range_overwrite_multiple_vrfs(afi)
+    end
+  end
+
+  # Creates single invalid ssm range under vrf default
+  #---------------------------------------------------
+  def create_single_invalid_ssm_range_single_vrf(afi)
+    range = '1.1.1.1/8'
+    p1 = Pim.new(afi, 'default')
+    assert_raises(CliError) do
+      p1.ssm_range = (range)
+    end
+  end
+
+  # Tests single invalid ssm range under vrf default
+  #---------------------------------------------------
+  def test_ssm_range_inv
+    %w(ipv4).each do |afi|
+      create_single_invalid_ssm_range_single_vrf(afi)
+    end
+  end
+
+  def test_bfd
+    %w(ipv4).each do |afi|
+      p1 = Pim.new(afi, 'default')
+      p2 = Pim.new(afi, 'red')
+      assert_equal(p1.default_bfd, p1.bfd)
+      assert_equal(p2.default_bfd, p1.bfd)
+      p1.bfd = true
+      p2.bfd = true
+      assert_equal(true, p1.bfd)
+      assert_equal(true, p2.bfd)
+      p1.bfd = p1.default_bfd
+      p2.bfd = p2.default_bfd
+      assert_equal(p1.default_bfd, p1.bfd)
+      assert_equal(p2.default_bfd, p1.bfd)
+    end
+  end
+end

data/tests/test_pim_group_list.rb

@@ -0,0 +1,147 @@
+# Copyright (c) 2013-2016 Cisco and/or its affiliates.
+#
+# Smitha Gopalan, November 2015
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#-----------------------------------------------------------
+# CLI: <afi> pim rp-address <rp-address> group-list <group>
+#      (under different VRFs)
+#-----------------------------------------------------------
+# Testcases: All Tests create and destroy all instances within a test
+#
+# 1. test_all_group_lists
+#
+# 2. test_single_rpaddr_single_invalid_grouplist_single_vrf:
+#         vrf default, ip pim rp-address 25.25.25.25 group-list 25.0.0.0/8
+#-------------------------------------------------------------------------------
+
+require_relative 'ciscotest'
+require_relative '../lib/cisco_node_utils/pim_group_list'
+
+include Cisco
+
+# TestPim - Minitest for PimGrouplist
+class TestPimGrpLst < CiscoTestCase
+  @skip_unless_supported = 'pim'
+
+  # Enables feature pim
+  #--------------------
+  def setup
+    super
+    config('no feature pim')
+  end
+
+  # Test Pim.group_lists
+  # - multiple vrfs, groups, rp_addrs
+  # - same rp_addr, different group
+  # - same rp_addr, different vrf
+  # - same group, different rp_addr
+  # - same group, different vrf
+  #--------------------------------------------
+  def all_group_lists(afi)
+    rp_addr1 = '11.11.11.11'
+    rp_addr2 = '22.22.22.22'
+    grouplist1 = '227.0.0.0/8'
+    grouplist2 = '228.0.0.0/8'
+
+    # Basic setup
+    vrf = 'default'
+    pd1 = PimGroupList.new(afi, vrf, rp_addr1, grouplist1)
+    pd2 = PimGroupList.new(afi, vrf, rp_addr2, grouplist2)
+    hash = PimGroupList.group_lists
+    assert(hash.key?(afi))
+    assert(hash[afi].key?(vrf))
+    assert_equal(2, hash[afi][vrf].keys.count,
+                 "hash[#{afi}][#{vrf}] should have 2 group lists")
+    assert(hash[afi][vrf].key?([rp_addr1, grouplist1]),
+           "hash[#{afi}][#{vrf}] does not contain [#{rp_addr1}, #{grouplist1}]")
+    assert(hash[afi][vrf].key?([rp_addr2, grouplist2]),
+           "hash[#{afi}][#{vrf}] does not contain [#{rp_addr2}, #{grouplist2}]")
+
+    # vrf with same rp_addrs/groups as in default
+    vrf = 'red'
+    pd3 = PimGroupList.new(afi, vrf, rp_addr1, grouplist1)
+    pd4 = PimGroupList.new(afi, vrf, rp_addr2, grouplist2)
+    hash = PimGroupList.group_lists
+    assert_equal(2, hash[afi].keys.count,
+                 "hash[#{afi}] does not have 2 vrfs")
+    assert(hash[afi].key?(vrf))
+    assert_equal(2, hash[afi][vrf].keys.count,
+                 "hash[#{afi}][#{vrf}] should have 2 group lists")
+    assert(hash[afi][vrf].key?([rp_addr1, grouplist1]),
+           "hash[#{afi}][#{vrf}] does not contain [#{rp_addr1}, #{grouplist1}]")
+    assert(hash[afi][vrf].key?([rp_addr2, grouplist2]),
+           "hash[#{afi}][#{vrf}] does not contain [#{rp_addr2}, #{grouplist2}]")
+
+    # different vrf, same rps / diff groups, etc.
+    vrf = 'black'
+    rp_addr7 = '7.7.7.7'
+    pd5 = PimGroupList.new(afi, vrf, rp_addr1, grouplist1)
+    pd6 = PimGroupList.new(afi, vrf, rp_addr1, grouplist2)
+    pd7 = PimGroupList.new(afi, vrf, rp_addr7, grouplist2)
+    hash = PimGroupList.group_lists
+    assert_equal(3, hash[afi].keys.count,
+                 "hash[#{afi}] does not have 3 vrfs")
+    assert(hash[afi].key?(vrf))
+    assert_equal(3, hash[afi][vrf].keys.count,
+                 "hash[#{afi}][#{vrf}] should have 3 group lists")
+    assert(hash[afi][vrf].key?([rp_addr1, grouplist1]),
+           "hash[#{afi}][#{vrf}] does not contain [#{rp_addr1}, #{grouplist2}]")
+    assert(hash[afi][vrf].key?([rp_addr1, grouplist2]),
+           "hash[#{afi}][#{vrf}] does not contain [#{rp_addr1}, #{grouplist2}]")
+    assert(hash[afi][vrf].key?([rp_addr7, grouplist2]),
+           "hash[#{afi}][#{vrf}] does not contain [#{rp_addr7}, #{grouplist2}]")
+
+    # Test removal
+    vrf = 'default'
+    pd1.destroy
+    hash = PimGroupList.group_lists
+    assert_equal(1, hash[afi][vrf].keys.count,
+                 "hash[#{afi}][#{vrf}] should have 1 group lists")
+    refute(hash[afi][vrf].key?([rp_addr1, grouplist1]),
+           "hash[#{afi}][#{vrf}] should not contain "\
+           "[#{rp_addr1}, #{grouplist1}]")
+
+    vrf = 'red'
+    pd3.destroy
+    hash = PimGroupList.group_lists
+    assert_equal(1, hash[afi][vrf].keys.count,
+                 "hash[#{afi}][#{vrf}] should have 1 group lists")
+    refute(hash[afi][vrf].key?([rp_addr1, grouplist1]),
+           "hash[#{afi}][#{vrf}] should not contain "\
+           "[#{rp_addr1}, #{grouplist1}]")
+
+    vrf = 'black'
+    pd5.destroy
+    pd7.destroy
+    hash = PimGroupList.group_lists
+    assert_equal(1, hash[afi][vrf].keys.count,
+                 "hash[#{afi}][#{vrf}] should have 1 group lists")
+    refute(hash[afi][vrf].key?([rp_addr1, grouplist1]),
+           "hash[#{afi}][#{vrf}] should not contain "\
+           "[#{rp_addr1}, #{grouplist1}]")
+    pd2.destroy
+    pd4.destroy
+    pd6.destroy
+    hash = PimGroupList.group_lists
+    assert_empty(hash[afi], 'hash[#{afi}] is not empty')
+  end
+
+  # Tests Pim.group_lists
+  #------------------------------------------
+  def test_all_group_lists
+    %w(ipv4).each do |afi|
+      all_group_lists(afi)
+    end
+  end
+end

data/tests/test_pim_rp_address.rb

@@ -0,0 +1,155 @@
+# Copyright (c) 2013-2016 Cisco and/or its affiliates.
+#
+# Smitha Gopalan, November 2015
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ---------------------------------------------------------------------
+# Cli: <afi> pim rp-address <rp-address> (under different VRFs)
+# ---------------------------------------------------------------------
+# Testcases: All Tests create and destroy all instances within a test
+#
+# 1. test_all_rp_addrs
+#
+# 2. test_single_invalid_rpaddr_single_vrf:
+#         vrf default, ip pim rp-address 256.256.256.256
+# ---------------------------------------------------------------------
+
+require_relative 'ciscotest'
+require_relative '../lib/cisco_node_utils/pim_rp_address'
+
+include Cisco
+
+# TestPimRpAddress - Minitest for PimRpAddress
+class TestPimRpAddress < CiscoTestCase
+  @skip_unless_supported = 'pim'
+
+  # Enables feature pim
+  #--------------------
+  def setup
+    super
+    config('no feature pim')
+  end
+
+  # Test PimRpAddress.rp_addresses
+  # - multiple vrfs, rp_addrs
+  # - same rp_addr, different vrf
+  #--------------------------------------------
+  def all_rp_addrs(afi)
+    rp_addr1 = '11.11.11.11'
+    rp_addr2 = '22.22.22.22'
+
+    # Basic setup
+    vrf = 'default'
+    pd1 = PimRpAddress.new(afi, vrf, rp_addr1)
+    pd2 = PimRpAddress.new(afi, vrf, rp_addr2)
+    hash = PimRpAddress.rp_addresses
+    assert(hash.key?(afi))
+    assert(hash[afi].key?(vrf))
+    assert_equal(2, hash[afi][vrf].keys.count,
+                 "hash[#{afi}][#{vrf}] should have 2 rp-address")
+    assert(hash[afi][vrf].key?(rp_addr1),
+           "hash[#{afi}][#{vrf}] does not contain #{rp_addr1}")
+    assert(hash[afi][vrf].key?(rp_addr2),
+           "hash[#{afi}][#{vrf}] does not contain #{rp_addr2}")
+
+    # vrf with same rp_addrs as in default
+    vrf = 'red'
+    pd3 = PimRpAddress.new(afi, vrf, rp_addr1)
+    pd4 = PimRpAddress.new(afi, vrf, rp_addr2)
+    hash = PimRpAddress.rp_addresses
+    assert_equal(2, hash[afi].keys.count,
+                 "hash[#{afi}] does not have 2 vrfs")
+    assert(hash[afi].key?(vrf))
+    assert_equal(2, hash[afi][vrf].keys.count,
+                 "hash[#{afi}][#{vrf}] should have 2 rp-address")
+    assert(hash[afi][vrf].key?(rp_addr1),
+           "hash[#{afi}][#{vrf}] does not contain #{rp_addr1}")
+    assert(hash[afi][vrf].key?(rp_addr2),
+           "hash[#{afi}][#{vrf}] does not contain #{rp_addr2}")
+
+    # different vrf, same/diff rps
+    vrf = 'black'
+    rp_addr7 = '7.7.7.7'
+    pd5 = PimRpAddress.new(afi, vrf, rp_addr1)
+    pd6 = PimRpAddress.new(afi, vrf, rp_addr2)
+    pd7 = PimRpAddress.new(afi, vrf, rp_addr7)
+    hash = PimRpAddress.rp_addresses
+    assert_equal(3, hash[afi].keys.count,
+                 "hash[#{afi}] does not have 3 vrfs")
+    assert(hash[afi].key?(vrf))
+    assert_equal(3, hash[afi][vrf].keys.count,
+                 "hash[#{afi}][#{vrf}] should have 3 rp-address")
+    assert(hash[afi][vrf].key?(rp_addr1),
+           "hash[#{afi}][#{vrf}] does not contain #{rp_addr1}")
+    assert(hash[afi][vrf].key?(rp_addr1),
+           "hash[#{afi}][#{vrf}] does not contain #{rp_addr2}")
+    assert(hash[afi][vrf].key?(rp_addr7),
+           "hash[#{afi}][#{vrf}] does not contain #{rp_addr7}")
+
+    # Test removal
+    vrf = 'default'
+    pd1.destroy
+    hash = PimRpAddress.rp_addresses
+    assert_equal(1, hash[afi][vrf].keys.count,
+                 "hash[#{afi}][#{vrf}] should have 1 rp-address")
+    refute(hash[afi][vrf].key?(rp_addr1),
+           "hash[#{afi}][#{vrf}] should not contain #{rp_addr1}")
+
+    vrf = 'red'
+    pd3.destroy
+    hash = PimRpAddress.rp_addresses
+    assert_equal(1, hash[afi][vrf].keys.count,
+                 "hash[#{afi}][#{vrf}] should have 1 rp-address")
+    refute(hash[afi][vrf].key?(rp_addr1),
+           "hash[#{afi}][#{vrf}] should not contain #{rp_addr1}")
+
+    vrf = 'black'
+    pd5.destroy
+    pd7.destroy
+    hash = PimRpAddress.rp_addresses
+    assert_equal(1, hash[afi][vrf].keys.count,
+                 "hash[#{afi}][#{vrf}] should have 1 rp-address")
+    refute(hash[afi][vrf].key?([rp_addr1]),
+           "hash[#{afi}][#{vrf}] should not contain [#{rp_addr1}]")
+    pd2.destroy
+    pd4.destroy
+    pd6.destroy
+    hash = PimRpAddress.rp_addresses
+    assert_empty(hash[afi], 'hash[#{afi}] is not empty')
+  end
+
+  # Tests PimRpAddress.rp_addresses
+  #------------------------------------------
+  def test_all_rp_addrs
+    %w(ipv4).each do |afi|
+      all_rp_addrs(afi)
+    end
+  end
+
+  # Creates single invalid rp address under vrf default
+  #---------------------------------------------------
+  def create_single_invalid_rpaddr_single_vrf(afi)
+    rp_addr = '256.256.256.256'
+    assert_raises(CliError) do
+      PimRpAddress.new(afi, 'default', rp_addr)
+    end
+  end
+
+  # Tests single invalid rp address under vrf default
+  #---------------------------------------------------
+  def test_rpaddr
+    %w(ipv4).each do |afi|
+      create_single_invalid_rpaddr_single_vrf(afi)
+    end
+  end
+end