Add latest changes from gitlab-org/gitlab@master

1 files changed, 24 insertions, 0 deletions

diff --git a/spec/features/markdown/sandboxed_mermaid_spec.rb b/spec/features/markdown/sandboxed_mermaid_spec.rb
index 0282d02d809..0ecba8df88b 100644
--- a/spec/features/markdown/sandboxed_mermaid_spec.rb
+++ b/spec/features/markdown/sandboxed_mermaid_spec.rb
@@ -53,4 +53,28 @@ RSpec.describe 'Sandboxed Mermaid rendering', :js, feature_category: :team_plann
       end
     end
   end
+
+  context 'in a project milestone' do
+    let(:milestone) { create(:project_milestone, project: project, description: description) }
+
+    it 'includes mermaid frame correctly' do
+      visit(project_milestone_path(project, milestone))
+
+      wait_for_requests
+
+      expect(page.html).to include(expected)
+    end
+  end
+
+  context 'in a group milestone' do
+    let(:group_milestone) { create(:group_milestone, description: description) }
+
+    it 'includes mermaid frame correctly' do
+      visit(group_milestone_path(group_milestone.group, group_milestone))
+
+      wait_for_requests
+
+      expect(page.html).to include(expected)
+    end
+  end
 end