Fix tests

1 files changed, 7 insertions, 8 deletions

diff --git a/test/components/Account.spec.js b/test/components/Account.spec.js
index 2c0979bb50..ef2ffe6d16 100644
--- a/test/components/Account.spec.js
+++ b/test/components/Account.spec.js
@@ -3,6 +3,7 @@
 import * as React from 'react';
 import { shallow } from 'enzyme';
 import Account from '../../app/components/Account';
+import { BackBarItem } from '../../app/components/NavigationBar';
 import type { AccountProps } from '../../app/components/Account';
 
 describe('components/Account', () => {
@@ -25,8 +26,10 @@ describe('components/Account', () => {
     const props = makeProps({
       onClose: () => done(),
     });
-    const component = getComponent(render(props), 'account__close');
-    click(component);
+    const component = render(props)
+      .find(BackBarItem)
+      .dive();
+    component.simulate('press');
   });
 
   it('should call logout callback', (done) => {
@@ -34,7 +37,7 @@ describe('components/Account', () => {
       onLogout: () => done(),
     });
     const component = getComponent(render(props), 'account__logout');
-    click(component);
+    component.simulate('press');
   });
 
   it('should call "buy more" callback', (done) => {
@@ -42,7 +45,7 @@ describe('components/Account', () => {
       onBuyMore: () => done(),
     });
     const component = getComponent(render(props), 'account__buymore');
-    click(component);
+    component.simulate('press');
   });
 
   it('should display "out of time" message when account expired', () => {
@@ -67,7 +70,3 @@ function render(props) {
 function getComponent(container, testName) {
   return container.findWhere((n) => n.prop('testName') === testName);
 }
-
-function click(component) {
-  component.prop('onPress')();
-}