nonnative 1.22.0 → 1.23.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5a6ee4ae9e2629fcae27b82ea0f4db839d6c131ab1687ed286b33eb80fc5a41a
-  data.tar.gz: 53192dec8216ce696fe9774f77b98904a6f13b041cbbe131bbaacaafc45cffda
+  metadata.gz: 2265811d2fdb090bd12cb35065038272974519ad6a7758bb2b7c281ca7b804c1
+  data.tar.gz: c08f324e5de548b1457f3ff16585569f7917340b6367dc0b9d1cc1ceae2d4645
 SHA512:
-  metadata.gz: 6da13105ad656ebfe2f5bc5c1ba643e12776b2747da9453c3e1da62729475e2490d67942482b7f0a9de480ccebd245fe798cb57cf5db4ee6ef483781b8657be4
-  data.tar.gz: 787afe51f6eed81281d439f91824c795a96f4673712302983977c976df949e23eeeee58a3d3c5e71829f4c69b6b8931531488f7bf6a7674c3156322b81d10195
+  metadata.gz: dfe2d40e5c27f6d9425fbfc7059ce3d3723fdb00451a7a72a399d359627f7424699b044115848c45a7e180606919f81d03a439d6942acfbbcd8584350e35499a
+  data.tar.gz: 6b2221ac50779bf621e5377ae1940f0621e4f1b4d6349aead6f9adac575d37f9f3d04e12aa65f9118fdc8b277d8021e653566c77cc1dee2815ef501d8b936073

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    nonnative (1.22.0)
+    nonnative (1.23.0)
       concurrent-ruby (~> 1.0, >= 1.0.5)
       cucumber (>= 4, < 5)
       grpc (>= 1, < 2)

data/README.md

@@ -297,8 +297,8 @@ Nonnative.load_configuration('configuration.yml')
 #### Proxies
 
 We allow different proxies to be configured. These proxies can be used to simulate all kind of situations. The proxies that can be configured are:
-- none (this is the default)
-- chaos
+- `none` (this is the default)
+- `chaos`
 
 Setup it up programmatically:
 
