test_bench-isolated 0 → 2.0.0.1

data/lib/test_bench_isolated/test_bench/output/device/substitute.rb

@@ -0,0 +1,60 @@
+module TestBenchIsolated
+  module TestBench
+    module Output
+      module Device
+        module Substitute
+          def self.build
+            Device.new
+          end
+  
+          class Device
+            def written_data
+              @written_data ||= String.new
+            end
+            attr_writer :written_data
+  
+            def flushed_data
+              @flushed_data ||= String.new
+            end
+            attr_writer :flushed_data
+  
+            attr_accessor :tty
+            def tty? = !!tty
+  
+            def write(data)
+              bytes_written = data.bytesize
+  
+              written_data << data
+  
+              bytes_written
+            end
+  
+            def flush
+              flushed_data << written_data
+            end
+  
+            def written?(data=nil)
+              if data.nil?
+                !written_data.empty?
+              else
+                written_data == data
+              end
+            end
+  
+            def flushed?(data=nil)
+              if data.nil?
+                !flushed_data.empty?
+              else
+                flushed_data == data
+              end
+            end
+  
+            def tty!
+              self.tty = true
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/test_bench_isolated/test_bench/output/digest.rb

@@ -0,0 +1,115 @@
+module TestBenchIsolated
+  module TestBench
+    module Output
+      class Digest
+        def last_digest
+          @last_digest ||= 0
+        end
+        attr_writer :last_digest
+  
+        def buffer
+          @buffer ||= String.new(encoding: 'BINARY', capacity: Defaults.size_bytes * 2)
+        end
+        attr_writer :buffer
+  
+        def self.digest(data)
+          instance = new
+          instance.update(data)
+          instance.digest
+        end
+  
+        def self.configure(receiver, inert: nil, attr_name: nil)
+          attr_name ||= :digest
+          inert = true if inert.nil?
+  
+          if inert
+            instance = Null.new
+          else
+            instance = self.new
+          end
+  
+          receiver.public_send(:"#{attr_name}=", instance)
+        end
+  
+        def clone
+          cloned_digest = Digest.new
+          cloned_digest.last_digest = last_digest
+          cloned_digest.buffer << buffer
+          cloned_digest
+        end
+  
+        def update(data)
+          digest_size_bytes = Defaults.size_bytes
+  
+          (0..data.bytesize).step(digest_size_bytes).each do |position|
+            next_bytes = data.byteslice(position, digest_size_bytes)
+            next_bytes.force_encoding('BINARY')
+  
+            buffer << next_bytes
+  
+            if buffer.length >= digest_size_bytes
+              self.last_digest = digest
+  
+              buffer.slice!(0, 8)
+            end
+          end
+  
+          digest
+        end
+  
+        def digest
+          buffer_bytes = buffer.unpack('C*')[0...8]
+  
+          buffer_int64 = buffer_bytes.reduce(0) do |int64, byte|
+            (int64 << 8) | byte
+          end
+  
+          (last_digest + buffer_int64) % (256 ** Defaults.size_bytes - 1)
+        end
+        alias :value :digest
+        alias :to_i :digest
+  
+        def digest?(data)
+          other_digest = self.class.digest(data)
+  
+          digest == other_digest
+        end
+  
+        def format
+          self.class.format(value)
+        end
+        alias :to_s :format
+  
+        def self.format(digest)
+          if digest < 0
+            # 0x8000...
+            sign_mask = 0x1 << (8 * Defaults.size_bytes - 1)
+            # 0x7FFF...
+            mask = (256 ** Defaults.size_bytes - 1) ^ sign_mask
+  
+            digest_uint64 = ~digest & mask | sign_mask
+          else
+            digest_uint64 = digest
+          end
+  
+          "0x%016X" % digest_uint64
+        end
+  
+        class Null
+          def update(_data)
+          end
+  
+          def digest?(_data)
+            false
+          end
+        end
+  
+        module Defaults
+          def self.size_bytes
+            8
+          end
+        end
+      end
+    end
+  end
+end

data/lib/test_bench_isolated/test_bench/output/output.rb

@@ -0,0 +1,68 @@
+module TestBenchIsolated
+  module TestBench
+    module Output
+      def self.included(cls)
+        cls.class_exec do
+          include Telemetry::Sink::Handler
+  
+          extend Build
+          extend RegisterTelemetry
+          extend Configure
+        end
+      end
+  
+      def writer
+        @writer ||= Writer::Substitute.build
+      end
+      attr_writer :writer
+  
+      def configure(writer: nil, device: nil, styling: nil)
+        Writer.configure(self, writer:, device:, styling:)
+      end
+  
+      module Build
+        def build(writer: nil, device: nil, styling: nil, **arguments)
+          instance = new
+          instance.configure(writer:, device:, styling:, **arguments)
+          instance
+        end
+      end
+  
+      module RegisterTelemetry
+        def register_telemetry(telemetry, **arguments)
+          instance = build(**arguments)
+          telemetry.register(instance)
+          instance
+        end
+        alias :register :register_telemetry
+      end
+  
+      module Configure
+        def configure(receiver, attr_name: nil, **arguments)
+          attr_name ||= :output
+  
+          instance = build(**arguments)
+          receiver.public_send(:"#{attr_name}=", instance)
+        end
+      end
+  
+      module Substitute
+        def self.build
+          Output.new
+        end
+  
+        class Output < Telemetry::Substitute::Sink
+          def handle(event_or_event_data)
+            if event_or_event_data.is_a?(Telemetry::Event)
+              event_data = Telemetry::Event::Export.(event_or_event_data)
+            else
+              event_data = event_or_event_data
+            end
+  
+            receive(event_data)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/test_bench_isolated/test_bench/output/writer/buffer.rb

