isomorfeus-puppetmaster 0.1.0

data/lib/isomorfeus/puppetmaster/rspec/matchers/become_closed.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module Capybara
+  module RSpecMatchers
+    module Matchers
+      class BecomeClosed
+        def initialize(options)
+          @options = options
+        end
+
+        def matches?(window)
+          @window = window
+          @wait_time = Capybara::Queries::BaseQuery.wait(@options, window.session.config.default_max_wait_time)
+          timer = Capybara::Helpers.timer(expire_in: @wait_time)
+          while window.exists?
+            return false if timer.expired?
+
+            sleep 0.01
+          end
+          true
+        end
+
+        def failure_message
+          "expected #{@window.inspect} to become closed after #{@wait_time} seconds"
+        end
+
+        def failure_message_when_negated
+          "expected #{@window.inspect} not to become closed after #{@wait_time} seconds"
+        end
+      end
+    end
+  end
+end

data/lib/isomorfeus/puppetmaster/rspec/matchers/compound.rb

@@ -0,0 +1,88 @@
+# frozen_string_literal: true
+
+if defined?(::RSpec::Expectations::Version)
+  module Capybara
+    module RSpecMatchers
+      module Matchers
+        module Compound
+          include ::RSpec::Matchers::Composable
+
+          def and(matcher)
+            And.new(self, matcher)
+          end
+
+          def and_then(matcher)
+            ::RSpec::Matchers::BuiltIn::Compound::And.new(self, matcher)
+          end
+
+          def or(matcher)
+            Or.new(self, matcher)
+          end
+
+          class CapybaraEvaluator
+            def initialize(actual)
+              @actual = actual
+              @match_results = Hash.new { |hsh, matcher| hsh[matcher] = matcher.matches?(@actual) }
+            end
+
+            def matcher_matches?(matcher)
+              @match_results[matcher]
+            end
+
+            def reset
+              @match_results.clear
+            end
+          end
+
+          # @api private
+          module Synchronizer
+            def match(_expected, actual)
+              @evaluator = CapybaraEvaluator.new(actual)
+              syncer = sync_element(actual)
+              begin
+                syncer.synchronize do
+                  @evaluator.reset
+                  raise ::Capybara::ElementNotFound unless synchronized_match?
+
+                  true
+                end
+              rescue StandardError
+                false
+              end
+            end
+
+            def sync_element(el)
+              if el.respond_to? :synchronize
+                el
+              elsif el.respond_to? :current_scope
+                el.current_scope
+              else
+                Capybara.string(el)
+              end
+            end
+          end
+
+          class And < ::RSpec::Matchers::BuiltIn::Compound::And
+            include Synchronizer
+
+          private
+
+            def synchronized_match?
+              [matcher_1_matches?, matcher_2_matches?].all?
+            end
+          end
+
+          class Or < ::RSpec::Matchers::BuiltIn::Compound::Or
+            include Synchronizer
+
+          private
+
+            def synchronized_match?
+              [matcher_1_matches?, matcher_2_matches?].any?
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/isomorfeus/puppetmaster/rspec/matchers/have_current_path.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+require 'capybara/rspec/matchers/base'
+
+module Capybara
+  module RSpecMatchers
+    module Matchers
+      class HaveCurrentPath < WrappedElementMatcher
+        def element_matches?(el)
+          el.assert_current_path(*@args)
+        end
+
+        def element_does_not_match?(el)
+          el.assert_no_current_path(*@args)
+        end
+
+        def description
+          "have current path #{current_path.inspect}"
+        end
+
+      private
+
+        def current_path
+          @args.first
+        end
+      end
+    end
+  end
+end

data/lib/isomorfeus/puppetmaster/rspec/matchers/have_selector.rb

