tests: add test cases for Client

1 files changed, 6 insertions, 3 deletions

diff --git a/test/utils/string_socket.rb b/test/utils/string_socket.rb
index fb59172..0cffa64 100644
--- a/test/utils/string_socket.rb
+++ b/test/utils/string_socket.rb
@@ -1,12 +1,15 @@
-class StringSocket
+class StringSocket < IO
+  # remove all methods
+  (IO.instance_methods - Object.instance_methods).each { |symbol| undef_method symbol }
+
   extend Forwardable
   def_delegators :@rio, :readpartial
   def_delegators :@wio, :<<, :write
 
   attr_reader :rio, :wio
 
-  def initialize(str)
-    @rio = StringIO.new(str)
+  def initialize(str = nil)
+    @rio = StringIO.new(str.to_s)
     @wio = StringIO.new
   end
 end