feat(MdBadge): add prop validation

vuematerial · Apr 10, 2018 · a53aff6 · a53aff6
1 parent e136096
commit a53aff6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/src/components/MdBadge/MdBadge.test.js b/src/components/MdBadge/MdBadge.test.js
@@ -8,10 +8,3 @@ test('should render the badge', async () => {
   expect(wrapper.hasClass('md-badge')).toBe(true)
   expect(wrapper.text()).toBe('Lorem ipsum')
 })
-
-test('should render the theme class', async () => {
-  const template = '<md-badge md-theme="alt">Lorem ipsum</md-badge>'
-  const wrapper = await mountTemplate(MdBadge, template)
-
-  expect(wrapper.hasClass('md-theme-alt')).toBe(true)
-})
diff --git a/src/components/MdBadge/MdBadge.vue b/src/components/MdBadge/MdBadge.vue
@@ -14,6 +14,7 @@
 
 <script>
   import MdComponent from 'core/MdComponent'
+  import MdPropValidator from 'core/utils/MdPropValidator'
 
   import MdBadgeStandalone from './MdBadgeStandalone'
 
@@ -26,7 +27,11 @@
       mdContent: [String, Number],
       mdPosition: {
         type: String,
-        default: 'top'
+        default: 'top',
+        ...MdPropValidator('md-position', [
+          'top',
+          'bottom'
+        ])
       },
       mdDense: Boolean
     },