diff --git a/app/javascript/mastodon/features/ui/components/__tests__/column-test.js b/app/javascript/mastodon/features/ui/components/__tests__/column-test.js
index 1e5e1d8dc..11cc1b6e8 100644
--- a/app/javascript/mastodon/features/ui/components/__tests__/column-test.js
+++ b/app/javascript/mastodon/features/ui/components/__tests__/column-test.js
@@ -21,13 +21,13 @@ describe('', () => {
);
- wrapper.find(ColumnHeader).simulate('click');
+ wrapper.find(ColumnHeader).find('button').simulate('click');
expect(global.requestAnimationFrame.mock.calls.length).toEqual(1);
});
it('does not try to scroll if there is no scrollable content', () => {
const wrapper = mount();
- wrapper.find(ColumnHeader).simulate('click');
+ wrapper.find(ColumnHeader).find('button').simulate('click');
expect(global.requestAnimationFrame.mock.calls.length).toEqual(0);
});
});
diff --git a/app/javascript/mastodon/features/ui/components/column_header.js b/app/javascript/mastodon/features/ui/components/column_header.js
index fdf9aab1a..e8bdd8054 100644
--- a/app/javascript/mastodon/features/ui/components/column_header.js
+++ b/app/javascript/mastodon/features/ui/components/column_header.js
@@ -1,5 +1,6 @@
import React from 'react';
import PropTypes from 'prop-types';
+import classNames from 'classnames';
export default class ColumnHeader extends React.PureComponent {
@@ -16,19 +17,20 @@ export default class ColumnHeader extends React.PureComponent {
}
render () {
- const { type, active, columnHeaderId } = this.props;
+ const { icon, type, active, columnHeaderId } = this.props;
+ let iconElement = '';
- let icon = '';
-
- if (this.props.icon) {
- icon = ;
+ if (icon) {
+ iconElement = ;
}
return (
-
- {icon}
- {type}
-
+
+
+
);
}