spikard 0.1.2 → 0.2.1

data/lib/spikard/testing.rb

@@ -1,220 +1,221 @@
-# frozen_string_literal: true
-
-require 'json'
-
-module Spikard
-  # Testing helpers that wrap the native Ruby extension.
-  module Testing
-    module_function
-
-    def create_test_client(app, config: nil)
-      unless defined?(Spikard::Native::TestClient)
-        raise LoadError, 'Spikard native test client is not available. Build the native extension before running tests.'
-      end
-
-      # Allow generated apps to stash a test config
-      if config.nil? && app.instance_variable_defined?(:@__spikard_test_config)
-        config = app.instance_variable_get(:@__spikard_test_config)
-      end
-
-      # Use default config if none provided
-      config ||= Spikard::ServerConfig.new
-
-      routes_json = JSON.generate(app.route_metadata)
-      handlers = app.handler_map.transform_keys(&:to_sym)
-      ws_handlers = app.websocket_handlers || {}
-      sse_producers = app.sse_producers || {}
-      native = Spikard::Native::TestClient.new(routes_json, handlers, config, ws_handlers, sse_producers)
-      TestClient.new(native)
-    end
-
-    # High level wrapper around the native test client.
-    class TestClient
-      def initialize(native)
-        @native = native
-      end
-
-      # Factory method for creating test client from an app
-      def self.new(app_or_native, config: nil)
-        # If passed a native client directly, use it
-        return super(app_or_native) if app_or_native.is_a?(Spikard::Native::TestClient)
-
-        # Otherwise, create test client from app
-        Spikard::Testing.create_test_client(app_or_native, config: config)
-      end
-
-      def request(method, path, **options)
-        payload = @native.request(method.to_s.upcase, path, options)
-        Response.new(payload)
-      end
-
-      def websocket(path)
-        native_ws = @native.websocket(path)
-        WebSocketTestConnection.new(native_ws)
-      end
-
-      def sse(path)
-        native_sse = @native.sse(path)
-        SseStream.new(native_sse)
-      end
-
-      def close
-        @native.close
-      end
-
-      %w[get post put patch delete head options trace].each do |verb|
-        define_method(verb) do |path, **options|
-          request(verb.upcase, path, **options)
-        end
-      end
-    end
-
-    # WebSocket test connection wrapper
-    class WebSocketTestConnection
-      def initialize(native_ws)
-        @native_ws = native_ws
-      end
-
-      def send_text(text)
-        @native_ws.send_text(JSON.generate(text))
-      end
-
-      def send_json(obj)
-        @native_ws.send_json(obj)
-      end
-
-      def receive_text
-        raw = @native_ws.receive_text
-        JSON.parse(raw)
-      rescue JSON::ParserError
-        raw
-      end
-
-      def receive_json
-        @native_ws.receive_json
-      end
-
-      def receive_bytes
-        receive_text
-      end
-
-      def receive_message
-        native_msg = @native_ws.receive_message
-        WebSocketMessage.new(native_msg)
-      end
-
-      def close
-        @native_ws.close
-      end
-    end
-
-    # WebSocket message wrapper
-    class WebSocketMessage
-      def initialize(native_msg)
-        @native_msg = native_msg
-      end
-
-      def as_text
-        raw = @native_msg.as_text
-        return unless raw
-
-        JSON.parse(raw)
-      rescue JSON::ParserError
-        raw
-      end
-
-      def as_json
-        @native_msg.as_json
-      end
-
-      def as_binary
-        @native_msg.as_binary
-      end
-
-      def close?
-        @native_msg.is_close
-      end
-    end
-
-    # SSE stream wrapper
-    class SseStream
-      def initialize(native_sse)
-        @native_sse = native_sse
-      end
-
-      def body
-        @native_sse.body
-      end
-
-      def events
-        parsed_chunks.map { |chunk| InlineSseEvent.new(chunk) }
-      end
-
-      def events_as_json
-        parsed_chunks.filter_map do |chunk|
-          JSON.parse(chunk)
-        rescue JSON::ParserError
-          nil
-        end
-      end
-
-      private
-
-      # rubocop:disable Metrics/AbcSize, Metrics/MethodLength
-      def parsed_chunks
-        raw = body.to_s.gsub("\r\n", "\n")
-        events = []
-        current = []
-
-        raw.each_line do |line|
-          stripped = line.chomp
-          if stripped.start_with?('data:')
-            current << stripped[5..].strip
-          elsif stripped.empty?
-            unless current.empty?
-              data = current.join("\n").strip
-              events << data unless data.empty?
-              current = []
-            end
-          end
-        end
-
-        unless current.empty?
-          data = current.join("\n").strip
-          events << data unless data.empty?
-        end
-
-        events
-      end
-      # rubocop:enable Metrics/AbcSize, Metrics/MethodLength
-    end
-
-    # SSE event wrapper
-    class SseEvent
-      def initialize(native_event)
-        @native_event = native_event
-      end
-
-      def data
-        @native_event.data
-      end
-
-      def as_json
-        @native_event.as_json
-      end
-    end
-
-    # Lightweight wrapper for parsed SSE events backed by strings.
-    class InlineSseEvent
-      attr_reader :data
-
-      def initialize(data)
-        @data = data
-      end
-
-      def as_json
-        JSON.parse(@data)
-      end
-    end
-  end
-end
+# frozen_string_literal: true
+
+require 'json'
+
+module Spikard
+  # Testing helpers that wrap the native Ruby extension.
+  module Testing
+    module_function
+
+    def create_test_client(app, config: nil)
+      unless defined?(Spikard::Native::TestClient)
+        raise LoadError, 'Spikard native test client is not available. Build the native extension before running tests.'
+      end
+
+      # Allow generated apps to stash a test config
+      if config.nil? && app.instance_variable_defined?(:@__spikard_test_config)
+        config = app.instance_variable_get(:@__spikard_test_config)
+      end
+
+      # Use default config if none provided
+      config ||= Spikard::ServerConfig.new
+
+      routes_json = JSON.generate(app.route_metadata)
+      handlers = app.handler_map.transform_keys(&:to_sym)
+      ws_handlers = app.websocket_handlers || {}
+      sse_producers = app.sse_producers || {}
+      dependencies = app.dependencies || {}
+      native = Spikard::Native::TestClient.new(routes_json, handlers, config, ws_handlers, sse_producers, dependencies)
+      TestClient.new(native)
+    end
+
+    # High level wrapper around the native test client.
+    class TestClient
+      def initialize(native)
+        @native = native
+      end
+
+      # Factory method for creating test client from an app
+      def self.new(app_or_native, config: nil)
+        # If passed a native client directly, use it
+        return super(app_or_native) if app_or_native.is_a?(Spikard::Native::TestClient)
+
+        # Otherwise, create test client from app
+        Spikard::Testing.create_test_client(app_or_native, config: config)
+      end
+
+      def request(method, path, **options)
+        payload = @native.request(method.to_s.upcase, path, options)
+        Response.new(payload)
+      end
+
+      def websocket(path)
+        native_ws = @native.websocket(path)
+        WebSocketTestConnection.new(native_ws)
+      end
+
+      def sse(path)
+        native_sse = @native.sse(path)
+        SseStream.new(native_sse)
+      end
+
+      def close
+        @native.close
+      end
+
+      %w[get post put patch delete head options trace].each do |verb|
+        define_method(verb) do |path, **options|
+          request(verb.upcase, path, **options)
+        end
+      end
+    end
+
+    # WebSocket test connection wrapper
+    class WebSocketTestConnection
+      def initialize(native_ws)
+        @native_ws = native_ws
+      end
+
+      def send_text(text)
+        @native_ws.send_text(JSON.generate(text))
+      end
+
+      def send_json(obj)
+        @native_ws.send_json(obj)
+      end
+
+      def receive_text
+        raw = @native_ws.receive_text
+        JSON.parse(raw)
+      rescue JSON::ParserError
+        raw
+      end
+
+      def receive_json
+        @native_ws.receive_json
+      end
+
+      def receive_bytes
+        receive_text
+      end
+
+      def receive_message
+        native_msg = @native_ws.receive_message
+        WebSocketMessage.new(native_msg)
+      end
+
+      def close
+        @native_ws.close
+      end
+    end
+
+    # WebSocket message wrapper
+    class WebSocketMessage
+      def initialize(native_msg)
+        @native_msg = native_msg
+      end
+
+      def as_text
+        raw = @native_msg.as_text
+        return unless raw
+
+        JSON.parse(raw)
+      rescue JSON::ParserError
+        raw
+      end
+
+      def as_json
+        @native_msg.as_json
+      end
+
+      def as_binary
+        @native_msg.as_binary
+      end
+
+      def close?
+        @native_msg.is_close
+      end
+    end
+
+    # SSE stream wrapper
+    class SseStream
+      def initialize(native_sse)
+        @native_sse = native_sse
+      end
+
+      def body
+        @native_sse.body
+      end
+
+      def events
+        parsed_chunks.map { |chunk| InlineSseEvent.new(chunk) }
+      end
+
+      def events_as_json
+        parsed_chunks.filter_map do |chunk|
+          JSON.parse(chunk)
+        rescue JSON::ParserError
+          nil
+        end
+      end
+
+      private
+
+      # rubocop:disable Metrics/AbcSize, Metrics/MethodLength
+      def parsed_chunks
+        raw = body.to_s.gsub("\r\n", "\n")
+        events = []
+        current = []
+
+        raw.each_line do |line|
+          stripped = line.chomp
+          if stripped.start_with?('data:')
+            current << stripped[5..].strip
+          elsif stripped.empty?
+            unless current.empty?
+              data = current.join("\n").strip
+              events << data unless data.empty?
+              current = []
+            end
+          end
+        end
+
+        unless current.empty?
+          data = current.join("\n").strip
+          events << data unless data.empty?
+        end
+
+        events
+      end
+      # rubocop:enable Metrics/AbcSize, Metrics/MethodLength
+    end
+
+    # SSE event wrapper
+    class SseEvent
+      def initialize(native_event)
+        @native_event = native_event
+      end
+
+      def data
+        @native_event.data
+      end
+
+      def as_json
+        @native_event.as_json
+      end
+    end
+
+    # Lightweight wrapper for parsed SSE events backed by strings.
+    class InlineSseEvent
+      attr_reader :data
+
+      def initialize(data)
+        @data = data
+      end
+
+      def as_json
+        JSON.parse(@data)
+      end
+    end
+  end
+end