Skip to content
代码片段 群组 项目
提交 fc2191ea 编辑于 作者: Pavel Shutsin's avatar Pavel Shutsin
浏览文件

Cleanup Ai::UsageEvents spec factories

Pure refactoring
上级 bddaa8ee
No related branches found
No related tags found
无相关合并请求
......@@ -6,7 +6,7 @@ class CreateAiCodeSuggestionEvents < Gitlab::Database::Migration[2.2]
def up
# rubocop:disable Migration/Datetime -- "timestamp" is a column name
create_table :ai_code_suggestion_events, # rubocop:disable Migration/EnsureFactoryForTable -- code_suggestion_event
create_table :ai_code_suggestion_events,
options: 'PARTITION BY RANGE (timestamp)',
primary_key: [:id, :timestamp] do |t|
t.bigserial :id, null: false
......
# frozen_string_literal: true
FactoryBot.define do
factory :code_suggestion_event, class: '::Ai::CodeSuggestionEvent' do
factory :ai_code_suggestion_event, class: '::Ai::CodeSuggestionEvent' do
event { 'code_suggestion_shown_in_ide' }
user
payload { { language: 'ruby', suggestion_size: 1, unique_tracking_id: SecureRandom.hex.slice(0, 20) } }
......
# frozen_string_literal: true
FactoryBot.define do
factory :duo_chat_event, class: '::Ai::DuoChatEvent' do
factory :ai_duo_chat_event, class: '::Ai::DuoChatEvent' do
event { 'request_duo_chat_response' }
user { build_stubbed(:user, :with_namespace) }
payload { {} }
......
......@@ -10,10 +10,12 @@
let_it_be(:user_contributor_2) { create(:user) }
let_it_be(:user_contributor_only_on_ch) { create(:user) }
let_it_be(:user_not_contributor) { create(:user) }
let_it_be(:code_suggestion_event_1) { create(:code_suggestion_event, :shown, user: user_contributor_1) }
let_it_be(:code_suggestion_event_2) { create(:code_suggestion_event, :shown, user: user_contributor_2) }
let_it_be(:code_suggestion_event_3) { create(:code_suggestion_event, :accepted, user: user_not_contributor) }
let_it_be(:code_suggestion_event_4) { create(:code_suggestion_event, :accepted, user: user_contributor_only_on_ch) }
let_it_be(:code_suggestion_event_1) { create(:ai_code_suggestion_event, :shown, user: user_contributor_1) }
let_it_be(:code_suggestion_event_2) { create(:ai_code_suggestion_event, :shown, user: user_contributor_2) }
let_it_be(:code_suggestion_event_3) { create(:ai_code_suggestion_event, :accepted, user: user_not_contributor) }
let_it_be(:code_suggestion_event_4) do
create(:ai_code_suggestion_event, :accepted, user: user_contributor_only_on_ch)
end
subject(:results) { described_class.new(user, resource: group).execute }
......
......@@ -39,10 +39,10 @@
let(:filter_params) { {} }
let(:expected_filters) { {} }
let_it_be(:code_suggestion_event_1) { create(:code_suggestion_event, :shown, user: user_1) }
let_it_be(:code_suggestion_event_2) { create(:code_suggestion_event, :accepted, user: user_1) }
let_it_be(:code_suggestion_event_3) { create(:code_suggestion_event, :accepted, user: user_2) }
let_it_be(:code_suggestion_event_4) { create(:code_suggestion_event, :accepted, user: user_3) }
let_it_be(:code_suggestion_event_1) { create(:ai_code_suggestion_event, :shown, user: user_1) }
let_it_be(:code_suggestion_event_2) { create(:ai_code_suggestion_event, :accepted, user: user_1) }
let_it_be(:code_suggestion_event_3) { create(:ai_code_suggestion_event, :accepted, user: user_2) }
let_it_be(:code_suggestion_event_4) { create(:ai_code_suggestion_event, :accepted, user: user_3) }
before do
allow(Gitlab::ClickHouse).to receive(:enabled_for_analytics?).and_return(true)
......
0% 加载中 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册