Fix: Addrinfo#ip_address

crystal-lang · Dec 27, 2016 · 48af685 · 48af685
1 parent cd51ef6
commit 48af685
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/spec/std/socket_spec.cr b/spec/std/socket_spec.cr
@@ -114,6 +114,13 @@ describe Socket::Addrinfo do
       end
     end
   end
+
+  describe "#ip_address" do
+    assert do
+      addrinfos = Socket::Addrinfo.udp("localhost", 80)
+      typeof(addrinfos.first.ip_address).should eq(Socket::IPAddress)
+    end
+  end
 end
 
 describe Socket::IPAddress do

diff --git a/src/socket/addrinfo.cr b/src/socket/addrinfo.cr
@@ -153,9 +153,11 @@ class Socket
       end
     end
 
+    @ip_address : IPAddress?
+
     # Returns an `IPAddress` matching this addrinfo.
     def ip_address
-      @ip_address = IPAddress.from(@addr, @addrlen)
+      @ip_address ||= IPAddress.from(to_unsafe, size)
     end
 
     def to_unsafe