gitlab-org--gitlab-foss/spec/factories/ci/build_trace_chunks.rb

# frozen_string_literal: true

FactoryBot.define do
  factory :ci_build_trace_chunk, class: 'Ci::BuildTraceChunk' do
    build factory: :ci_build
    chunk_index { 0 }
    data_store { :redis }

    trait :redis_with_data do
      data_store { :redis }

      transient do
        initial_data { 'test data' }
      end

      after(:create) do |build_trace_chunk, evaluator|
        Ci::BuildTraceChunks::Redis.new.set_data(build_trace_chunk, evaluator.initial_data)
      end
    end

    trait :redis_without_data do
      data_store { :redis }
    end

    trait :database_with_data do
      data_store { :database}

      transient do
        initial_data { 'test data' }
      end

      after(:build) do |build_trace_chunk, evaluator|
        Ci::BuildTraceChunks::Database.new.set_data(build_trace_chunk, evaluator.initial_data)
      end
    end

    trait :database_without_data do
      data_store { :database }
    end

    trait :fog_with_data do
      data_store { :fog }

      transient do
        initial_data { 'test data' }
      end

      after(:create) do |build_trace_chunk, evaluator|
        Ci::BuildTraceChunks::Fog.new.set_data(build_trace_chunk, evaluator.initial_data)
      end
    end

    trait :fog_without_data do
      data_store { :fog }
    end
  end
end