ruby-sfn-local 0.1.13 → 0.1.17

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 75dd7b0209a27d5fedaac3a899ab43ce8c9cc814ae8736e9ebfab57e11a81397
-  data.tar.gz: a4caabe1ecab7fbae2b9a71d3ddd760f95c709813d67c67972248cd15d7628b8
+  metadata.gz: 4a8d8a6fc30d2a2965dfda5c56c1aad5af809b7599f829828c829cf9930efbaf
+  data.tar.gz: ac740780539e3c0f93a7f3613f5b332f727d063bbbc63476b50c19c536489418
 SHA512:
-  metadata.gz: 46d027dcc3ba4688427c36f35ffd40c725a6edef9727320aea01c7c6c100f8dc2933dbe01437104d71546b037ea7830458dccb3e14b1be7cb443c47628077298
-  data.tar.gz: 77cc4a235b5e881fac19faff75c9a36c2d57ef08b95a7c941d214bb823166db9149f9e53241a51436ec94c0d8e85965564dbe9caa0b40c3f2caae0fbe2a8fe02
+  metadata.gz: f0f125829bc0d612a8dfd84f06434434ff684e61cdc51328af8c2ffe0df809cfbf00141d5d436b6a091ecb1413a3b1c58da097eae2b0fbd0fddab7c7e2be7b78
+  data.tar.gz: 1b1de6bb7d076d4930594eb5d153c1e243067d5c48f17376a5e500b3d9988679009caa24d34174f3dd3d9c6104b14fa66ddd93ce016b6dda9dcac835c6b3ff46

data/lib/sfn/execution.rb

@@ -4,8 +4,8 @@ module Sfn
   class Execution
     attr_accessor :uuid, :state_machine, :test_case, :arn, :output, :profile
 
-    def self.call(state_machine, test_case, mock_data, input)
-      new(state_machine, test_case).exec(mock_data, input)
+    def self.call(state_machine, test_case, mock_data, input, dry_run = false)
+      new(state_machine, test_case).exec(mock_data, input, dry_run)
     end
 
     def initialize(state_machine, test_case)
@@ -14,14 +14,15 @@ module Sfn
       self.test_case = test_case.camelize
     end
 
-    def exec(mock_data, input)
+    def exec(mock_data, input, dry_run = false)
       MockData.write_context(state_machine.name, test_case, mock_data)
       self.arn = AwsCli.run('stepfunctions', 'start-execution',
                             { name: uuid,
                               'state-machine': "#{state_machine.arn}##{test_case}",
-                              input: "'#{input.to_json}'" },
+                              input: "'#{input.to_json}'" 
+                            },
                             'executionArn')
-      self.output, self.profile = ExecutionLog.parse(arn)
+      self.output, self.profile = ExecutionLog.parse(arn, dry_run)
       self
     end
   end

data/lib/sfn/execution_log.rb

@@ -17,7 +17,7 @@ module Sfn
     EVENTS = %w[stateEnteredEventDetails stateExitedEventDetails executionSucceededEventDetails
                 executionFailedEventDetails taskScheduledEventDetails].freeze
 
-    def self.parse(execution_arn)
+    def self.parse(execution_arn, dry_run = false)
       profile = {}
       root_state_name = {}
       output = nil
@@ -28,7 +28,7 @@ module Sfn
       JSON.parse(events_json).each do |event|
         parsed_event = new(event)
         output ||= parsed_event.output
-        error  ||= parsed_event.error(events_json)
+        error  ||= parsed_event.error(events_json, dry_run)
         state_name = parsed_event.state_name
 
         unless state_name.nil?
@@ -57,12 +57,14 @@ module Sfn
       try_parse(event.dig('executionSucceededEventDetails', 'output'))
     end
 
-    def error(events_json = '{}')
+    def error(events_json = '{}', dry_run = false)
       return if event['executionFailedEventDetails'].nil?
 
       raise ExecutionError.new(event['executionFailedEventDetails']['cause'],
-                               event['executionFailedEventDetails']['error'],
-                               events_json)
+                              event['executionFailedEventDetails']['error'],
+                              events_json) unless dry_run
+      
+      event['executionFailedEventDetails']
     end
 
     def profile

data/lib/sfn/mock_macros/optimised_step_function.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Sfn
+  module MockMacros
+    module OptimisedStepFunction
+      def self.response(data)
+        data = [data] if data.is_a?(Hash)
+        data.map! do |val|
+          if val.key?(:error)
+            { Throw: { Error: val[:error], Cause: val[:cause] } }
+          else
+            val[:output] ||= val[:payload]
+            val[:output] ||= val[:response]
+            { Return: { Output: val[:output], Status: 'SUCCEDED' } }
+          end
+        end
+        out = {}
+        data.each_with_index do |val, idx|
+          out[idx.to_s] = val
+        end
+        out
+      end
+    end
+  end
+end

data/lib/sfn/mock_macros.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'sfn/mock_macros/lambda'
+require 'sfn/mock_macros/optimised_step_function'
 require 'sfn/mock_macros/step_function'
 require 'sfn/mock_macros/api_gateway'
 require 'sfn/mock_macros/sns'
@@ -9,6 +10,7 @@ require 'sfn/mock_macros/sqs'
 module Sfn
   module MockMacros
     include Lambda
+    include OptimisedStepFunction
     include StepFunction
     include ApiGateway
     include Sns
@@ -30,10 +32,14 @@ module Sfn
       Sqs.response(data)
     end
 
-    def self.step_function_response(data)
+    def self.step_function_response(data, optimised = false)
       StepFunction.response(data)
     end
 
+    def self.optimised_step_function_response(data)
+      OptimisedStepFunction.response(data)
+    end
+
     def self.gateway_payload(data)
       warn '[DEPRECATION] `gateway_payload` is deprecated.  Please use `gateway_response` instead.'
       ApiGateway.response(data)

data/lib/sfn/state_machine.rb

@@ -40,9 +40,13 @@ module Sfn
       Collection.instance.delete_by_arn(arn)
     end
 
-    def run(mock_data = {}, input = {}, test_name = nil)
+    def dry_run(mock_data = {}, input = {}, test_name = nil)
+      execution = run(mock_data, input, test_name, true)
+    end
+    
+    def run(mock_data = {}, input = {}, test_name = nil, dry_run = false)
       test_name ||= OpenSSL::Digest::SHA512.digest(mock_data.merge({ input: input }).to_json)
-      executions[test_name] ||= Execution.call(self, test_name, mock_data, input)
+      executions[test_name] ||= Execution.call(self, test_name, mock_data, input, dry_run)
       executions[test_name]
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-sfn-local
 version: !ruby/object:Gem::Version
-  version: 0.1.13
+  version: 0.1.17
 platform: ruby
 authors:
 - Gianni Mazza
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-07-02 00:00:00.000000000 Z
+date: 2022-09-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake-release
@@ -68,6 +68,7 @@ files:
 - lib/sfn/mock_macros.rb
 - lib/sfn/mock_macros/api_gateway.rb
 - lib/sfn/mock_macros/lambda.rb
+- lib/sfn/mock_macros/optimised_step_function.rb
 - lib/sfn/mock_macros/sns.rb
 - lib/sfn/mock_macros/sqs.rb
 - lib/sfn/mock_macros/step_function.rb