1 files changed, 18 insertions, 0 deletions
diff --git a/spec/lib/api/entities/draft_note_spec.rb b/spec/lib/api/entities/draft_note_spec.rb
new file mode 100644
index 00000000000..59555319bb1
--- /dev/null
+++ b/spec/lib/api/entities/draft_note_spec.rb
@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe API::Entities::DraftNote, feature_category: :code_review_workflow do
+  let_it_be(:entity) { create(:draft_note, :on_discussion) }
+  let_it_be(:json) { entity.as_json }
+
+  it 'exposes correct attributes' do
+    expect(json["id"]).to eq entity.id
+    expect(json["author_id"]).to eq entity.author_id
+    expect(json["merge_request_id"]).to eq entity.merge_request_id
+    expect(json["resolve_discussion"]).to eq entity.resolve_discussion
+    expect(json["discussion_id"]).to eq entity.discussion_id
+    expect(json["note"]).to eq entity.note
+    expect(json["position"].transform_keys(&:to_sym)).to eq entity.position.to_h
+  end
+end