streamdal 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ca014e8334d38a589d1cd3aab58362ac3095fc077a93c076301f2e1552f9b375
-  data.tar.gz: 77110e0cc07e851e7b21ef831bed28495353bd70fce865159c41b18d4a40f7d0
+  metadata.gz: bd49e6aba633028e6bd0af296c6ed7a69ca00f2b4bc9c26646d40bdd1e19d313
+  data.tar.gz: 745eb525e4290b1d23bda53000683e9a8761fe070a4b50bb011605f2f03d64a5
 SHA512:
-  metadata.gz: 227c07d36fd54e8bf35a79b434417f59c80e45c72678ca82ada629b8268b744b614616d28b196e48b85c93f7a031330ebc8884a8ca826553eb647953069d7890
-  data.tar.gz: 9c83c8388b8e871d60b1b670b6f919b6a05878c731b6b8978545f9b782885c195cb30d02f1ec160cba58974ad53f99d9506410301185c5e036128eca4c56bb3b
+  metadata.gz: c65cd7b9a1342e31092bbb183e55537b26cb1be2ca34609d5df44ae763ec6bdcf514febb0d85f95a02107ec5d6b65f66b02956ef2b4516cce178e4185af6712e
+  data.tar.gz: 22319f8bffbafea0aa41748f7a745f44698c6061589ebe1ba8ee2621127e0f4fb3b4d5990d476e172d3e54f45c66f88310c7d295becab73ae0cc0172a79bc432

data/lib/hostfunc.rb

@@ -43,9 +43,9 @@ module Streamdal
       req = Streamdal::Protos::WASMRequest.decode(data)
 
       begin
-        req_body = self._get_request_body_for_mode(req)
+        req_body = _get_request_body_for_mode(req)
       rescue => e
-        return self._http_request_response(caller, 400, e.to_s, {})
+        return _http_request_response(caller, 400, e.to_s, {})
       end
 
       # Attempt to make HTTP request
@@ -53,10 +53,17 @@ module Streamdal
       begin
         response = _make_http_request(req.step.http_request.request, req_body)
       rescue => e
-        return self._http_request_response(caller, 400, "Unable to execute HTTP request: #{e}", {})
+        return _http_request_response(caller, 400, "Unable to execute HTTP request: #{e}", {})
       end
 
-      self._http_request_response(caller, response.code, response.body, response.headers)
+      # Convert body to utf8
+      out = encode(response.body)
+
+      _http_request_response(caller, response.code, out, response.headers)
+    end
+
+    def encode(str)
+      str.force_encoding('ascii-8bit').encode('utf-8', invalid: :replace, undef: :replace, replace: '?')
     end
 
     private
@@ -95,22 +102,18 @@ module Streamdal
         }
 
         req.inter_step_result.to_json
-      when :HTTP_REQUEST_BODY_MODE_STATIC
-        http_req.body
       else
-        raise 'invalid http request body mode'
+        http_req.body
       end
     end
 
     ##
     # Performs an http request
     def _make_http_request(req, body)
-      if req.nil?
-        raise 'req is required'
-      end
+      raise 'req is required' if req.nil?
 
       options = {
-        headers: { "Content-Type": 'application/json', },
+        headers: { "Content-Type": 'application/json' }
       }
 
       req.headers.each { |key, value| options.headers[key] = value }
@@ -137,5 +140,25 @@ module Streamdal
         raise ArgumentError, "Invalid http request method: #{req.method}"
       end
     end
+
+    # Called by host functions to write memory to wasm instance so that
+    # the wasm module can read the result of a host function call
+    def write_to_memory(caller, res)
+      alloc = caller.export('alloc').to_func
+      memory = caller.export('memory').to_memory
+
+      # Serialize protobuf message
+      resp = res.to_proto
+
+      # Allocate memory for response
+      resp_ptr = alloc.call(resp.length)
+
+      # Write response to memory
+      memory.write(resp_ptr, resp)
+
+      # return 64bit integer where first 32 bits is the pointer, and the last 32 is the length
+      resp_ptr << 32 | resp.length
+    end
+
   end
 end

data/lib/spec_helper.rb

@@ -1,2 +1,12 @@
 require 'simplecov'
-SimpleCov.start
+SimpleCov.start
+
+# This environment variable exists so that we can run WASM tests
+# via the CI process for libs/wasm* and not just through the CI
+# of the ruby-sdk
+WASM_DIR = ENV['WASM_DIR'] || File.join(File.dirname(__FILE__), '..', 'test-assets', 'wasm')
+
+# Load the WebAssembly module
+def load_wasm(name)
+  File.read(File.join(WASM_DIR, name))
+end