@@ -328,3 +328,20 @@ servers:
       type: chaos
       port: 20000
 ```
+
+##### Fault Injection
+
+The `chaos` proxy allows you to simulate failures by injecting them. We currently support the following:
+- `close_all` - Closes the socket as soon as it connects.
+- `delay` - This delays the communication between the connection.
+- `invalid_data` - This takes the input and rearranges it to produce invalid data.
+
+Setup it up programmatically:
+
+```ruby
+name = 'name of server in configuration'
+server = Nonnative.pool.server_by_name(name)
+
+server.proxy.close_all # To use close_all.
+server.proxy.reset # To reset it back to a good state.
+```

data/lib/nonnative.rb

@@ -35,6 +35,11 @@ require 'nonnative/proxy_factory'
 require 'nonnative/proxy'
 require 'nonnative/no_proxy'
 require 'nonnative/chaos_proxy'
+require 'nonnative/socket_pair'
+require 'nonnative/close_all_socket_pair'
+require 'nonnative/delay_socket_pair'
+require 'nonnative/invalid_data_socket_pair'
+require 'nonnative/socket_pair_factory'
 
 module Nonnative
   class << self

data/lib/nonnative/chaos_proxy.rb

@@ -24,6 +24,14 @@ module Nonnative
       apply_state :close_all
     end
 
+    def delay
+      apply_state :delay
+    end
+
+    def invalid_data
+      apply_state :invalid_data
+    end
+
     def reset
       apply_state :none
     end
@@ -38,57 +46,21 @@ module Nonnative
 
     def perform_start
       loop do
-        thread = Thread.start(tcp_server.accept) { |local_socket| connect(local_socket) }
+        thread = Thread.start(tcp_server.accept) do |local_socket|
+          SocketPairFactory.create(read_state, port).connect(local_socket)
+          connections.delete(Thread.current.object_id)
+        end
+        thread.report_on_exception = false
         connections[thread.object_id] = thread
       end
     end
 
-    def connect(local_socket)
-      return local_socket.close if state?(:close_all)
-
-      remote_socket = create_remote_socket
-      return unless remote_socket
-
-      loop do
-        ready = select([local_socket, remote_socket], nil, nil)
-
-        break if write(ready, local_socket, remote_socket)
-        break if write(ready, remote_socket, local_socket)
-      end
-    rescue Errno::ECONNRESET
-      # Just ignore it.
-    ensure
-      local_socket.close
-      remote_socket&.close
-      connections.delete(Thread.current.object_id)
-    end
-
-    def create_remote_socket
-      timeout.perform do
-        ::TCPSocket.new('0.0.0.0', port)
-      rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH
-        sleep 0.01
-        retry
-      end
-    end
-
-    def write(ready, socket1, socket2)
-      if ready[0].include?(socket1)
-        data = socket1.recv(1024)
-        return true if data.empty?
-
-        socket2.write(data)
-      end
-
-      false
-    end
-
     def apply_state(state)
       mutex.synchronize { @state = state }
     end
 
-    def state?(state)
-      mutex.synchronize { @state == state }
+    def read_state
+      mutex.synchronize { state }
     end
   end
 end

data/lib/nonnative/close_all_socket_pair.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module Nonnative
+  class CloseAllSocketPair < SocketPair
+    def connect(local_socket)
+      local_socket.close
+    end
+  end
+end

data/lib/nonnative/delay_socket_pair.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module Nonnative
+  class DelaySocketPair < SocketPair
+    def read(socket)
+      sleep 2
+
+      super socket
+    end
+  end
+end

data/lib/nonnative/http_client.rb

@@ -8,31 +8,33 @@ module Nonnative
 
     protected
 
-    def get(pathname, headers = {})
+    def get(pathname, headers = {}, timeout = 60)
       with_exception do
         uri = URI.join(host, pathname)
-        RestClient.get(uri.to_s, headers)
+        RestClient::Request.execute(method: :get, url: uri.to_s, headers: headers, timeout: timeout)
       end
     end
 
-    def post(pathname, payload, headers = {})
+    def post(pathname, payload, headers = {}, timeout = 60)
       with_exception do
         uri = URI.join(host, pathname)
-        RestClient.post(uri.to_s, payload.to_json, headers)
+        RestClient::Request.execute(method: :post, url: uri.to_s, payload: payload.to_json, headers: headers,
+                                    timeout: timeout)
       end
     end
 
-    def delete(pathname, headers = {})
+    def delete(pathname, headers = {}, timeout = 60)
       with_exception do
         uri = URI.join(host, pathname)
-        RestClient.delete(uri.to_s, headers)
+        RestClient::Request.execute(method: :delete, url: uri.to_s, headers: headers, timeout: timeout)
       end
     end
 
-    def put(pathname, payload, headers = {})
+    def put(pathname, payload, headers = {}, timeout = 60)
       with_exception do
         uri = URI.join(host, pathname)
-        RestClient.put(uri.to_s, payload.to_json, headers)
+        RestClient::Request.execute(method: :put, url: uri.to_s, payload: payload.to_json, headers: headers,
+                                    timeout: timeout)
       end
     end
 
@@ -42,6 +44,8 @@ module Nonnative
 
     def with_exception
       yield
+    rescue RestClient::Exceptions::ReadTimeout => e
+      raise e
     rescue RestClient::ExceptionWithResponse => e
       e.response
     end

data/lib/nonnative/invalid_data_socket_pair.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module Nonnative
+  class InvalidDataSocketPair < SocketPair
+    def write(socket, data)
+      data = data.chars.shuffle.join
+
+      super socket, data
+    end
+  end
+end

data/lib/nonnative/proxy.rb

@@ -4,11 +4,10 @@ module Nonnative
   class Proxy
     def initialize(service)
       @service = service
-      @timeout = Nonnative::Timeout.new(service.timeout)
     end
 
     protected
 
-    attr_reader :service, :timeout
+    attr_reader :service
   end
 end

data/lib/nonnative/socket_pair.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+module Nonnative
+  class SocketPair
+    def initialize(port)
+      @port = port
+    end
+
+    def connect(local_socket)
+      remote_socket = create_remote_socket
+
+      loop do
+        ready = select([local_socket, remote_socket], nil, nil)
+
+        break if pipe(ready, local_socket, remote_socket)
+        break if pipe(ready, remote_socket, local_socket)
+      end
+    ensure
+      local_socket.close
+      remote_socket&.close
+    end
+
+    protected
+
+    attr_reader :port
+
+    def create_remote_socket
+      ::TCPSocket.new('0.0.0.0', port)
+    end
+
+    def pipe(ready, socket1, socket2)
+      if ready[0].include?(socket1)
+        data = read(socket1)
+        return true if data.empty?
+
+        write socket2, data
+      end
+
+      false
+    end
+
+    def read(socket)
+      socket.recv(1024)
+    end
+
+    def write(socket, data)
+      socket.write(data)
+    end
+  end
+end

data/lib/nonnative/socket_pair_factory.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module Nonnative
+  class SocketPairFactory
+    class << self
+      def create(type, port)
+        case type
+        when :close_all
+          CloseAllSocketPair.new(port)
+        when :delay
+          DelaySocketPair.new(port)
+        when :invalid_data
+          InvalidDataSocketPair.new(port)
+        else
+          SocketPair.new(port)
+        end
+      end
+    end
+  end
+end

data/lib/nonnative/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Nonnative
-  VERSION = '1.22.0'
+  VERSION = '1.23.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: nonnative
 version: !ruby/object:Gem::Version
-  version: 1.22.0
+  version: 1.23.0
 platform: ruby
 authors:
 - Alex Falkowski
-autorequire:
+autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-07-10 00:00:00.000000000 Z
+date: 2020-07-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: concurrent-ruby
@@ -284,15 +284,18 @@ files:
 - lib/nonnative.rb
 - lib/nonnative/before.rb
 - lib/nonnative/chaos_proxy.rb
+- lib/nonnative/close_all_socket_pair.rb
 - lib/nonnative/command.rb
 - lib/nonnative/configuration.rb
 - lib/nonnative/configuration_process.rb
 - lib/nonnative/configuration_proxy.rb
 - lib/nonnative/configuration_server.rb
+- lib/nonnative/delay_socket_pair.rb
 - lib/nonnative/error.rb
 - lib/nonnative/grpc_server.rb
 - lib/nonnative/http_client.rb
 - lib/nonnative/http_server.rb
+- lib/nonnative/invalid_data_socket_pair.rb
 - lib/nonnative/manual.rb
 - lib/nonnative/no_proxy.rb
 - lib/nonnative/observability.rb
@@ -302,6 +305,8 @@ files:
 - lib/nonnative/proxy_factory.rb
 - lib/nonnative/server.rb
 - lib/nonnative/service.rb
+- lib/nonnative/socket_pair.rb
+- lib/nonnative/socket_pair_factory.rb
 - lib/nonnative/start_error.rb
 - lib/nonnative/startup.rb
 - lib/nonnative/stop_error.rb
@@ -313,7 +318,7 @@ homepage: https://github.com/alexfalkowski/nonnative
 licenses:
 - Unlicense
 metadata: {}
-post_install_message:
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -328,8 +333,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
-signing_key:
+rubygems_version: 3.0.8
+signing_key: 
 specification_version: 4
 summary: Allows you to keep using the power of ruby to test other systems
 test_files: []