1 files changed, 21 insertions, 0 deletions
diff --git a/spec/lib/api/entities/ml/mlflow/run_spec.rb b/spec/lib/api/entities/ml/mlflow/run_spec.rb
new file mode 100644
index 00000000000..84234f474f5
--- /dev/null
+++ b/spec/lib/api/entities/ml/mlflow/run_spec.rb
@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe API::Entities::Ml::Mlflow::Run do
+  let_it_be(:candidate) { create(:ml_candidates) }
+
+  subject { described_class.new(candidate).as_json }
+
+  it 'has run key' do
+    expect(subject).to have_key(:run)
+  end
+
+  it 'has the id' do
+    expect(subject[:run][:info][:run_id]).to eq(candidate.iid.to_s)
+  end
+
+  it 'data is empty' do
+    expect(subject[:run][:data]).to be_empty
+  end
+end