prefab-cloud-ruby 0

data/test/test_local_config_parser.rb

@@ -0,0 +1,147 @@
+# frozen_string_literal: true
+
+require 'test_helper'
+
+class TestLocalConfigParser < Minitest::Test
+  FILE_NAME = 'example-config.yaml'
+  DEFAULT_MATCH = 'default'
+
+  def setup
+    super
+    @mock_resolver = MockResolver.new
+  end
+
+  def test_parse_int_config
+    key = :sample_int
+    parsed = Prefab::LocalConfigParser.parse(key, 123, {}, FILE_NAME)[key]
+    config = parsed[:config]
+
+    assert_equal FILE_NAME, parsed[:source]
+    assert_equal DEFAULT_MATCH, parsed[:match]
+    assert_equal :CONFIG, config.config_type
+    assert_equal key.to_s, config.key
+    assert_equal 1, config.rows.size
+    assert_equal 1, config.rows[0].values.size
+    assert_equal 123, config.rows[0].values[0].value.int
+  end
+
+  def test_flag_with_a_value
+    key = :flag_with_a_value
+    value = stringify_keys({ feature_flag: true, value: 'all-features' })
+    parsed = Prefab::LocalConfigParser.parse(key, value, {}, FILE_NAME)[key]
+    config = parsed[:config]
+
+    assert_equal FILE_NAME, parsed[:source]
+    assert_equal key, parsed[:match]
+    assert_equal :FEATURE_FLAG, config.config_type
+    assert_equal key.to_s, config.key
+    assert_equal 1, config.rows.size
+    assert_equal 1, config.rows[0].values.size
+
+    value_row = config.rows[0].values[0]
+    assert_equal 'all-features', Prefab::ConfigValueUnwrapper.deepest_value(value_row.value, key, {}, @mock_resolver).unwrap
+  end
+
+  def test_flag_in_user_key
+    key = :flag_in_user_key
+    value = stringify_keys({ 'feature_flag': 'true', value: true,
+                             criterion: { operator: 'PROP_IS_ONE_OF', property: 'user.key', values: %w[abc123 xyz987] } })
+    parsed = Prefab::LocalConfigParser.parse(key, value, {}, FILE_NAME)[key]
+    config = parsed[:config]
+
+    assert_equal FILE_NAME, parsed[:source]
+    assert_equal key, parsed[:match]
+    assert_equal :FEATURE_FLAG, config.config_type
+    assert_equal key.to_s, config.key
+    assert_equal 1, config.rows.size
+    assert_equal 1, config.rows[0].values.size
+    assert_equal 1, config.rows[0].values[0].criteria.size
+
+    value_row = config.rows[0].values[0]
+    assert_equal true, Prefab::ConfigValueUnwrapper.deepest_value(value_row.value, key, {}, @mock_resolver).unwrap
+
+    assert_equal 'user.key', value_row.criteria[0].property_name
+    assert_equal :PROP_IS_ONE_OF, value_row.criteria[0].operator
+    assert_equal %w[abc123 xyz987], value_row.criteria[0].value_to_match.string_list.values
+  end
+
+  def test_provided_values
+    with_env('LOOKUP_ENV', 'from env') do 
+      key = :test_provided
+      value = stringify_keys({type: 'provided', source: 'ENV_VAR', lookup: 'LOOKUP_ENV'})
+      parsed = Prefab::LocalConfigParser.parse(key, value, {}, FILE_NAME)[key]
+      config = parsed[:config]
+
+      assert_equal FILE_NAME, parsed[:source]
+      assert_equal 'LOOKUP_ENV', parsed[:match]
+      assert_equal :CONFIG, config.config_type
+      assert_equal key.to_s, config.key
+      assert_equal 1, config.rows.size
+      assert_equal 1, config.rows[0].values.size
+      
+      value_row = config.rows[0].values[0]            
+      provided = value_row.value.provided
+      assert_equal :ENV_VAR, provided.source
+      assert_equal 'LOOKUP_ENV', provided.lookup
+      assert_equal 'from env', Prefab::ConfigValueUnwrapper.deepest_value(value_row.value, config, {}, @mock_resolver).unwrap
+      reportable_value = Prefab::ConfigValueUnwrapper.deepest_value(value_row.value, config, {}, @mock_resolver).reportable_value
+      assert_equal 'from env', reportable_value
+    end
+  end
+
+  def test_confidential_provided_values
+    with_env('LOOKUP_ENV', 'from env') do
+      key = :test_provided
+      value = stringify_keys({type: 'provided', source: 'ENV_VAR', lookup: 'LOOKUP_ENV', confidential: true})
+      parsed = Prefab::LocalConfigParser.parse(key, value, {}, FILE_NAME)[key]
+      config = parsed[:config]
+
+      value_row = config.rows[0].values[0]
+      provided = value_row.value.provided
+      assert_equal :ENV_VAR, provided.source
+      assert_equal 'LOOKUP_ENV', provided.lookup
+      assert_equal 'from env', Prefab::ConfigValueUnwrapper.deepest_value(value_row.value, config, {}, @mock_resolver).unwrap
+      reportable_value = Prefab::ConfigValueUnwrapper.deepest_value(value_row.value, config, {}, @mock_resolver).reportable_value
+      assert reportable_value.start_with? Prefab::ConfigValueUnwrapper::CONFIDENTIAL_PREFIX
+    end
+  end
+
+  def test_confidential_values
+    key = :test_confidential
+    value = stringify_keys({value: 'a confidential string', confidential: true})
+    parsed = Prefab::LocalConfigParser.parse(key, value, {}, FILE_NAME)[key]
+    config = parsed[:config]
+
+    assert_equal FILE_NAME, parsed[:source]
+    assert_equal :CONFIG, config.config_type
+    assert_equal key.to_s, config.key
+    assert_equal 1, config.rows.size
+    assert_equal 1, config.rows[0].values.size
+
+    value_row = config.rows[0].values[0]
+    config_value = value_row.value
+    assert_equal 'a confidential string', Prefab::ConfigValueUnwrapper.deepest_value(config_value, key, {}, @mock_resolver).unwrap
+    reportable_value = Prefab::ConfigValueUnwrapper.deepest_value(config_value, key, {}, @mock_resolver).reportable_value
+    assert reportable_value.start_with? Prefab::ConfigValueUnwrapper::CONFIDENTIAL_PREFIX
+  end
+
+  private
+
+  def stringify_keys(hash)
+    deep_transform_keys(hash, &:to_s)
+  end
+
+  def deep_transform_keys(hash, &block)
+    result = {}
+    hash.each do |key, value|
+      result[yield(key)] = value.is_a?(Hash) ? deep_transform_keys(value, &block) : value
+    end
+    result
+  end
+
+  class MockResolver
+    def get(key)
+      raise "unexpected key"
+    end
+  end
+end

