1 files changed, 5 insertions, 1 deletions
diff --git a/spec/serializers/ci/dag_stage_entity_spec.rb b/spec/serializers/ci/dag_stage_entity_spec.rb
index 5c6aa7faee4..0262ccdac68 100644
--- a/spec/serializers/ci/dag_stage_entity_spec.rb
+++ b/spec/serializers/ci/dag_stage_entity_spec.rb
@@ -2,7 +2,7 @@
 
 require 'spec_helper'
 
-describe Ci::DagStageEntity do
+RSpec.describe Ci::DagStageEntity do
   let_it_be(:pipeline) { create(:ci_pipeline) }
   let_it_be(:request) { double(:request) }
 
@@ -27,5 +27,9 @@ describe Ci::DagStageEntity do
       expect(job_group[:size]).to eq 1
       expect(job_group[:jobs]).not_to be_empty
     end
+
+    it "matches schema" do
+      expect(subject.to_json).to match_schema('entities/dag_stage')
+    end
   end
 end