module Net::SSH::Test::Extensions::IO

def self.with_test_extension(&block)
  orig_value = @extension_enabled
  @extension_enabled = true
  begin
    yield
  ensure
    @extension_enabled = orig_value
  end
end