add SocketMethods#kgio_addr!

This refreshes (or sets) the @kgio_addr ivar for sockets
that didn't go through kgio_accept or kgio_tryaccept.

1 files changed, 19 insertions, 0 deletions

diff --git a/test/test_kgio_addr.rb b/test/test_kgio_addr.rb
new file mode 100644
index 0000000..8650f5e
--- /dev/null
+++ b/test/test_kgio_addr.rb
@@ -0,0 +1,19 @@
+# -*- encoding: binary -*-
+require 'test/unit'
+$-w = true
+require 'kgio'
+
+class TestKgioAddr < Test::Unit::TestCase
+  def test_tcp
+    addr = ENV["TEST_HOST"] || '127.0.0.1'
+    tcp = TCPServer.new(addr, 0)
+    port = tcp.addr[1]
+    client = Kgio::TCPSocket.new(addr, port)
+    accepted = tcp.accept
+    assert ! accepted.instance_eval { defined?(@kgio_addr) }
+    accepted.extend Kgio::SocketMethods
+    s = accepted.kgio_addr!
+    assert_equal addr, s
+    assert_equal addr, accepted.instance_variable_get(:@kgio_addr)
+  end
+end