@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+
+require 'capybara/rspec/matchers/base'
+
+module Capybara
+  module RSpecMatchers
+    module Matchers
+      class HaveSelector < WrappedElementMatcher
+        def element_matches?(el)
+          el.assert_selector(*@args, &@filter_block)
+        end
+
+        def element_does_not_match?(el)
+          el.assert_no_selector(*@args, &@filter_block)
+        end
+
+        def description
+          "have #{query.description}"
+        end
+
+        def query
+          @query ||= Capybara::Queries::SelectorQuery.new(*session_query_args, &@filter_block)
+        end
+      end
+
+      class HaveAllSelectors < WrappedElementMatcher
+        def element_matches?(el)
+          el.assert_all_of_selectors(*@args, &@filter_block)
+        end
+
+        def does_not_match?(_actual)
+          raise ArgumentError, 'The have_all_selectors matcher does not support use with not_to/should_not'
+        end
+
+        def description
+          'have all selectors'
+        end
+      end
+
+      class HaveNoSelectors < WrappedElementMatcher
+        def element_matches?(el)
+          el.assert_none_of_selectors(*@args, &@filter_block)
+        end
+
+        def does_not_match?(_actual)
+          raise ArgumentError, 'The have_none_of_selectors matcher does not support use with not_to/should_not'
+        end
+
+        def description
+          'have no selectors'
+        end
+      end
+
+      class HaveAnySelectors < WrappedElementMatcher
+        def element_matches?(el)
+          el.assert_any_of_selectors(*@args, &@filter_block)
+        end
+
+        def does_not_match?(_actual)
+          el.assert_none_of_selectors(*@args, &@filter_block)
+        end
+
+        def description
+          'have any selectors'
+        end
+      end
+    end
+  end
+end

data/lib/isomorfeus/puppetmaster/rspec/matchers/have_text.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require 'capybara/rspec/matchers/base'
+
+module Capybara
+  module RSpecMatchers
+    module Matchers
+      class HaveText < WrappedElementMatcher
+        def element_matches?(el)
+          el.assert_text(*@args)
+        end
+
+        def element_does_not_match?(el)
+          el.assert_no_text(*@args)
+        end
+
+        def description
+          "text #{format(text)}"
+        end
+
+        def format(content)
+          content.inspect
+        end
+
+      private
+
+        def text
+          @args[0].is_a?(Symbol) ? @args[1] : @args[0]
+        end
+      end
+    end
+  end
+end

data/lib/isomorfeus/puppetmaster/rspec/matchers/have_title.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+require 'capybara/rspec/matchers/base'
+
+module Capybara
+  module RSpecMatchers
+    module Matchers
+      class HaveTitle < WrappedElementMatcher
+        def element_matches?(el)
+          el.assert_title(*@args)
+        end
+
+        def element_does_not_match?(el)
+          el.assert_no_title(*@args)
+        end
+
+        def description
+          "have title #{title.inspect}"
+        end
+
+      private
+
+        def title
+          @args.first
+        end
+      end
+    end
+  end
+end

data/lib/isomorfeus/puppetmaster/rspec/matchers/match_selector.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require 'capybara/rspec/matchers/have_selector'
+
+module Capybara
+  module RSpecMatchers
+    module Matchers
+      class MatchSelector < HaveSelector
+        def element_matches?(el)
+          el.assert_matches_selector(*@args, &@filter_block)
+        end
+
+        def element_does_not_match?(el)
+          el.assert_not_matches_selector(*@args, &@filter_block)
+        end
+
+        def description
+          "match #{query.description}"
+        end
+
+        def query
+          @query ||= Capybara::Queries::MatchQuery.new(*session_query_args, &@filter_block)
+        end
+      end
+    end
+  end
+end

data/lib/isomorfeus/puppetmaster/rspec/matchers/match_style.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+require 'capybara/rspec/matchers/base'
+
+module Capybara
+  module RSpecMatchers
+    module Matchers
+      class MatchStyle < WrappedElementMatcher
+        def element_matches?(el)
+          el.assert_matches_style(*@args)
+        end
+
+        def does_not_match?(_actual)
+          raise ArgumentError, 'The match_style matcher does not support use with not_to/should_not'
+        end
+
+        def description
+          'match style'
+        end
+      end
+    end
+  end
+end
+
+module Capybara
+  module RSpecMatchers
+    module Matchers
+      ##
+      # @deprecated
+      class HaveStyle < MatchStyle
+        def initialize(*args, &filter_block)
+          warn 'HaveStyle matcher is deprecated, please use the MatchStyle matcher instead'
+          super
+        end
+      end
+    end
+  end
+end

data/lib/isomorfeus/puppetmaster/self_forwardable.rb

