Add better selector via css

gibbok · Apr 16, 2020 · e0fe86d · e0fe86d
1 parent 9778973
commit e0fe86d
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/ColorPickerPalette.spec.tsx b/src/ColorPickerPalette.spec.tsx
@@ -17,10 +17,9 @@ describe('ColorPickerPalette', () => {
 
   const onSelectColorCB = jest.fn((x) => x);
   const wrapper = mount(<ColorPickerPalette onSelectColor={onSelectColorCB} />);
-  const canvas = wrapper.find('canvas');
-  const divs = wrapper.find('div > div');
-  const colorDiv = divs.at(7);
-  const prevColorDiv = divs.at(6);
+  const canvas = wrapper.find('[data-cy="canvas"]');
+  const colorDiv = wrapper.find('[data-cy="picked-color"]');
+  const prevColorDiv = wrapper.find('[data-cy="picked-prevColor"]');
 
   describe('should render main components', () => {
     expect(canvas).toHaveLength(1);