require 'twine_test'

class TestOutputProcessor < TwineTest
  def setup
    super

    @twine_file = build_twine_file 'en', 'fr' do
      add_section 'Section' do
        add_definition key1: 'value1', tags: ['tag1']
        add_definition key2: 'value2', tags: ['tag1', 'tag2']
        add_definition key3: 'value3', tags: ['tag2']
        add_definition key4: { en: 'value4-en', fr: 'value4-fr' }
      end
    end
  end

  def test_includes_all_keys_by_default
    processor = Twine::Processors::OutputProcessor.new(@twine_file, {})
    result = processor.process('en')

    assert_equal %w(key1 key2 key3 key4), result.definitions_by_key.keys.sort
  end

  def test_filter_by_tag
    processor = Twine::Processors::OutputProcessor.new(@twine_file, { tags: [['tag1']] })
    result = processor.process('en')

    assert_equal %w(key1 key2), result.definitions_by_key.keys.sort
  end

  def test_filter_by_multiple_tags
    processor = Twine::Processors::OutputProcessor.new(@twine_file, { tags: [['tag1', 'tag2']] })
    result = processor.process('en')

    assert_equal %w(key1 key2 key3), result.definitions_by_key.keys.sort
  end

  def test_filter_untagged
    processor = Twine::Processors::OutputProcessor.new(@twine_file, { tags: [['tag1']], untagged: true })
    result = processor.process('en')

    assert_equal %w(key1 key2 key4), result.definitions_by_key.keys.sort
  end

  def test_include_translated
    processor = Twine::Processors::OutputProcessor.new(@twine_file, { include: :translated })
    result = processor.process('fr')

    assert_equal %w(key4), result.definitions_by_key.keys.sort
  end

  def test_include_untranslated
    processor = Twine::Processors::OutputProcessor.new(@twine_file, { include: :untranslated })
    result = processor.process('fr')

    assert_equal %w(key1 key2 key3), result.definitions_by_key.keys.sort
  end

  class TranslationFallback < TwineTest
    def setup
      super

      @twine_file = build_twine_file 'en', 'fr', 'de' do
        add_section 'Section' do
          add_definition key1: { en: 'value1-en', fr: 'value1-fr' }
        end
      end
    end

    def test_fallback_to_default_language
      processor = Twine::Processors::OutputProcessor.new(@twine_file, {})
      result = processor.process('de')
      
      assert_equal 'value1-en', result.definitions_by_key['key1'].translations['de']
    end

    def test_fallback_to_developer_language
      processor = Twine::Processors::OutputProcessor.new(@twine_file, {developer_language: 'fr'})
      result = processor.process('de')
      
      assert_equal 'value1-fr', result.definitions_by_key['key1'].translations['de']
    end
  end

end