Added IPv6#each, IPv6#broadcast_u128 and Prefix128#host_prefix

1 files changed, 26 insertions, 1 deletions

diff --git a/test/ipaddress/ipv6_test.rb b/test/ipaddress/ipv6_test.rb
index 4c84e4b..898938f 100644
--- a/test/ipaddress/ipv6_test.rb
+++ b/test/ipaddress/ipv6_test.rb
@@ -127,7 +127,22 @@ class IPv6Test < Test::Unit::TestCase
   def test_method_network_u128
     assert_equal 42540766411282592856903984951653826560, @ip.network_u128
   end
-  
+
+  def test_method_broadcast_u128
+    assert_equal 42540766411282592875350729025363378175, @ip.broadcast_u128
+  end
+
+  def test_method_size
+    ip = @klass.new("2001:db8::8:800:200c:417a/64")
+    assert_equal 2**64, ip.size
+    ip = @klass.new("2001:db8::8:800:200c:417a/32")
+    assert_equal 2**96, ip.size
+    ip = @klass.new("2001:db8::8:800:200c:417a/120")
+    assert_equal 2**8, ip.size
+    ip = @klass.new("2001:db8::8:800:200c:417a/124")
+    assert_equal 2**4, ip.size
+  end
+
   def test_method_include?
     assert_equal true, @ip.include?(@ip)
     # test prefix on same address
@@ -199,6 +214,16 @@ class IPv6Test < Test::Unit::TestCase
       assert_equal net, ip.network.to_string
     end
   end
+
+  def test_method_each
+    ip = @klass.new("2001:db8::4/125")
+    arr = []
+    ip.each {|i| arr << i.compressed}
+    expected = ["2001:db8::","2001:db8::1","2001:db8::2",
+                "2001:db8::3","2001:db8::4","2001:db8::5",
+                "2001:db8::6","2001:db8::7"]
+    assert_equal expected, arr
+  end
   
   def test_classmethod_expand
     compressed = "2001:db8:0:cd30::"