Use component selector in serverless function_row_spec.jswinh-serverless-tests-selectors

1 files changed, 17 insertions, 12 deletions

diff --git a/spec/frontend/serverless/components/function_row_spec.js b/spec/frontend/serverless/components/function_row_spec.js
index 414fdc5cd82..979f98c4832 100644
--- a/spec/frontend/serverless/components/function_row_spec.js
+++ b/spec/frontend/serverless/components/function_row_spec.js
@@ -1,27 +1,32 @@
 import functionRowComponent from '~/serverless/components/function_row.vue';
 import { shallowMount } from '@vue/test-utils';
+import Timeago from '~/vue_shared/components/time_ago_tooltip.vue';
 
 import { mockServerlessFunction } from '../mock_data';
 
-const createComponent = func =>
-  shallowMount(functionRowComponent, { propsData: { func }, sync: false }).vm;
-
 describe('functionRowComponent', () => {
-  it('Parses the function details correctly', () => {
-    const vm = createComponent(mockServerlessFunction);
+  let wrapper;
 
-    expect(vm.$el.querySelector('b').innerHTML).toEqual(mockServerlessFunction.name);
-    expect(vm.$el.querySelector('span').innerHTML).toEqual(mockServerlessFunction.image);
-    expect(vm.$el.querySelector('timeago-stub').getAttribute('time')).not.toBe(null);
+  const createComponent = func => {
+    wrapper = shallowMount(functionRowComponent, { propsData: { func }, sync: false });
+  };
 
-    vm.$destroy();
+  afterEach(() => {
+    wrapper.destroy();
+  });
+
+  it('Parses the function details correctly', () => {
+    createComponent(mockServerlessFunction);
+
+    expect(wrapper.find('b').text()).toBe(mockServerlessFunction.name);
+    expect(wrapper.find('span').text()).toBe(mockServerlessFunction.image);
+    expect(wrapper.find(Timeago).attributes('time')).not.toBe(null);
   });
 
   it('handles clicks correctly', () => {
-    const vm = createComponent(mockServerlessFunction);
+    createComponent(mockServerlessFunction);
+    const { vm } = wrapper;
 
     expect(vm.checkClass(vm.$el.querySelector('p'))).toBe(true); // check somewhere inside the row
-
-    vm.$destroy();
   });
 });