1 files changed, 14 insertions, 0 deletions
diff --git a/spec/lib/api/entities/ml/mlflow/run_info_spec.rb b/spec/lib/api/entities/ml/mlflow/run_info_spec.rb
index d5a37f53e21..db8f106c9fe 100644
--- a/spec/lib/api/entities/ml/mlflow/run_info_spec.rb
+++ b/spec/lib/api/entities/ml/mlflow/run_info_spec.rb
@@ -33,6 +33,20 @@ RSpec.describe API::Entities::Ml::Mlflow::RunInfo do
     end
   end
 
+  describe 'run_name' do
+    context 'when nil' do
+      it { is_expected.not_to have_key(:run_name) }
+    end
+
+    context 'when not nil' do
+      before do
+        allow(candidate).to receive(:name).and_return('hello')
+      end
+
+      it { expect(subject[:run_name]).to eq('hello') }
+    end
+  end
+
   describe 'experiment_id' do
     it 'is the experiment iid as string' do
       expect(subject[:experiment_id]).to eq(candidate.experiment.iid.to_s)