From dfb2dd2500d8c4a3011f629f95fd11ac008039ea Mon Sep 17 00:00:00 2001
From: Andrej Ocenas <mr.ocenas@gmail.com>
Date: Tue, 12 Mar 2019 23:23:57 +0100
Subject: [PATCH] Add simple test for the ColorPicker

---
 .../ColorPicker/ColorPicker.test.tsx          | 23 +++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 packages/grafana-ui/src/components/ColorPicker/ColorPicker.test.tsx

diff --git a/packages/grafana-ui/src/components/ColorPicker/ColorPicker.test.tsx b/packages/grafana-ui/src/components/ColorPicker/ColorPicker.test.tsx
new file mode 100644
index 00000000000..8bf41f307f3
--- /dev/null
+++ b/packages/grafana-ui/src/components/ColorPicker/ColorPicker.test.tsx
@@ -0,0 +1,23 @@
+import React from 'react';
+import renderer from 'react-test-renderer';
+import { ColorPicker } from './ColorPicker';
+import { ColorPickerTrigger } from './ColorPickerTrigger';
+
+describe('ColorPicker', () => {
+  it('renders ColorPickerTrigger component by default', () => {
+    expect(
+      renderer.create(<ColorPicker color="#EAB839" onChange={() => {}} />).root.findByType(ColorPickerTrigger)
+    ).toBeTruthy();
+  });
+
+  it('renders custom trigger when supplied', () => {
+    const div = renderer
+      .create(
+        <ColorPicker color="#EAB839" onChange={() => {}}>
+          {() => <div>Custom trigger</div>}
+        </ColorPicker>
+      )
+      .root.findByType('div');
+    expect(div.children[0]).toBe('Custom trigger');
+  });
+});