@@ -0,0 +1,31 @@
+module Isomorfeus
+  module Puppetmaster
+    module SelfForwardable
+      def self.extended(base)
+          base.define_singleton_method(:document_forward) do |methods|
+            methods.each do |method|
+              define_method(method) do |*args, &block|
+                @driver.send("document_#{method}", self, *args, &block)
+              end
+            end
+          end
+
+          base.define_singleton_method(:frame_forward) do |methods|
+            methods.each do |method|
+              define_method(method) do |*args|
+                @driver.send("frame_#{method}", self, *args)
+              end
+            end
+          end
+
+          base.define_singleton_method(:node_forward) do |methods|
+            methods.each do |method|
+              define_method(method) do |*args|
+                @driver.send("node_#{method}", self, *args)
+              end
+            end
+          end
+      end
+    end
+  end
+end

data/lib/isomorfeus/puppetmaster/server.rb

@@ -0,0 +1,128 @@
+module Isomorfeus
+  module Puppetmaster
+    class Server
+      class Timer
+        def initialize(expire_in:)
+          @start = current
+          @expire_in = expire_in
+        end
+
+        def expired?
+          raise Isomorfeus::Puppetmaster::FrozenInTime, 'Time appears to be frozen. Puppetmaster does not work with libraries which freeze time, consider using time travelling instead' if stalled?
+
+          current - @start >= @expire_in
+        end
+
+        def stalled?
+          @start == current
+        end
+
+        private
+
+        if defined?(Process::CLOCK_MONOTONIC)
+          def current; Process.clock_gettime Process::CLOCK_MONOTONIC; end
+        else
+          def current; Time.now.to_f; end
+        end
+      end
+
+      class << self
+        def ports
+          @ports ||= {}
+        end
+      end
+
+      attr_reader :app, :port, :host
+
+      def initialize(app,
+                     port: Isomorfeus::Puppetmaster.server_port,
+                     host: Isomorfeus::Puppetmaster.server_host,
+                     extra_middleware: [])
+        @app = app
+        @extra_middleware = extra_middleware
+        @server_thread = nil # suppress warnings
+        @host = host
+        @port = port
+        @port ||= Isomorfeus::Puppetmaster::Server.ports[port_key]
+        @port ||= find_available_port(host)
+        @checker = Isomorfeus::Puppetmaster::Server::Checker.new(@host, @port)
+      end
+
+      def reset_error!
+        middleware.clear_error
+      end
+
+      def error
+        middleware.error
+      end
+
+      def using_ssl?
+        @checker.ssl?
+      end
+
+      def scheme
+        using_ssl? ? 'https' : 'http'
+      end
+
+      def responsive?
+        return false if @server_thread&.join(0)
+
+        res = @checker.request { |http| http.get('/__identify__') }
+
+        return res.body == app.object_id.to_s if res.is_a?(Net::HTTPSuccess) || res.is_a?(Net::HTTPRedirection)
+      rescue SystemCallError, Net::ReadTimeout, OpenSSL::SSL::SSLError
+        false
+      end
+
+      def wait_for_pending_requests
+        timer = Isomorfeus::Puppetmaster::Server::Timer.new(expire_in: 60)
+        while pending_requests?
+          raise 'Requests did not finish in 60 seconds' if timer.expired?
+
+          sleep 0.01
+        end
+      end
+
+      def boot
+        unless responsive?
+          Isomorfeus::Puppetmaster::Server.ports[port_key] = port
+
+          @server_thread = Thread.new do
+            Isomorfeus::Puppetmaster.server.call(middleware, port, host)
+          end
+
+          timer = Isomorfeus::Puppetmaster::Server::Timer.new(expire_in: 60)
+          until responsive?
+            raise 'Rack application timed out during boot' if timer.expired?
+
+            @server_thread.join(0.1)
+          end
+        end
+
+        self
+      end
+
+    private
+
+      def middleware
+        @middleware ||= ::Isomorfeus::Puppetmaster::Server::Middleware.new(app, @extra_middleware)
+      end
+
+      def port_key
+        app.object_id
+      end
+
+      def pending_requests?
+        middleware.pending_requests?
+      end
+
+      def find_available_port(host)
+        server = TCPServer.new(host, 0)
+        server.addr[1]
+      ensure
+        server&.close
+      end
+    end
+  end
+end
+