1 files changed, 3 insertions, 2 deletions
diff --git a/spec/lib/gitlab/data_builder/wiki_page_spec.rb b/spec/lib/gitlab/data_builder/wiki_page_spec.rb
index ec768cf9719..276ddb1e0af 100644
--- a/spec/lib/gitlab/data_builder/wiki_page_spec.rb
+++ b/spec/lib/gitlab/data_builder/wiki_page_spec.rb
@@ -4,8 +4,8 @@ require 'spec_helper'
 
 RSpec.describe Gitlab::DataBuilder::WikiPage do
   let_it_be(:project) { create(:project, :repository, :wiki_repo) }
-  let(:wiki_page) { create(:wiki_page, wiki: project.wiki) }
-  let(:user) { create(:user) }
+  let_it_be(:wiki_page) { create(:wiki_page, wiki: project.wiki) }
+  let_it_be(:user) { create(:user) }
 
   describe '.build' do
     let(:data) { described_class.build(wiki_page, user, 'create') }
@@ -19,5 +19,6 @@ RSpec.describe Gitlab::DataBuilder::WikiPage do
     it { expect(data[:object_attributes]).to include(wiki_page.hook_attrs) }
     it { expect(data[:object_attributes]).to include(url: Gitlab::UrlBuilder.build(wiki_page)) }
     it { expect(data[:object_attributes]).to include(action: 'create') }
+    it { expect(data[:object_attributes]).to include(diff_url: Gitlab::UrlBuilder.build(wiki_page, action: :diff, version_id: wiki_page.version.id)) }
   end
 end