From 6c602261dd26f8eebf8a4ebd32e0bed8a08522ef Mon Sep 17 00:00:00 2001
From: David <david.acevedo@appfolio.com>
Date: Fri, 11 Dec 2020 10:53:22 -0800
Subject: [PATCH] feat: update input group to have type dropdown

---
 docs/lib/examples/InputGroupButton.js | 46 ++++++++++++---------------
 src/InputGroup.js                     |  7 ++++
 src/__tests__/InputGroup.spec.js      |  7 ++++
 3 files changed, 35 insertions(+), 25 deletions(-)

diff --git a/docs/lib/examples/InputGroupButton.js b/docs/lib/examples/InputGroupButton.js
index 13d42864b..0972f7ec1 100644
--- a/docs/lib/examples/InputGroupButton.js
+++ b/docs/lib/examples/InputGroupButton.js
@@ -24,34 +24,30 @@ const Example = (props) => {
         <Input />
       </InputGroup>
       <br />
-      <InputGroup>
+      <InputGroup type="dropdown" isOpen={dropdownOpen} toggle={toggleDropDown}>
         <Input />
-        <Dropdown isOpen={dropdownOpen} toggle={toggleDropDown}>
-          <DropdownToggle caret>
-            Button Dropdown
-          </DropdownToggle>
-          <DropdownMenu>
-            <DropdownItem header>Header</DropdownItem>
-            <DropdownItem disabled>Action</DropdownItem>
-            <DropdownItem>Another Action</DropdownItem>
-            <DropdownItem divider />
-            <DropdownItem>Another Action</DropdownItem>
-          </DropdownMenu>
-        </Dropdown>
+        <DropdownToggle caret>
+          Button Dropdown
+        </DropdownToggle>
+        <DropdownMenu>
+          <DropdownItem header>Header</DropdownItem>
+          <DropdownItem disabled>Action</DropdownItem>
+          <DropdownItem>Another Action</DropdownItem>
+          <DropdownItem divider />
+          <DropdownItem>Another Action</DropdownItem>
+        </DropdownMenu>
       </InputGroup>
       <br />
-      <InputGroup>
-        <Dropdown isOpen={splitButtonOpen} toggle={toggleSplit}>
-          <Button outline>Split Button</Button>
-          <DropdownToggle split outline />
-          <DropdownMenu>
-            <DropdownItem header>Header</DropdownItem>
-            <DropdownItem disabled>Action</DropdownItem>
-            <DropdownItem>Another Action</DropdownItem>
-            <DropdownItem divider />
-            <DropdownItem>Another Action</DropdownItem>
-          </DropdownMenu>
-        </Dropdown>
+      <InputGroup type="dropdown" isOpen={splitButtonOpen} toggle={toggleSplit}>
+        <Button outline>Split Button</Button>
+        <DropdownToggle split outline />
+        <DropdownMenu>
+          <DropdownItem header>Header</DropdownItem>
+          <DropdownItem disabled>Action</DropdownItem>
+          <DropdownItem>Another Action</DropdownItem>
+          <DropdownItem divider />
+          <DropdownItem>Another Action</DropdownItem>
+        </DropdownMenu>
         <Input placeholder="and..." />
         <Button color="secondary">I'm a button</Button>
       </InputGroup>
diff --git a/src/InputGroup.js b/src/InputGroup.js
index f0a73d7e0..6d638ad94 100644
--- a/src/InputGroup.js
+++ b/src/InputGroup.js
@@ -2,9 +2,11 @@ import React from 'react';
 import PropTypes from 'prop-types';
 import classNames from 'classnames';
 import { mapToCssModules, tagPropType } from './utils';
+import Dropdown from './Dropdown';
 
 const propTypes = {
   tag: tagPropType,
+  type: PropTypes.bool,
   size: PropTypes.string,
   className: PropTypes.string,
   cssModule: PropTypes.object,
@@ -19,6 +21,7 @@ const InputGroup = (props) => {
     className,
     cssModule,
     tag: Tag,
+    type,
     size,
     ...attributes
   } = props;
@@ -28,6 +31,10 @@ const InputGroup = (props) => {
     size ? `input-group-${size}` : null
   ), cssModule);
 
+  if (props.type === 'dropdown') {
+    return <Dropdown {...attributes} className={classes} />
+  }
+
   return (
     <Tag {...attributes} className={classes} />
   );
diff --git a/src/__tests__/InputGroup.spec.js b/src/__tests__/InputGroup.spec.js
index 281fae8ba..78b26cba7 100644
--- a/src/__tests__/InputGroup.spec.js
+++ b/src/__tests__/InputGroup.spec.js
@@ -1,6 +1,7 @@
 import React from 'react';
 import { shallow } from 'enzyme';
 import { InputGroup } from '../';
+import Dropdown from '../Dropdown';
 
 describe('InputGroup', () => {
   it('should render with "div" tag', () => {
@@ -39,4 +40,10 @@ describe('InputGroup', () => {
 
     expect(wrapper.type()).toBe('main');
   });
+
+  it('should render Dropdown when type is dropdown', () => {
+    const wrapper = shallow(<InputGroup type="dropdown" />);
+
+    expect(wrapper.type()).toBe(Dropdown);
+  });
 });