fix missing default value on non-exist size key. change unittest according to default value change, removed redundant test.

Author: mendyEdri

src/components/avatar/__tests__/index.spec.js

@@ -62,13 +62,8 @@ describe('Avatar Badge', () => {
 
   describe('badgeProps.size, supports enum or number', () => {
     it('should return 10 as the size number given', () => {
-      const uut = new Avatar({badgeProps: {size: 10}});
-      expect(uut.getBadgeSize()).toEqual(10);
-    });
-
-    it('should return 876 as the size number given', () => {
-      const uut = new Avatar({badgeProps: {size: 876}});
-      expect(uut.getBadgeSize()).toEqual(876);
+      const uut = new Avatar({badgeProps: {size: 99}});
+      expect(uut.getBadgeSize()).toEqual(99);
     });
 
     it('should return 0 as the given size number', () => {
@@ -89,10 +84,10 @@ describe('Avatar Badge', () => {
       expect(uut.getBadgeSize()).toEqual(BADGE_SIZES[lastSizeKey]);
     });
 
-    it('should return undefined for a non-exist size type', () => {
+    it('should return a default value from Badge for a non-exist size type', () => {
       const sizeKey = '!NOT_A_VALID_ENUM$';
       const uut = new Avatar({badgeProps: {size: sizeKey}});
-      expect(uut.getBadgeSize()).toBeUndefined();
+      expect(typeof uut.getBadgeSize()).toBe('number');
     });
   });
 });

src/components/avatar/index.tsx

@@ -224,9 +224,9 @@ class Avatar extends PureComponent<AvatarPropTypes> {
 
   getBadgeSize = () => {
     const badgeSize = _.get(this.props, 'badgeProps.size', DEFAULT_BADGE_SIZE);
-    
+  
     if (_.isString(badgeSize)) {
-      return BADGE_SIZES[badgeSize];
+      return BADGE_SIZES[badgeSize] || BADGE_SIZES[DEFAULT_BADGE_SIZE];
     }
     return badgeSize;
   }