fix(getTarget): return array like when allElements parameter is true (#…

…1687)

* return array like when allElements parameter in getTarget is true

* Pass a dummy object to test addEventListener

Set ref.current to a dummy object with addEventListener and
removeEventListener.

src/__tests__/TooltipPopoverWrapper.spec.js

@@ -364,6 +364,7 @@ describe('Tooltip', () => {
 
   it('should not throw when passed a ref object as the target', () => {
     const targetObj = React.createRef();
+    targetObj.current = createSpyObj('div', ['addEventListener', 'removeEventListener']);
     const event = createSpyObj('event', ['preventDefault']);
     const wrapper = mount(
       <TooltipPopoverWrapper target={targetObj}>Yo!</TooltipPopoverWrapper>,
@@ -373,6 +374,9 @@ describe('Tooltip', () => {
     instance.toggle(event);
 
     wrapper.detach();
+
+    expect(targetObj.current.addEventListener).toHaveBeenCalled();
+    expect(targetObj.current.removeEventListener).toHaveBeenCalled();
   });
 
   describe('multi target', () => {

src/__tests__/utils.spec.js

@@ -130,6 +130,15 @@ describe('Utils', () => {
       document.querySelectorAll.mockRestore();
     });
 
+    it('should return elements as array like object if allElement param is true', () => {
+      const data = {};
+      const spy = jest.fn(() => data);
+      const elements = Utils.getTarget(spy, true);
+      expect(elements).toHaveProperty('length');
+      expect(elements).toContain(data);
+      expect(spy).toHaveBeenCalled();
+    });
+
     it('should query the document for the target if the target is a string', () => {
       const element = document.createElement('div');
       element.className = 'thing';

src/utils.js

@@ -301,10 +301,17 @@ export function isArrayOrNodeList(els) {
 
 export function getTarget(target, allElements) {
   const els = findDOMElements(target);
-  if (isArrayOrNodeList(els) && !allElements) {
-    return els[0];
+  if (allElements) {
+    if (isArrayOrNodeList(els)) {
+      return els;
+    }
+    return [els];
+  } else {
+    if (isArrayOrNodeList(els)) {
+      return els[0];
+    }
+    return els;
   }
-  return els;
 }
 
 export const defaultToggleEvents = ['touchstart', 'click'];