From f9d615fdee7e8f55bbdd1e33b9bf8b415a2f3121 Mon Sep 17 00:00:00 2001
From: kialam <digitalanime@gmail.com>
Date: Tue, 19 Feb 2019 11:28:17 -0500
Subject: [PATCH] Fix unit tests.

---
 __tests__/components/AnsibleSelect.test.jsx   | 13 -------
 .../screens/OrganizationAdd.test.jsx          | 34 +++++++++++++++++++
 2 files changed, 34 insertions(+), 13 deletions(-)

diff --git a/__tests__/components/AnsibleSelect.test.jsx b/__tests__/components/AnsibleSelect.test.jsx
index 8f54b0d522..c80207cc9d 100644
--- a/__tests__/components/AnsibleSelect.test.jsx
+++ b/__tests__/components/AnsibleSelect.test.jsx
@@ -33,19 +33,6 @@ describe('<AnsibleSelect />', () => {
     expect(spy).toHaveBeenCalled();
   });
 
-  test('content not rendered when data property is falsey', () => {
-    const wrapper = mount(
-      <AnsibleSelect
-        value="foo"
-        name="bar"
-        onChange={() => { }}
-        label={label}
-        data={null}
-      />
-    );
-    expect(wrapper.find('FormGroup')).toHaveLength(0);
-    expect(wrapper.find('Select')).toHaveLength(0);
-  });
   test('Returns correct select options if defaultSelected props is passed', () => {
     const wrapper = mount(
       <AnsibleSelect
diff --git a/__tests__/pages/Organizations/screens/OrganizationAdd.test.jsx b/__tests__/pages/Organizations/screens/OrganizationAdd.test.jsx
index 6a41d1d1ff..c9ea5414b4 100644
--- a/__tests__/pages/Organizations/screens/OrganizationAdd.test.jsx
+++ b/__tests__/pages/Organizations/screens/OrganizationAdd.test.jsx
@@ -2,6 +2,7 @@ import React from 'react';
 import { mount } from 'enzyme';
 import { MemoryRouter } from 'react-router-dom';
 import { I18nProvider } from '@lingui/react';
+import { ConfigContext } from '../../../../src/context';
 import OrganizationAdd from '../../../../src/pages/Organizations/screens/OrganizationAdd';
 
 describe('<OrganizationAdd />', () => {
@@ -157,4 +158,37 @@ describe('<OrganizationAdd />', () => {
     });
     expect(createInstanceGroupsFn).toHaveBeenCalledWith('/api/v2/organizations/1/instance_groups', 1);
   });
+
+  test('AnsibleSelect component renders if there are virtual environments', () => {
+    const config = {
+      custom_virtualenvs: ['foo', 'bar'],
+    };
+    const wrapper = mount(
+      <MemoryRouter>
+        <I18nProvider>
+          <ConfigContext.Provider value={config}>
+            <OrganizationAdd api={{}} />
+          </ConfigContext.Provider>
+        </I18nProvider>
+      </MemoryRouter>
+    ).find('OrganizationAdd').find('AnsibleSelect');
+    expect(wrapper.find('Select')).toHaveLength(1);
+    expect(wrapper.find('SelectOption')).toHaveLength(2);
+  });
+
+  test('AnsibleSelect component does not render if there are 0 virtual environments', () => {
+    const config = {
+      custom_virtualenvs: [],
+    };
+    const wrapper = mount(
+      <MemoryRouter>
+        <I18nProvider>
+          <ConfigContext.Provider value={config}>
+            <OrganizationAdd api={{}} />
+          </ConfigContext.Provider>
+        </I18nProvider>
+      </MemoryRouter>
+    ).find('OrganizationAdd').find('AnsibleSelect');
+    expect(wrapper.find('Select')).toHaveLength(0);
+  });
 });