feat(Pagination): support device

STLighter · Aug 5, 2019 · 52054e1 · 52054e1
1 parent 67f0d49
commit 52054e1
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/src/pagination/pagination.jsx b/src/pagination/pagination.jsx
@@ -20,6 +20,7 @@ class Pagination extends Component {
         prefix: PropTypes.string,
         pure: PropTypes.bool,
         rtl: PropTypes.bool,
+        device: PropTypes.oneOf(['desktop', 'tablet', 'phone']),
         className: PropTypes.string,
         /**
          * 自定义国际化文案对象
@@ -579,7 +580,8 @@ class Pagination extends Component {
             prefix,
             pure,
             rtl,
-            type,
+            device,
+            type: paginationType,
             size,
             shape,
             className,
@@ -612,6 +614,12 @@ class Pagination extends Component {
         const sizeSelector = this.renderPageSizeSelector();
         const isStart = pageSizePosition === 'start';
 
+        let type = paginationType;
+
+        if (device === 'phone' && type === 'normal') {
+            type = 'simple';
+        }
+
         const classes = cx({
             [`${prefix}pagination`]: true,
             [`${prefix}${size}`]: size,

diff --git a/test/pagination/index-spec.js b/test/pagination/index-spec.js
@@ -680,4 +680,13 @@ describe('Pagination', () => {
                 )
         );
     });
+
+    it('should support device transfer type to simple', () => {
+        wrapper.setProps({
+            device: 'phone',
+            type: 'normal'
+        });
+
+        assert(wrapper.find('.next-simple').length > 0);
+    })
 });