@@ -0,0 +1,59 @@
+module TestBenchIsolated
+  module TestBench
+    module Output
+      class Writer
+        class Buffer
+          attr_accessor :limit
+  
+          def contents
+            @contents ||= String.new
+          end
+          attr_writer :contents
+  
+          def self.build
+            instance = new
+            instance.limit = 0
+            instance
+          end
+  
+          def self.configure(receiver, attr_name: nil)
+            attr_name ||= :buffer
+  
+            instance = build
+            receiver.public_send(:"#{attr_name}=", instance)
+          end
+  
+          def receive(data)
+            bytes = data.bytesize
+  
+            if not limit.nil?
+              final_size = contents.bytesize + bytes
+  
+              if final_size > limit
+                bytes = limit - contents.bytesize
+  
+                data = data[0...bytes]
+              end
+            end
+  
+            contents << data
+  
+            bytes
+          end
+  
+          def limit?
+            !limit.nil?
+          end
+  
+          def flush(*devices)
+            devices.each do |device|
+              device.write(contents)
+            end
+  
+            contents.clear
+          end
+        end
+      end
+    end
+  end
+end

data/lib/test_bench_isolated/test_bench/output/writer/defaults.rb

@@ -0,0 +1,13 @@
+module TestBenchIsolated
+  module TestBench
+    module Output
+      class Writer
+        module Defaults
+          def self.device
+            STDOUT
+          end
+        end
+      end
+    end
+  end
+end

data/lib/test_bench_isolated/test_bench/output/writer/style.rb

@@ -0,0 +1,52 @@
+module TestBenchIsolated
+  module TestBench
+    module Output
+      class Writer
+        module Style
+          Error = Class.new(RuntimeError)
+  
+          def self.control_code(style)
+            control_codes.fetch(style) do
+              raise Error, "Invalid style #{style.inspect}"
+            end
+          end
+  
+          def self.control_codes
+            @sgr_codes ||= {
+              :reset => '0',
+  
+              :bold => '1',
+              :faint => '2',
+              :italic => '3',
+              :underline => '4',
+  
+              :reset_intensity => '22',
+              :reset_italic => '23',
+              :reset_underline => '24',
+  
+              :black => '30',
+              :red => '31',
+              :green => '32',
+              :yellow => '33',
+              :blue => '34',
+              :magenta => '35',
+              :cyan => '36',
+              :white => '37',
+              :reset_fg => '39',
+  
+              :black_bg => '40',
+              :red_bg => '41',
+              :green_bg => '42',
+              :yellow_bg => '43',
+              :blue_bg => '44',
+              :magenta_bg => '45',
+              :cyan_bg => '46',
+              :white_bg => '47',
+              :reset_bg => '49'
+            }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/test_bench_isolated/test_bench/output/writer/substitute.rb

@@ -0,0 +1,40 @@
+module TestBenchIsolated
+  module TestBench
+    module Output
+      class Writer
+        module Substitute
+          def self.build
+            Writer.build
+          end
+  
+          def self.included(cls)
+            cls.class_exec do
+              extend Build
+            end
+          end
+  
+          def written_data
+            device.written_data
+          end
+          alias :written_text :written_data
+  
+          def styling!
+            self.styling_policy = Styling.on
+          end
+  
+          module Build
+            def build
+              instance = new
+              instance.buffer.limit = 0
+              instance
+            end
+          end
+  
+          class Writer < Writer
+            include Substitute
+          end
+        end
+      end
+    end
+  end
+end

data/lib/test_bench_isolated/test_bench/output/writer.rb