data/lib/streamdal.rb

@@ -31,7 +31,6 @@ DEFAULT_HEARTBEAT_INTERVAL = 1 # 1 second
 MAX_PAYLOAD_SIZE = 1024 * 1024 # 1 megabyte
 
 module Streamdal
-
   OPERATION_TYPE_PRODUCER = 2
   OPERATION_TYPE_CONSUMER = 1
   CLIENT_TYPE_SDK = 1
@@ -39,7 +38,6 @@ module Streamdal
 
   # Data class to hold instantiated wasm functions
   class WasmFunction
-
     ##
     # Instance of an initialized wasm module and associated memory store
 
@@ -66,7 +64,6 @@ module Streamdal
     end
   end
 
-
   class Client
 
     ##
@@ -117,20 +114,14 @@ module Streamdal
 
       # Exit any remaining threads
       @workers.each do |w|
-        if w.running?
-          w.exit
-        end
+        w.exit if w.running?
       end
     end
 
     def process(data, audience)
-      if data.empty?
-        raise 'data is required'
-      end
+      raise 'data is required' if data.empty?
 
-      if audience.nil?
-        raise 'audience is required'
-      end
+      raise 'audience is required' if audience.nil?
 
       resp = Streamdal::Protos::SDKResponse.new
       resp.status = :EXEC_STATUS_TRUE
@@ -215,13 +206,9 @@ module Streamdal
               break
             end
 
-            if @cfg[:dry_run]
-              @log.debug "Running step '#{step.name}' in dry-run mode"
-            end
+            @log.debug "Running step '#{step.name}' in dry-run mode" if @cfg[:dry_run]
 
-            if wasm_resp.output_payload.length.positive?
-              resp.data = wasm_resp.output_payload
-            end
+            resp.data = wasm_resp.output_payload if wasm_resp.output_payload.length.positive?
 
             _handle_schema(aud, step, wasm_resp)
 
@@ -309,17 +296,11 @@ module Streamdal
     private
 
     def _validate_cfg(cfg)
-      if cfg[:streamdal_url].nil? || cfg[:streamdal_url].empty?
-        raise 'streamdal_url is required'
-      end
+      raise 'streamdal_url is required' if cfg[:streamdal_url].nil? || cfg[:streamdal_url].empty?
 
-      if cfg[:streamdal_token].nil? || cfg[:streamdal_token].empty?
-        raise 'streamdal_token is required'
-      end
+      raise 'streamdal_token is required' if cfg[:streamdal_token].nil? || cfg[:streamdal_token].empty?
 
-      if cfg[:service_name].nil? || cfg[:streamdal_token].empty?
-        raise 'service_name is required'
-      end
+      raise 'service_name is required' if cfg[:service_name].nil? || cfg[:streamdal_token].empty?
 
       if cfg[:log].nil? || cfg[:streamdal_token].empty?
         logger = Logger.new($stdout)
@@ -327,13 +308,9 @@ module Streamdal
         cfg[:log] = logger
       end
 
-      if cfg[:pipeline_timeout].nil?
-        cfg[:pipeline_timeout] = DEFAULT_PIPELINE_TIMEOUT
-      end
+      cfg[:pipeline_timeout] = DEFAULT_PIPELINE_TIMEOUT if cfg[:pipeline_timeout].nil?
 
-      if cfg[:step_timeout].nil?
-        cfg[:step_timeout] = DEFAULT_STEP_TIMEOUT
-      end
+      cfg[:step_timeout] = DEFAULT_STEP_TIMEOUT if cfg[:step_timeout].nil?
     end
 
     def _handle_command(cmd)
@@ -414,9 +391,7 @@ module Streamdal
     def _get_function(step)
       # We cache functions so we can eliminate the wasm bytes from steps to save on memory
       # And also to avoid re-initializing the same function multiple times
-      if @functions.key?(step._wasm_id)
-        return @functions[step._wasm_id]
-      end
+      return @functions[step._wasm_id] if @functions.key?(step._wasm_id)
 
       engine = Wasmtime::Engine.new
       mod = Wasmtime::Module.new(engine, step._wasm_bytes)
@@ -450,17 +425,11 @@ module Streamdal
     end
 
     def _call_wasm(step, data, isr)
