ResultPath=$ replaces complete output

lib/floe/workflow/reference_path.rb

@@ -25,12 +25,9 @@ def get(context)
       def set(context, value)
         result = context.dup
 
-        # If the payload is '$' then merge the value into the context
-        # otherwise store the value under the path
-        #
-        # TODO: how to handle non-hash values, raise error if path=$ and value not a hash?
+        # If the payload is '$' then replace the output with the value
         if path.empty?
-          result.merge!(value)
+          result = value.dup
         else
           child    = result
           keys     = path.dup

spec/workflow/reference_path_spec.rb

@@ -50,19 +50,27 @@
 
   describe "#set" do
     let(:payload) { "$" }
-    let(:input) { {} }
+    let(:input) { {"old" => "key"} }
 
     context "with a simple path" do
       it "sets the output at the top-level" do
         expect(subject.set(input, "foo" => "bar")).to eq("foo" => "bar")
       end
     end
 
+    context "with a top level path" do
+      let(:payload) { "$.hash" }
+
+      it "sets the output at the correct nested level" do
+        expect(subject.set(input, "foo" => "bar")).to eq("old" => "key", "hash" => {"foo" => "bar"})
+      end
+    end
+
     context "with a nested path" do
       let(:payload) { "$.nested.hash" }
 
       it "sets the output at the correct nested level" do
-        expect(subject.set(input, "foo" => "bar")).to eq("nested" => {"hash" => {"foo" => "bar"}})
+        expect(subject.set(input, "foo" => "bar")).to eq("old" => "key", "nested" => {"hash" => {"foo" => "bar"}})
       end
     end
 

spec/workflow/states/task_spec.rb

@@ -51,7 +51,7 @@
 
         workflow.current_state.run_nonblock!
 
-        expect(ctx.output).to eq("foo" => {"bar" => "baz"}, "bar" => {"baz" => "foo"}, "response" => ["192.168.1.2"])
+        expect(ctx.output).to eq("response" => ["192.168.1.2"])
       end
 
       context "with an error" do
@@ -80,7 +80,7 @@
 
           workflow.current_state.run_nonblock!
 
-          expect(ctx.output).to eq("foo" => {"bar" => "baz"}, "bar" => {"baz" => "foo"}, "ip_addrs" => ["192.168.1.2"])
+          expect(ctx.output).to eq("ip_addrs" => ["192.168.1.2"])
         end
       end