Remove badge element – now handled via data attribute

Author: jzempel

packages/avatars/src/Avatar.js

@@ -7,7 +7,7 @@
 
 import React from 'react';
 import PropTypes from 'prop-types';
-import { StyledAvatar, StyledBadge } from './styled';
+import { StyledAvatar } from './styled';
 
 const SIZE = {
   EXTRASMALL: 'extrasmall',
@@ -27,40 +27,28 @@ const STATUS = {
 const Avatar = ({ isSystem, size, status, children, badge, ...other }) => {
   let computedStatus = status;
 
-  if (status === STATUS.AVAILABLE && badge !== undefined) {
+  if (status === STATUS.AVAILABLE && badge) {
     computedStatus = STATUS.ACTIVE;
   }
 
-  const FormattedBadge = () => {
-    if (status === undefined || status === STATUS.AWAY) {
-      return null;
-    }
-
-    if (computedStatus === STATUS.AVAILABLE) {
-      return <StyledBadge />;
-    }
-
-    return <StyledBadge>{badge}</StyledBadge>;
-  };
-
   return (
     <StyledAvatar
       isSystem={isSystem}
       size={size}
       status={computedStatus}
+      data-badge={badge}
       aria-live="polite"
       {...other}
     >
       {children}
-      <FormattedBadge />
     </StyledAvatar>
   );
 };
 
 Avatar.propTypes = {
   /** Applies system styling */
   isSystem: PropTypes.bool,
-  badge: PropTypes.node,
+  badge: PropTypes.oneOf([PropTypes.string, PropTypes.number]),
   size: PropTypes.oneOf([SIZE.EXTRASMALL, SIZE.SMALL, SIZE.LARGE]),
   status: PropTypes.oneOf([STATUS.AVAILABLE, STATUS.AWAY]),
   children: PropTypes.node

packages/avatars/src/Avatar.spec.js

@@ -35,19 +35,15 @@ describe('Avatar', () => {
   });
 
   it('applies active styling if provided with badge', () => {
-    const { container } = render(
-      <Avatar status="available" badge={<span data-test-id="badge">2</span>} />
-    );
+    const { container } = render(<Avatar status="available" badge="2" />);
 
     expect(container.firstChild).toHaveClass('is-active');
   });
 
   it('renders badge if provided with status', () => {
-    const { getByTestId } = render(
-      <Avatar status="available" badge={<span data-test-id="badge">2</span>} />
-    );
+    const { container } = render(<Avatar status="available" badge="2" />);
 
-    expect(getByTestId('badge')).not.toBeUndefined();
+    expect(container.firstChild).toHaveAttribute('data-badge');
   });
 
   it('does not render badge if away status is provided', () => {

packages/avatars/src/styled.js

@@ -58,19 +58,6 @@ StyledAvatar.propTypes = {
   status: PropTypes.oneOf([STATUS.AVAILABLE, STATUS.ACTIVE, STATUS.AWAY])
 };
 
-const BADGE_COMPONENT_ID = 'avatars.badge';
-
-/**
- * Accepts all `<figcaption>` attributes
- */
-export const StyledBadge = styled.figcaption.attrs({
-  'data-garden-id': BADGE_COMPONENT_ID,
-  'data-garden-version': PACKAGE_VERSION,
-  className: AvatarStyles['c-avatar__badge']
-})`
-  ${props => retrieveTheme(BADGE_COMPONENT_ID, props)};
-`;
-
 const TEXT_COMPONENT_ID = 'avatars.text';
 
 /**

packages/avatars/src/styled.spec.js

@@ -7,7 +7,7 @@
 
 import React from 'react';
 import { render, renderRtl } from 'garden-test-utils';
-import { StyledAvatar, StyledBadge, StyledText } from './styled';
+import { StyledAvatar, StyledText } from './styled';
 
 describe('Styled Elements', () => {
   describe('StyledAvatar', () => {
@@ -52,14 +52,6 @@ describe('Styled Elements', () => {
     });
   });
 
-  describe('StyledBadge', () => {
-    it('renders badge styling by default', () => {
-      const { container } = render(<StyledBadge />);
-
-      expect(container.firstChild).toHaveClass(`c-avatar__badge`);
-    });
-  });
-
   describe('StyledText', () => {
     it('renders text styling by default', () => {
       const { container } = render(<StyledText />);