[FlatButton] Test to check icon color in FlatButton

Author: tintin1343

src/FlatButton/FlatButton.spec.js

@@ -4,6 +4,7 @@ import {shallow} from 'enzyme';
 import {assert} from 'chai';
 import FlatButton from './FlatButton';
 import getMuiTheme from '../styles/getMuiTheme';
+import ActionAndroid from '../svg-icons/action/android';
 
 describe('<FlatButton />', () => {
   const muiTheme = getMuiTheme();
@@ -92,6 +93,18 @@ describe('<FlatButton />', () => {
     assert.ok(icon.is({color: flatButtonTheme.primaryTextColor}));
   });
 
+  it('colors the icon with the passed color in prop', () => {
+    const wrapper = shallowWithContext(
+      <FlatButton
+        backgroundColor="#a4c639"
+        hoverColor="#8AA62F"
+        icon={<ActionAndroid color={'white'} />}
+      />
+    );
+    const icon = wrapper.children().at(0);
+    assert.equal(icon.prop('color'), 'white', 'icon should have same color as that of color prop');
+  });
+
   it('colors the button the secondary theme color', () => {
     const wrapper = shallowWithContext(
       <FlatButton secondary={true} icon={<span className="test-icon" />}>Button</FlatButton>