data/test/test_log_path_aggregator.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+require 'test_helper'
+require 'timecop'
+
+class TestLogPathAggregator < Minitest::Test
+  MAX_WAIT = 2
+  SLEEP_TIME = 0.01
+
+  def test_push
+    client = new_client
+    aggregator = Prefab::LogPathAggregator.new(client: client, max_paths: 2, sync_interval: 1000)
+
+    aggregator.push('test.test_log_path_aggregator.test_push.1', ::Logger::INFO)
+    aggregator.push('test.test_log_path_aggregator.test_push.2', ::Logger::DEBUG)
+
+    assert_equal 2, aggregator.data.size
+
+    # we've reached the limit, so no more
+    aggregator.push('test.test_log_path_aggregator.test_push.3', ::Logger::INFO)
+    assert_equal 2, aggregator.data.size
+
+    assert_only_expected_logs
+  end
+
+  def test_sync
+    Timecop.freeze do
+      client = new_client(namespace: 'this.is.a.namespace')
+
+      2.times { client.log.info('here is a message') }
+      3.times { client.log.error('here is a message') }
+
+      requests = wait_for_post_requests(client) do
+        client.log_path_aggregator.send(:sync)
+      end
+
+      assert_equal '/api/v1/known-loggers', requests[0][0]
+      sent_logger = requests[0][1]
+      assert_equal 'this.is.a.namespace', sent_logger.namespace
+      assert_equal Prefab::TimeHelpers.now_in_ms, sent_logger.start_at
+      assert_equal Prefab::TimeHelpers.now_in_ms, sent_logger.end_at
+      assert_equal client.instance_hash, sent_logger.instance_hash
+      assert_includes sent_logger.loggers, PrefabProto::Logger.new(logger_name: 'test.test_log_path_aggregator.test_sync', infos: 2, errors: 3)
+      assert_includes sent_logger.loggers, PrefabProto::Logger.new(logger_name: 'cloud.prefab.client.client', debugs: 1) # spot test that internal logging is here too
+
+      assert_logged [
+                      'WARN  2023-08-09 15:18:12 -0400: cloud.prefab.client.configclient No success loading checkpoints',
+                      'ERROR 2023-08-09 15:18:12 -0400: test.test_log_path_aggregator.test_sync here is a message'
+                    ]
+    end
+  end
+
+  private
+
+  def new_client(overrides = {})
+    super(**{
+      prefab_datasources: Prefab::Options::DATASOURCES::ALL,
+      api_key: '123-development-yourapikey-SDK',
+      collect_sync_interval: 1000 # we'll trigger sync manually in our test
+    }.merge(overrides))
+  end
+end