From 05f93032f51c2a9e5358ef25bcbdb05c1f86fdcb Mon Sep 17 00:00:00 2001
From: "Keith J. Grant" <barkeep@keithjgrant.com>
Date: Fri, 12 Mar 2021 08:54:19 -0800
Subject: [PATCH] fix CodeEditor tests

---
 awx/ui_next/src/components/CodeEditor/CodeEditor.test.jsx | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/awx/ui_next/src/components/CodeEditor/CodeEditor.test.jsx b/awx/ui_next/src/components/CodeEditor/CodeEditor.test.jsx
index 0af1599feb..6f834d9108 100644
--- a/awx/ui_next/src/components/CodeEditor/CodeEditor.test.jsx
+++ b/awx/ui_next/src/components/CodeEditor/CodeEditor.test.jsx
@@ -1,5 +1,5 @@
 import React from 'react';
-import { mount } from 'enzyme';
+import { mountWithContexts } from '../../../testUtils/enzymeHelpers';
 import CodeEditor from './CodeEditor';
 
 describe('CodeEditor', () => {
@@ -9,7 +9,7 @@ describe('CodeEditor', () => {
 
   it('should pass value and mode through to ace editor', () => {
     const onChange = jest.fn();
-    const wrapper = mount(
+    const wrapper = mountWithContexts(
       <CodeEditor value={'---\nfoo: bar'} onChange={onChange} mode="yaml" />
     );
     const aceEditor = wrapper.find('AceEditor');
@@ -20,7 +20,7 @@ describe('CodeEditor', () => {
 
   it('should trigger onChange prop', () => {
     const onChange = jest.fn();
-    const wrapper = mount(
+    const wrapper = mountWithContexts(
       <CodeEditor value="---" onChange={onChange} mode="yaml" />
     );
     const aceEditor = wrapper.find('AceEditor');
@@ -30,7 +30,7 @@ describe('CodeEditor', () => {
 
   it('should render in read only mode', () => {
     const onChange = jest.fn();
-    const wrapper = mount(
+    const wrapper = mountWithContexts(
       <CodeEditor value="---" onChange={onChange} mode="yaml" readOnly />
     );
     const aceEditor = wrapper.find('AceEditor');