lavin 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ed47d43b5e5f91e1af2c3e319fa03d3588aee9826215394038359e559773a173
-  data.tar.gz: 17261c42f46a7c4e1ecff28adbd92434c5ec66ac28d4c208d6195a55bf49f69d
+  metadata.gz: d3a7216c049391b24ccad2eb6c6d6424114761a11b1b4aa15824c5c58ca2bf19
+  data.tar.gz: 3ec27a41fdc255d3eafb3a168c39ad8b744b28ab5e436d38d8eaf9f94aa22f23
 SHA512:
-  metadata.gz: 747a493c85dd8fe99c5da92d9fc8380248f80a7b0483bfe42ba3f48fc8a06c89ecc45dfa02f424304b75d28709dfd7c35bc9137a658b22cb449dab91a4bb13ff
-  data.tar.gz: 304de7aa069cb83a61f3530bec34df37349e4572cdc4eb23ec20b86daa4a75606f207248f7b4efe5f3338d8f4166ba94c71ee33e856bc22e6cb36720585627cf
+  metadata.gz: c6689ccce7532d8438013a8e3af32052f5b8df828169a026270f03d16a9078f11677b36f2a6c8989ec27cd8bfa3248d4bb1a9bbd2ddc35cccd4d8f1df143e182
+  data.tar.gz: 4b2b728d3d5c775471c3ffdcf52c7b2a475d6c7e76d6aae4288b1e1154747e6f81bc802bf3f04f951aa1dec02de690671f110e01db8baed5c5442717f93f2b85

data/lib/lavin/client.rb

@@ -17,7 +17,7 @@ module Lavin
     }.freeze
 
     attr_reader :internet, :base_url
-    attr_accessor :request_count, :cookie
+    attr_accessor :request_count, :cookie, :report_statistics
 
     def initialize(base_url = nil)
       raise NoCurrentAsyncTaskError unless Async::Task.current?
@@ -26,6 +26,7 @@ module Lavin
       @base_url = base_url
       @request_count = 0
       @cookie = nil
+      @report_statistics = true
     end
 
     def close
@@ -41,7 +42,7 @@ module Lavin
 
       status, headers, body = process(response)
 
-      Statistics.register_request(method:, url:, status:, duration:)
+      Statistics.register_request(method:, url:, status:, duration:) if report_statistics
 
       {status:, headers:, body: body}
     end

data/lib/lavin/hook.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+#
+
+module Lavin
+  class Hook
+    attr_reader :user, :block
+
+    def initialize(user:, &block)
+      @user = user
+      @block = block
+    end
+
+    def run(context: nil)
+      call(context:)
+      Runner.yield
+    end
+
+    def call(context:)
+      report_statistics = context.client.report_statistics
+      context.client.report_statistics = false
+      context.instance_exec(&block)
+    rescue => error
+      puts "Caught an error - #{error.class}: #{error.message}"
+      puts error.backtrace
+    ensure
+      context.client.report_statistics = report_statistics
+    end
+  end
+end

data/lib/lavin/http_client.rb

@@ -9,7 +9,7 @@ module Lavin
 
     def initialize(**kwargs)
       super(**kwargs)
-      @client = Client.new(config[:base_url])
+      @client = kwargs.fetch(:client) { Client.new(config[:base_url]) }
     end
 
     def cleanup

data/lib/lavin/runner.rb

@@ -57,7 +57,7 @@ module Lavin
       rescue StandardError => error
         puts "Failed to run in thread: #{error.message}"
         thread.join
-        thread = nil
+        self.thread = nil
         raise
       end
 

data/lib/lavin/statistics.rb

@@ -28,6 +28,10 @@ module Lavin
         data[:total_requests]
       end
 
+      def total_steps
+        data[:step_summary][:count]
+      end
+
       def duration
         return data[:duration] if data[:duration]
 
@@ -42,8 +46,13 @@ module Lavin
         data[:requests][key] << result
       end
 
-      def register_step
-        # TODO
+      def register_step(user:, step_name:, failure: nil)
+        data[:step_summary][:count] += 1
+        data[:step_summary][:success] += 1 unless failure
+        data[:step_summary][:failure] += 1 if failure
+        key = [user, step_name]
+        data[:steps][key][:success] += 1 unless failure
+        data[:steps][key][:failure] += 1 if failure
       end
 
       def stats
@@ -68,7 +77,7 @@ module Lavin
             statuses: statuses.tally,
             avg_duration: avg_duration,
             min_duration: min_duration,
-            max_duration: max_duration,
+            max_duration: max_duration
           }
         end
 
@@ -76,6 +85,8 @@ module Lavin
           duration: duration,
           total_requests: total_requests,
           rate: duration ? format("%.2f", total_requests / duration) : 0,
+          step_summary: data[:step_summary],
+          steps: data[:steps],
           requests: requests
         ).tap do |stats|
           # FIXME remove!
@@ -88,6 +99,14 @@ module Lavin
 
         show_summary(values)
 
+        show_steps(values) do |(user, step), hash|
+          format(
+            "%-24<user_step>s %8<success>d %8<failure>d",
+            user_step: "#{user}.#{step}",
+            **hash
+          )
+        end
+
         show_table(values) do |request_values|
           format(
             "%-6<method>s %-100<url>s %6<requests>d %12<avg_duration>fs %12<min_duration>fs %12<max_duration>fs",
@@ -116,7 +135,12 @@ module Lavin
           start: nil,
           duration: nil,
           total_requests: 0,
-          steps: [],
+          step_summary: {
+            count: 0,
+            success: 0,
+            failure: 0
+          },
+          steps: Hash.new { |h, k| h[k] = {success: 0, failure: 0} },
           requests: Hash.new { |h, k| h[k] = [] }
         }
       end