-      if step.nil?
-        raise 'step is required'
-      end
+      raise 'step is required' if step.nil?
 
-      if data.nil?
-        raise 'data is required'
-      end
+      raise 'data is required' if data.nil?
 
-      if isr.nil?
-        isr = Streamdal::Protos::InterStepResult.new
-      end
+      isr = Streamdal::Protos::InterStepResult.new if isr.nil?
 
       req = Streamdal::Protos::WASMRequest.new
       req.step = step.clone
@@ -477,7 +446,7 @@ module Streamdal
         resp.exit_code = :WASM_EXIT_CODE_ERROR
         resp.exit_msg = "Failed to execute WASM: #{e}"
         resp.output_payload = ''
-        return resp
+        resp
       end
     end
 
@@ -516,9 +485,7 @@ module Streamdal
       # Register with Streamdal External gRPC API
       resps = @stub.register(_gen_register_request, metadata: _metadata)
       resps.each do |r|
-        if @exit
-          break
-        end
+        break if @exit
 
         _handle_command(r)
       end
@@ -564,9 +531,7 @@ module Streamdal
 
       _add_audience(aud)
 
-      if @pipelines.key?(aud_str)
-        return @pipelines[aud_str]
-      end
+      return @pipelines[aud_str] if @pipelines.key?(aud_str)
 
       []
     end
@@ -612,18 +577,14 @@ module Streamdal
 
     def _get_active_tails_for_audience(aud)
       aud_str = aud_to_str(aud)
-      if @tails.key?(aud_str)
-        return @tails[aud_str].values
-      end
+      return @tails[aud_str].values if @tails.key?(aud_str)
 
       []
     end
 
     def _send_tail(aud, pipeline_id, original_data, new_data)
       tails = _get_active_tails_for_audience(aud)
-      if tails.empty?
-        return nil
-      end
+      return nil if tails.empty?
 
       tails.each do |tail|
         req = Streamdal::Protos::TailResponse.new
@@ -640,19 +601,13 @@ module Streamdal
     end
 
     def _notify_condition(pipeline, step, aud, cond, data, cond_type)
-      if cond.nil?
-        return nil
-      end
+      return nil if cond.nil?
 
-      if cond.notification.nil?
-        return nil
-      end
+      return nil if cond.notification.nil?
 
       @log.debug 'Notifying'
 
