1 files changed, 4 insertions, 10 deletions
diff --git a/spec/models/discussion_spec.rb b/spec/models/discussion_spec.rb
index 187d0bc0150..2a67c60b978 100644
--- a/spec/models/discussion_spec.rb
+++ b/spec/models/discussion_spec.rb
@@ -595,23 +595,17 @@ describe Discussion, model: true do
     let(:truncated_lines) { subject.truncated_diff_lines }
 
     context "when diff is greater than allowed number of truncated diff lines " do
-      let(:initial_line_count) { subject.diff_lines.count }
-      let(:truncated_line_count) { truncated_lines.count }
-
       it "returns fewer lines"  do
-        expect(initial_line_count).to be > described_class::NUMBER_OF_TRUNCATED_DIFF_LINES
+        expect(subject.diff_lines.count).to be > described_class::NUMBER_OF_TRUNCATED_DIFF_LINES
 
-        expect(truncated_line_count).to be <= described_class::NUMBER_OF_TRUNCATED_DIFF_LINES
+        expect(truncated_lines.count).to be <= described_class::NUMBER_OF_TRUNCATED_DIFF_LINES
       end
     end
 
     context "when some diff lines are meta" do
-      let(:initial_meta_lines?) { subject.diff_lines.any?(&:meta?) }
-      let(:truncated_meta_lines?) { truncated_lines.any?(&:meta?) }
-
       it "returns no meta lines"  do
-        expect(initial_meta_lines?).to be true
-        expect(truncated_meta_lines?).to be false
+        expect(subject.diff_lines).to include(be_meta)
+        expect(truncated_lines).not_to include(be_meta)
       end
     end
   end