fix: Flow: Add tests for Map and Set and fix base type

__tests__/flow/flow.js.flow

@@ -1,5 +1,6 @@
 // @flow
 import produce, {
+ enableMapSet,
  setAutoFreeze,
  setUseProxies,
  produce as produce2,
@@ -8,6 +9,7 @@ import produce, {
  original
 } from "../../dist/index.js.flow"
 
+enableMapSet()
 setAutoFreeze(true)
 setUseProxies(true)
 
@@ -136,7 +138,7 @@ type State = {| user: {| age: number |} |}
 }
 
 {
- const arr: Array<number> = [1];
+ const arr: Array<number> = [1]
  produce(arr, draftState => {
  const a = original(draftState)
  if (a) {
@@ -146,3 +148,41 @@ type State = {| user: {| age: number |} |}
  }
  })
 }
+
+function testMap(state: Map<number, string>) {
+ // normal
+ state = produce(state, draft => {
+ if (draft.has(1)) {
+ return
+ }
+ // $ExpectError
+ draft.set(1, 2)
+ draft.set(1, "tada")
+ })
+ // curried, no initial state
+ const f1 = produce(draft=> { draft.set(2, "ok") })
+ state = f1(state)
+ // curried, with initial state
+ const f2 = produce(draft=> { draft.set(2, "ok") }, state)
+ state = f2(state)
+ state = f2(undefined)
+}
+
+function testSet(state: Set<number>) {
+ // normal
+ state = produce(state, draft => {
+ if (draft.has(1)) {
+ return
+ }
+ // $ExpectError
+ draft.add("me")
+ draft.add(100)
+ })
+ // curried, no initial state
+ const f1 = produce(draft => { draft.add(draft.size) })
+ state = f1(state)
+ // curried, with initial state
+ const f2 = produce(draft => { draft.add(draft.size) }, state)
+ state = f2(state)
+ state = f2(undefined)
+}

src/types/index.js.flow

@@ -8,7 +8,7 @@ export interface Patch {
 
 export type PatchListener = (patches: Patch[], inversePatches: Patch[]) => void
 
-type Base = {...} | Array<any> | Map<any> | Set<any>
+type Base = {...} | Array<any>
 interface IProduce {
  /**
  * Immer takes a state, and runs a function against it.