ruby-sfn-local 0.1.12 → 0.1.14

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 05122d4c47a198bda25cdb97029a921f3e84a081f2bab2283124c8ee4eefb102
-  data.tar.gz: 212f8edb5259942315dd586ed34c453618adc245164db27b307123b66d9f8ba5
+  metadata.gz: ebdc76d783d70c6ffd4a53ad77d2ccad609f51454b6d8888cc03b71c9d2a3f8a
+  data.tar.gz: 8d4bd280dab1cc12fb4bd2fb25995c6afb4096ed7862d65df6a7d7c10c57483c
 SHA512:
-  metadata.gz: 31b1037fbad8a590aa1df037ab4f57709666e462c8ef9cc05242de74d891461bde6eee0381eaac26b449ef3b7c5864d6df3b6fa7e55ba3403f0e2e9e6cfba77b
-  data.tar.gz: 128ab49678e6a0d6622811f7610d135ea89d054a2700f915efe40a8e6c37f2259afc84ea08f64c6ca30f72783f33f7b7b83d9fe27cf7168419d45f12037bf653
+  metadata.gz: fff1c781578513f06501c8e01c25d72ea30ede3515170e17c4d86fa4f6d0fe6da27a6e0b26d87882c8152c4a343d3a65052b0ab30d183168799bc02d42cb2e41
+  data.tar.gz: 2df1e930018156d74acdfadc64d9c9f4bfa63a3c27190067562e556856cc982b1e8df4c8a7f1a554679912115527411043d56c5a733019ead05cd16c89313806

data/lib/sfn/execution_log.rb

@@ -19,6 +19,7 @@ module Sfn
 
     def self.parse(execution_arn)
       profile = {}
+      root_state_name = {}
       output = nil
       error = nil
       state_name = nil
@@ -26,19 +27,19 @@ module Sfn
                                { 'execution-arn': execution_arn.to_s, query: "'events[?#{EVENTS.join(' || ')}]'" })
       JSON.parse(events_json).each do |event|
         parsed_event = new(event)
-
         output ||= parsed_event.output
         error  ||= parsed_event.error(events_json)
-        last_state_name = state_name
         state_name = parsed_event.state_name
 
         unless state_name.nil?
           profile[state_name] ||= { 'input' => [], 'output' => [], 'parameters' => [] }
           profile[state_name]['input'] << parsed_event.profile['input'] unless parsed_event.profile['input'].nil?
           profile[state_name]['output'] << parsed_event.profile['output'] unless parsed_event.profile['output'].nil?
+          root_state_name[parsed_event.event['id']] = state_name
         end
-        if !last_state_name.nil? && !parsed_event.profile['parameters'].nil?
-          profile[last_state_name]['parameters'] << parsed_event.profile['parameters']
+
+        if !root_state_name[parsed_event.event['previousEventId']].nil? && !parsed_event.profile['parameters'].nil?
+          profile[root_state_name[parsed_event.event['previousEventId']]]['parameters'] << parsed_event.profile['parameters']
         end
       end
       [output, 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

@@ -29,7 +29,7 @@ module Sfn
 
     def initialize(name, arn = nil)
       self.path = "#{Sfn.configuration.definition_path}/#{name}.json"
-      self.name = name.split("/").last
+      self.name = name.split('/').last
       self.arn = arn || self.class.find_by_name(self.name)&.arn || create_state_machine
       self.executions = {}
     end
@@ -41,7 +41,7 @@ module Sfn
     end
 
     def run(mock_data = {}, input = {}, test_name = nil)
-      test_name ||= OpenSSL::Digest::SHA512.digest(mock_data.merge({input: input}).to_json)
+      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]
     end
@@ -63,8 +63,8 @@ module Sfn
 
     def load_definition
       local_definition_path = Tempfile.new(['name', '.json']).path
-      
-      definition = File.read(self.path)
+
+      definition = File.read(path)
       local_definition = definition.gsub(/"MaxConcurrency": [0-9]+/, '"MaxConcurrency": 1')
 
       File.open(local_definition_path, 'w') { |file| file.puts local_definition }

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-sfn-local
 version: !ruby/object:Gem::Version
-  version: 0.1.12
+  version: 0.1.14
 platform: ruby
 authors:
 - Gianni Mazza
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-06-23 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