chore: Add locale

iamkun · Mar 7, 2019 · 37ab09f · 37ab09f
1 parent 08be44a
commit 37ab09f
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/src/plugin/badMutable/index.js b/src/plugin/badMutable/index.js
@@ -22,5 +22,12 @@ export default (o, c) => { // locale needed later
     this.init()
     return this
   }
+
+  const oldLocale = proto.locale
+  proto.locale = function (preset, object) {
+    if (!preset) return this.$L
+    this.$L = oldLocale.bind(this)(preset, object).$L
+    return this
+  }
 }
 
diff --git a/test/plugin/badMutable.test.js b/test/plugin/badMutable.test.js
@@ -2,6 +2,7 @@ import MockDate from 'mockdate'
 import moment from 'moment'
 import dayjs from '../../src'
 import badMutable from '../../src/plugin/badMutable'
+import '../../src/locale/zh-cn'
 
 dayjs.extend(badMutable)
 
@@ -148,3 +149,15 @@ it('daysInMonth', () => {
   expect(d.daysInMonth()).toBe(m.daysInMonth())
   expect(d.format()).toBe(m.format())
 })
+
+it('Locale', () => {
+  const d = dayjs()
+  const m = moment()
+  const format = 'MMMM'
+  expect(d.locale()).toBe(m.locale())
+  expect(d.format(format)).toBe(m.format(format))
+  d.locale('zh-cn')
+  m.locale('zh-cn')
+  expect(d.locale()).toBe(m.locale())
+  expect(d.format(format)).toBe(m.format(format))
+})