@@ -127,11 +151,27 @@ module Lavin
           Lavin results:
           Test ran for #{values.duration}s
           Total number of requests: #{values.total_requests}
-          Rate: #{format("%.2f", values.total_requests/values.duration)} rps
+          Rate: #{format("%.2f", values.total_requests / values.duration)} rps
+
+          Total number of steps: #{values.total_steps}
+          Step success rate: #{format("%.2f %%", 100 * values.successful_steps.to_f / values.total_steps)}
 
         RESULT
       end
 
+      def show_steps(values)
+        puts format(
+          "%-24<user_step>s %8<success>s %8<failure>s",
+          user_step: "Steps",
+          success: "Success",
+          failure: "Failure"
+        )
+        divider = "-" * 42
+        puts divider
+        values.each_step { |step_values| puts yield step_values }
+        puts "#{divider}\n\n"
+      end
+
       def show_table(values)
         puts format(
           "%-6<method>s %-100<url>s %-6<requests>s %12<avg_duration>s %12<min_duration>s %12<max_duration>s",

data/lib/lavin/stats.rb

@@ -2,13 +2,15 @@
 
 module Lavin
   class Stats
-    attr_reader :duration, :total_requests, :rate, :requests
+    attr_reader :duration, :total_requests, :rate, :requests, :step_summary, :steps
 
-    def initialize(duration:, total_requests:, rate:, requests: [])
+    def initialize(duration:, total_requests:, rate:, requests: [], step_summary: {}, steps: [])
       @duration = duration
       @total_requests = total_requests
       @rate = rate
       @requests = requests
+      @step_summary = step_summary
+      @steps = steps
     end
 
     def empty?
@@ -20,10 +22,28 @@ module Lavin
         duration:,
         total_requests:,
         rate:,
-        requests:
+        requests:,
+        step_summary:,
+        steps:
       }
     end
 
+    def total_steps
+      @step_summary[:count]
+    end
+
+    def successful_steps
+      @step_summary[:success]
+    end
+
+    def failed_steps
+      @step_summary[:failure]
+    end
+
+    def each_step(&block)
+      steps.each(&block)
+    end
+
     def each_request(&block)
       requests.each(&block)
     end

data/lib/lavin/step.rb

@@ -1,10 +1,13 @@
 # frozen_string_literal: true
+#
 
 module Lavin
   class Step
-    attr_reader :user, :block, :repeat
+    attr_reader :name, :user, :block, :repeat
 
-    def initialize(repeat: 1, &block)
+    def initialize(name:, user:, repeat: 1, &block)
+      @name = name
+      @user = user
       @repeat = repeat
       @block = block
     end
@@ -18,11 +21,11 @@ module Lavin
 
     def call(context:)
       context.instance_exec(&block)
-      # Report Success!
+      Statistics.register_step(user: user.name, step_name: name)
     rescue => error
       puts "Caught an error - #{error.class}: #{error.message}"
       puts error.backtrace
-      # Report Failure!
+      Statistics.register_step(user: user.name, step_name: name, failure: error.message)
     end
   end
 end

data/lib/lavin/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Lavin
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/lavin/worker.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'lavin/step'
+require 'lavin/hook'
 
 module Lavin
   module Worker
@@ -8,21 +9,22 @@ module Lavin
       def before(&block)
         return @before unless block
 
-        @before = block
+        @before = Hook.new(user: self, &block)
       end
 
       def after(&block)
         return @after unless block
 
-        @after = block
+        @after = Hook.new(user: self, &block)
       end
 
       def steps
         @steps ||= []
       end
 
-      def step(**options, &block)
-        steps << Step.new(**options, &block)
+      def step(name: nil, **options, &block)
+        name ||= "Step##{steps.size + 1}"
+        steps << Step.new(user: self, name:, **options, &block)
       end
     end
 
@@ -38,11 +40,11 @@ module Lavin
     end
 
     def run
-      self.class.before.call.then { Runner.yield } if self.class.before
+      self.class.before.run(context: self).then { Runner.yield } if self.class.before
 
       run_step until finished?
 
-      self.class.after.call.then { Runner.yield } if self.class.after
+      self.class.after.run(context: self).then { Runner.yield } if self.class.after
     end
 
     private

data.tar.gz.sig

@@ -1 +1 @@
-w���r�We��sk�8ƚ�J�A��sl���k*�z
4��\��w�x
+�r�)ꮠo���%e�[n�a��/&�m�˧*�#
�RG�1���2�[L{@u��E��/ıi�����k�D�3{�<f�
�o�jNo�*/�C��Q����	��Һ���y�'��;�UC��Ф�)���4}q�M�����G��G~��R<�`)�+����n��{�=U~��r̒ ���MW��l
�.Y=��Q�i��5*S���R��bwC��!��C�J����~0D?�B쬗��r����

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lavin
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Sammy Henningsson
@@ -29,7 +29,7 @@ cert_chain:
   DVzXaUnsmwP+jQ1PkDa5q8ibBzMd2c6Hmm87UDqPxZtML0bF9SjrpbyLMjwtXaMA
   WDPp0ajpdUZ9GPHsrVNYXiOfQIqcmlmpYVsH1o7vuneUIcIDMrnMDChh
   -----END CERTIFICATE-----
-date: 2022-10-30 00:00:00.000000000 Z
+date: 2022-10-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: async
@@ -113,6 +113,7 @@ files:
 - lib/lavin.rb
 - lib/lavin/client.rb
 - lib/lavin/error.rb
+- lib/lavin/hook.rb
 - lib/lavin/http_client.rb
 - lib/lavin/runner.rb
 - lib/lavin/statistics.rb