-      if @cfg[:dry_run]
-        return nil
-      end
+      return nil if @cfg[:dry_run]
 
       @metrics.incr(CounterEntry.new(Metrics::COUNTER_NOTIFY, aud, {
         "service": @cfg[:service_name],
@@ -702,15 +657,12 @@ module Streamdal
       t.start_tail_workers
 
       _set_active_tail(t)
-
     end
 
     def _set_active_tail(tail)
       key = aud_to_str(tail.request.audience)
 
-      unless @tails.key?(key)
-        @tails[key] = {}
-      end
+      @tails[key] = {} unless @tails.key?(key)
 
       @tails[key][tail.request.id] = tail
     end
@@ -718,9 +670,7 @@ module Streamdal
     def _set_paused_tail(tail)
       key = aud_to_str(tail.request.aud)
 
-      unless @paused_tails.key?(key)
-        @paused_tails[key] = {}
-      end
+      @paused_tails[key] = {} unless @paused_tails.key?(key)
 
       @paused_tails[key][tail.request.id] = tail
     end
@@ -735,18 +685,14 @@ module Streamdal
         # Remove from active tails
         @tails[key].delete(cmd.tail.request.id)
 
-        if @tails[key].empty?
-          @tails.delete(key)
-        end
+        @tails.delete(key) if @tails[key].empty?
       end
 
-      if @paused_tails.key?(key) && @paused_tails[key].key?(cmd.tail.request.id)
-        @paused_tails[key].delete(cmd.tail.request.id)
+      return unless @paused_tails.key?(key) && @paused_tails[key].key?(cmd.tail.request.id)
 
-        if @paused_tails[key].empty?
-          @paused_tails.delete(key)
-        end
-      end
+      @paused_tails[key].delete(cmd.tail.request.id)
+
+      @paused_tails.delete(key) if @paused_tails[key].empty?
     end
 
     def _stop_all_tails
@@ -762,9 +708,7 @@ module Streamdal
           t.stop_tail
           tails[aud].delete(tail.request.id)
 
-          if tails[aud].empty?
-            tails.delete(aud)
-          end
+          tails.delete(aud) if tails[aud].empty?
         end
       end
     end
@@ -795,54 +739,30 @@ module Streamdal
     def _remove_active_tail(aud, tail_id)
       key = aud_to_str(aud)
 
-      if @tails.key?(key) && @tails[key].key?(tail_id)
-        t = @tails[key][tail_id]
-        t.stop_tail
+      return unless @tails.key?(key) && @tails[key].key?(tail_id)
 
-        @tails[key].delete(tail_id)
+      t = @tails[key][tail_id]
+      t.stop_tail
 
-        if @tails[key].empty?
-          @tails.delete(key)
-        end
+      @tails[key].delete(tail_id)
 
-        t
-      end
+      @tails.delete(key) if @tails[key].empty?
+
+      t
     end
 
     def _remove_paused_tail(aud, tail_id)
       key = aud_to_str(aud)
 
-      if @paused_tails.key?(key) && @paused_tails[key].key?(tail_id)
-        t = @paused_tails[key][tail_id]
+      return unless @paused_tails.key?(key) && @paused_tails[key].key?(tail_id)
 
-        @paused_tails[key].delete(tail_id)
+      t = @paused_tails[key][tail_id]
 
-        if @paused_tails[key].empty?
-          @paused_tails.delete(key)
-        end
+      @paused_tails[key].delete(tail_id)
 
-        t
-      end
-    end
-
-    # Called by host functions to write memory to wasm instance so that
-    # the wasm module can read the result of a host function call
-    def write_to_memory(caller, res)
-      alloc = caller.export('alloc').to_func
-      memory = caller.export('memory').to_memory
-
-      # Serialize protobuf message
-      resp = res.to_proto
+      @paused_tails.delete(key) if @paused_tails[key].empty?
 
-      # Allocate memory for response
-      resp_ptr = alloc.call(resp.length)
-
-      # Write response to memory
-      memory.write(resp_ptr, resp)
-
-      # return 64bit integer where first 32 bits is the pointer, and the last 32 is the length
-      resp_ptr << 32 | resp.length
+      t
     end
-
   end
-end
+end

data/lib/wasm_spec.rb

@@ -0,0 +1,259 @@
+require_relative 'spec_helper'
+require 'steps/sp_steps_httprequest_pb'
+require 'steps/sp_steps_detective_pb'
+require 'steps/sp_steps_transform_pb'
+require_relative 'streamdal'
+
+class TestClient < Streamdal::Client
+
+  attr_accessor :kv
+
+  # Ignore rubocop warning
+  # rubocop:disable Lint/MissingSuper
+  def initialize
+    @cfg = {
+      step_timeout: 2
+    }
+    @functions = {}
+
+    logger = Logger.new($stdout)
+    logger.level = Logger::ERROR
+
+    @log = logger
+    @kv = Streamdal::KeyValue.new
+    @hostfunc = Streamdal::HostFunc.new(@kv)
+  end
+end
+
+RSpec.describe 'WASM' do
+  let(:client) { TestClient.new }
+
+  context '_call_wasm' do
+    it 'raises an error if step is nil' do
+      expect { client.send(:_call_wasm, nil, nil, nil) }.to raise_error('step is required')
+    end
+
+    it 'raises an error if data is nil' do
+      step = Streamdal::Protos::HttpRequestStep.new
+      expect { client.send(:_call_wasm, step, nil, nil) }.to raise_error('data is required')
+    end
+  end
+
+  context 'detective.wasm' do
+    before(:each) do
+      wasm_bytes = load_wasm('detective.wasm')
+
+      @step = Streamdal::Protos::PipelineStep.new
+      @step.name = 'detective'
+      @step._wasm_bytes = wasm_bytes.b
+      @step._wasm_id = SecureRandom.uuid
+      @step._wasm_function = 'f'
+      @step.detective = Streamdal::Protos::DetectiveStep.new
+      @step.detective.path = 'object.field'
+      @step.detective.args = Google::Protobuf::RepeatedField.new(:string, ['streamdal'])
+      @step.detective.negate = false
+      @step.detective.type = :DETECTIVE_TYPE_STRING_CONTAINS_ANY
+    end
+
+    it 'detects email in JSON payload' do
+      data = '{"object":{"field":"streamdal@gmail.com"}}'
+
+      res = client.send(:_call_wasm, @step, data, nil)
+
+      expect(res).not_to be_nil
+      expect(res.exit_code).to eq(:WASM_EXIT_CODE_TRUE)
+      expect(res.output_payload).to eq(data)
+
+    end
+
+    it 'does not detect email in JSON payload' do
+      data = '{"object":{"field":"mark@gmail.com"}}'
+
+      res = client.send(:_call_wasm, @step, data, nil)
+      expect(res).not_to be_nil
+      expect(res.exit_code).to eq(:WASM_EXIT_CODE_FALSE)
+    end
+  end
+
+  context 'httprequest.wasm' do
+    before(:each) do
+      wasm_bytes = load_wasm('httprequest.wasm')
+
+      http_req_step = Streamdal::Protos::HttpRequestStep.new
+      http_req_step.request = Streamdal::Protos::HttpRequest.new
+      http_req_step.request.url = 'https://www.google.com/404_me'
+      http_req_step.request.method = :HTTP_REQUEST_METHOD_GET
+      http_req_step.request.body = ''
+
+      @step = Streamdal::Protos::PipelineStep.new
+      @step.name = 'http request'
+      @step._wasm_bytes = wasm_bytes.b
+      @step._wasm_id = SecureRandom.uuid
+      @step._wasm_function = 'f'
+      @step.http_request = http_req_step
+    end
+
+    it 'returns false on 404' do
+      res = client.send(:_call_wasm, @step, '', nil)
+
+      expect(res).not_to be_nil
+      expect(res.exit_msg).to eq('Request returned non-200 response code: 404')
+      expect(res.exit_code).to eq(:WASM_EXIT_CODE_FALSE)
+    end
+  end
+
+  context 'inferschema.wasm' do
+    before(:each) do
+      wasm_bytes = load_wasm('inferschema.wasm')
+
+      @step = Streamdal::Protos::PipelineStep.new
+      @step.name = 'schema inference'
+      @step._wasm_bytes = wasm_bytes.b
+      @step._wasm_id = SecureRandom.uuid
+      @step._wasm_function = 'f'
+      @step.infer_schema = Streamdal::Protos::InferSchemaStep.new
+    end
+
+    it 'infers schema' do
+      payload = '{"object": {"payload": "test"}}'
+      res = client.send(:_call_wasm, @step, payload, nil)
+
+      expected_schema = '{"$schema":"http://json-schema.org/draft-07/schema#","properties":{"object":{"properties":{"payload":{"type":"string"}},"required":["payload"],"type":"object"}},"required":["object"],"type":"object"}'
+
+      expect(res).not_to be_nil
+      expect(res.exit_msg).to eq('inferred fresh schema')
+      expect(res.exit_code).to eq(:WASM_EXIT_CODE_TRUE)
+      expect(res.output_payload).to eq(payload)
+      expect(res.output_step).to eq(expected_schema)
+    end
+  end
+
+  context 'transform.wasm' do
+    before(:each) do
+      wasm_bytes = load_wasm('transform.wasm')
+
+      @step = Streamdal::Protos::PipelineStep.new
+      @step.name = 'transform'
+      @step._wasm_bytes = wasm_bytes.b
+      @step._wasm_id = SecureRandom.uuid
+      @step._wasm_function = 'f'
+    end
+
+    it 'deletes a field from a payload' do
+      @step.transform = Streamdal::Protos::TransformStep.new
+      @step.transform.type = :TRANSFORM_TYPE_DELETE_FIELD
+      @step.transform.delete_field_options = Streamdal::Protos::TransformDeleteFieldOptions.new
+      @step.transform.delete_field_options.paths = Google::Protobuf::RepeatedField.new(:string, ['object.another'])
+
+      payload = '{"object": {"payload": "old val", "another": "field"}}'
+      res = client.send(:_call_wasm, @step, payload, nil)
+
+      expect(res).not_to be_nil
+      expect(res.exit_msg).to eq('Successfully transformed payload')
+      expect(res.exit_code).to eq(:WASM_EXIT_CODE_TRUE)
+      expect(res.output_payload).to eq('{"object": {"payload": "old val"}}')
+    end
+
+    it 'replaces a fields value with a new one' do
+      @step.transform = Streamdal::Protos::TransformStep.new
+      @step.transform.type = :TRANSFORM_TYPE_REPLACE_VALUE
+      @step.transform.replace_value_options = Streamdal::Protos::TransformReplaceValueOptions.new
+      @step.transform.replace_value_options.path = 'object.payload'
+      @step.transform.replace_value_options.value = '"new val"'
+
+      payload = '{"object": {"payload": "old val"}}'
+      res = client.send(:_call_wasm, @step, payload, nil)
+
+      expect(res).not_to be_nil
+      expect(res.exit_msg).to eq('Successfully transformed payload')
+      expect(res.exit_code).to eq(:WASM_EXIT_CODE_TRUE)
+      expect(res.output_payload).to eq('{"object": {"payload": "new val"}}')
+    end
+
+    it 'truncates the value of a field' do
+      @step.transform = Streamdal::Protos::TransformStep.new
+      @step.transform.type = :TRANSFORM_TYPE_TRUNCATE_VALUE
+      @step.transform.truncate_options = Streamdal::Protos::TransformTruncateOptions.new
+      @step.transform.truncate_options.type = :TRANSFORM_TRUNCATE_TYPE_LENGTH
+      @step.transform.truncate_options.path = 'object.payload'
+      @step.transform.truncate_options.value = 3
+
+      payload = '{"object": {"payload": "old val"}}'
+      res = client.send(:_call_wasm, @step, payload, nil)
+
+      expect(res).not_to be_nil
+      expect(res.exit_msg).to eq('Successfully transformed payload')
+      expect(res.exit_code).to eq(:WASM_EXIT_CODE_TRUE)
+      expect(res.output_payload).to eq('{"object": {"payload": "old"}}')
+    end
+
+    it 'performs dynamic transformation' do
+      # TODO: add this test
+    end
+  end
+
+  context 'validjson.wasm' do
+    before(:each) do
+      wasm_bytes = load_wasm('validjson.wasm')
+
+      @step = Streamdal::Protos::PipelineStep.new
+      @step.name = 'validate json'
+      @step._wasm_bytes = wasm_bytes.b
+      @step._wasm_id = SecureRandom.uuid
+      @step._wasm_function = 'f'
+      @step.valid_json = Streamdal::Protos::ValidJSONStep.new
+
+    end
+
+    it 'validates a valid JSON payload' do
+
+      payload = '{"object": {"payload": "test"}}'
+      res = client.send(:_call_wasm, @step, payload, nil)
+
+      expect(res).not_to be_nil
+      expect(res.exit_code).to eq(:WASM_EXIT_CODE_TRUE)
+      expect(res.output_payload).to eq(payload)
+    end
+
+    it 'returns false on invalid JSON payload' do
+      payload = '{"object": {"payload": "test"'
+      res = client.send(:_call_wasm, @step, payload, nil)
+
+      expect(res).not_to be_nil
+      expect(res.exit_code).to eq(:WASM_EXIT_CODE_FALSE)
+    end
+  end
+
+  context 'kv.wasm' do
+    before(:each) do
+      wasm_bytes = load_wasm('kv.wasm')
+
+      client.kv.set('test', 'test')
+
+      @step = Streamdal::Protos::PipelineStep.new
+      @step.name = 'kv exists'
+      @step._wasm_bytes = wasm_bytes.b
+      @step._wasm_id = SecureRandom.uuid
+      @step._wasm_function = 'f'
+      @step.kv = Streamdal::Protos::KVStep.new
+      @step.kv.key = 'test'
+      @step.kv.mode = :KV_MODE_STATIC
+      @step.kv.action = :KV_ACTION_EXISTS
+    end
+
+    it 'returns true if a key exists' do
+      res = client.send(:_call_wasm, @step, '', nil)
+
+      expect(res).not_to be_nil
+      expect(res.exit_code).to eq(:WASM_EXIT_CODE_TRUE)
+    end
+
+    it 'returns false when a key doesnt exist' do
+      @step.kv.key = 'not_exists'
+      res = client.send(:_call_wasm, @step, '', nil)
+
+      expect(res).not_to be_nil
+      expect(res.exit_code).to eq(:WASM_EXIT_CODE_FALSE)
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: streamdal
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
 platform: ruby
 authors:
 - Mark Gregan
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-06-19 00:00:00.000000000 Z
+date: 2024-06-20 00:00:00.000000000 Z
 dependencies: []
 description: 
 email: mark@streamdal.com
@@ -33,6 +33,7 @@ files:
 - lib/tail_spec.rb
 - lib/validation.rb
 - lib/validation_spec.rb
+- lib/wasm_spec.rb
 homepage: https://docs.streamdal.com
 licenses:
 - Apache-2.0
@@ -45,7 +46,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - '='
     - !ruby/object:Gem::Version
-      version: 0.0.2
+      version: 0.0.4
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="