gitlab-org--gitlab-foss/app/models/ci/unit_test.rb

46 lines
1.4 KiB
Ruby

# frozen_string_literal: true
module Ci
class UnitTest < ApplicationRecord
extend Gitlab::Ci::Model
MAX_NAME_SIZE = 255
MAX_SUITE_NAME_SIZE = 255
validates :project, :key_hash, :name, :suite_name, presence: true
has_many :unit_test_failures, class_name: 'Ci::UnitTestFailure'
belongs_to :project
scope :by_project_and_keys, -> (project, keys) { where(project_id: project.id, key_hash: keys) }
class << self
def find_or_create_by_batch(project, unit_test_attrs)
# Insert records first. Existing ones will be skipped.
insert_all(build_insert_attrs(project, unit_test_attrs))
# Find all matching records now that we are sure they all are persisted.
by_project_and_keys(project, gather_keys(unit_test_attrs))
end
private
def build_insert_attrs(project, unit_test_attrs)
# NOTE: Rails 6.1 will add support for insert_all on relation so that
# we will be able to do project.test_cases.insert_all.
unit_test_attrs.map do |attrs|
attrs.merge(
project_id: project.id,
name: attrs[:name].truncate(MAX_NAME_SIZE),
suite_name: attrs[:suite_name].truncate(MAX_SUITE_NAME_SIZE)
)
end
end
def gather_keys(unit_test_attrs)
unit_test_attrs.map { |attrs| attrs[:key_hash] }
end
end
end
end