diff --git a/test/servers/ServersDropdown.test.js b/test/servers/ServersDropdown.test.js index 3f8013f0..dde12a82 100644 --- a/test/servers/ServersDropdown.test.js +++ b/test/servers/ServersDropdown.test.js @@ -2,6 +2,7 @@ import { identity } from 'ramda'; import React from 'react'; import { ServersDropdown } from '../../src/servers/ServersDropdown'; import { shallow } from 'enzyme'; +import { DropdownItem, DropdownToggle } from 'reactstrap'; describe('', () => { let wrapped; @@ -13,10 +14,10 @@ describe('', () => { afterEach(() => wrapped.unmount()); it('contains the list of servers', () => { - expect(wrapped.find('DropdownItem').length).toEqual(servers.length); + expect(wrapped.find(DropdownItem).length).toEqual(servers.length + 2); }); it('contains a toggle with proper title', () => { - expect(wrapped.find('DropdownToggle').length).toEqual(1); + expect(wrapped.find(DropdownToggle).length).toEqual(1); }); });