@@ -0,0 +1,212 @@
+module TestBenchIsolated
+  module TestBench
+    module Output
+      class Writer
+        def device
+          @device ||= Device::Substitute.build
+        end
+        attr_writer :device
+  
+        def styling_policy
+          @styling_policy ||= Styling.default
+        end
+        alias :styling :styling_policy
+        attr_writer :styling_policy
+  
+        def digest
+          @digest ||= Digest.new
+        end
+        attr_writer :digest
+  
+        def sequence
+          @sequence ||= 0
+        end
+        attr_writer :sequence
+  
+        def column_sequence
+          @column_sequence ||= 0
+        end
+        attr_writer :column_sequence
+  
+        def buffer
+          @buffer ||= Buffer.new
+        end
+        attr_writer :buffer
+  
+        def configure
+          Buffer.configure(self)
+        end
+  
+        def self.build(device=nil, styling: nil, inert_digest: nil)
+          device ||= Defaults.device
+  
+          instance = new
+          instance.device = device
+          instance.styling_policy = styling
+  
+          Digest.configure(instance, inert: inert_digest)
+  
+          instance.configure
+  
+          instance
+        end
+  
+        def self.configure(receiver, writer: nil, styling: nil, inert_digest: nil, device: nil, attr_name: nil)
+          attr_name ||= :writer
+  
+          if not writer.nil?
+            instance = writer
+          else
+            instance = build(device, styling:, inert_digest:)
+          end
+  
+          receiver.public_send(:"#{attr_name}=", instance)
+        end
+  
+        def sync
+          @sync.nil? ? @sync = true : @sync
+        end
+  
+        def tty
+          @tty.nil? ? @tty = device_tty? : @tty
+        end
+        attr_writer :tty
+        alias :tty? :tty
+  
+        def puts(text=nil)
+          if not text.nil?
+            text = text.chomp
+  
+            print(text)
+          end
+  
+          style(:reset)
+  
+          if tty?
+            write("\e[0K")
+          end
+  
+          write("\n")
+  
+          self.column_sequence = 0
+        end
+  
+        def style(style, *additional_styles)
+          control_code = Style.control_code(style)
+          control_codes = [control_code]
+  
+          additional_styles.each do |style|
+            control_code = Style.control_code(style)
+            control_codes << control_code
+          end
+  
+          if styling?
+            write("\e[#{control_codes.join(';')}m")
+          end
+  
+          self
+        end
+  
+        def print(text)
+          self.column_sequence += text.length
+  
+          write(text)
+  
+          self
+        end
+  
+        def write(data)
+          if sync
+            bytes_written = write!(data)
+          else
+            bytes_written = buffer.receive(data)
+          end
+  
+          self.sequence += bytes_written
+  
+          data = data[0...bytes_written]
+          digest.update(data)
+  
+          bytes_written
+        end
+  
+        def write!(data)
+          device.write(data)
+  
+          data.bytesize
+        end
+  
+        def device_tty?
+          device.tty?
+        end
+  
+        def flush
+          buffer.flush(device)
+        end
+  
+        def sync=(sync)
+          @sync = sync
+  
+          if sync
+            flush
+          end
+        end
+  
+        def written?(data=nil)
+          if data.nil?
+            sequence > 0
+          else
+            digest.digest?(data)
+          end
+        end
+  
+        def current?(sequence)
+          sequence >= self.sequence
+        end
+  
+        def styling?
+          Styling.styling?(styling_policy, tty?)
+        end
+  
+        module Styling
+          Error = Class.new(RuntimeError)
+  
+          def self.styling?(policy, console)
+            assure_styling(policy, console)
+          end
+  
+          def self.assure_styling(policy, console=nil)
+            console ||= false
+  
+            case policy
+            when on
+              true
+            when off
+              false
+            when detect
+              console ? true : false
+            else
+              raise Error, "Unknown styling policy #{policy.inspect}"
+            end
+          end
+  
+          def self.on = :on
+          def self.off = :off
+          def self.detect = :detect
+  
+          def self.default
+            policy = ENV.fetch('TEST_BENCH_OUTPUT_STYLING') do
+              return default!
+            end
+  
+            policy.to_sym
+          end
+  
+          def self.default!
+            :detect
+          end
+        end
+      end
+    end
+  end
+end

data/lib/test_bench_isolated/test_bench/output.rb

@@ -0,0 +1,14 @@
+require 'test_bench_isolated/test_bench/telemetry'
+
+require 'test_bench_isolated/test_bench/output/digest'
+
+require 'test_bench_isolated/test_bench/output/device/null'
+require 'test_bench_isolated/test_bench/output/device/substitute'
+
+require 'test_bench_isolated/test_bench/output/writer/buffer'
+require 'test_bench_isolated/test_bench/output/writer'
+require 'test_bench_isolated/test_bench/output/writer/style'
+require 'test_bench_isolated/test_bench/output/writer/substitute'
+require 'test_bench_isolated/test_bench/output/writer/defaults'
+
+require 'test_bench_isolated/test_bench/output/output'

data/lib/test_bench_isolated/test_bench/random/controls/seed.rb

@@ -0,0 +1,27 @@
+module TestBenchIsolated
+  module TestBench
+    class Random
+      module Controls
+        module Seed
+          def self.example
+            0x1111_1111_1111_1111
+          end
+  
+          def self.text
+            example.to_s(36)
+          end
+  
+          module Namespace
+            def self.example
+              'some-seed-namespace'
+            end
+  
+            def self.other_example
+              'some-other-seed-namespace'
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/test_bench_isolated/test_bench/random/controls.rb

@@ -0,0 +1 @@
+require 'test_bench_isolated/test_bench/random/controls/seed'