Fix broken tests

2 files changed, 4 insertions, 2 deletions

diff --git a/spec/javascripts/notes/components/issue_note_awards_list_spec.js b/spec/javascripts/notes/components/issue_note_awards_list_spec.js
index d4156e29ba0..3b6c34f1494 100644
--- a/spec/javascripts/notes/components/issue_note_awards_list_spec.js
+++ b/spec/javascripts/notes/components/issue_note_awards_list_spec.js
@@ -39,7 +39,8 @@ describe('issue_note_awards_list component', () => {
   });
 
   it('should render awarded emojis', () => {
-    expect(vm.$el.querySelectorAll('.js-awards-block button').length).toEqual(awardsMock.length);
+    expect(vm.$el.querySelector('.js-awards-block button [data-name="flag_tz"]')).toBeDefined();
+    expect(vm.$el.querySelector('.js-awards-block button [data-name="cartwheel_tone3"]')).toBeDefined();
   });
 
   it('should be possible to remove awareded emoji', () => {
diff --git a/spec/javascripts/notes/components/issue_note_body_spec.js b/spec/javascripts/notes/components/issue_note_body_spec.js
index c0e06b0985b..81f07ed47cc 100644
--- a/spec/javascripts/notes/components/issue_note_body_spec.js
+++ b/spec/javascripts/notes/components/issue_note_body_spec.js
@@ -40,6 +40,7 @@ describe('issue_note_body component', () => {
   });
 
   it('should render awards list', () => {
-    expect(vm.$el.querySelectorAll('.js-awards-block button').length).toEqual(note.award_emoji.length);
+    expect(vm.$el.querySelector('.js-awards-block button [data-name="baseball"]')).toBeDefined();
+    expect(vm.$el.querySelector('.js-awards-block button [data-name="bath_tone3"]')).toBeDefined();
   });
 });