333

schangxiang@126.com

2025-09-19 18966e02fb573c7e2bb0c6426ed792b38b910940

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
'use strict';
 
const is = require('is-type-of');
const sleep = require('ko-sleep');
const mm = require('./mm');
 
const mockDatas = mm.datas;
// support generator
mm.datas = function(mod, method, datas, timeout) {
  if (!is.generatorFunction(mod[method])) {
    return mockDatas.call(mm, mod, method, datas, timeout);
  }
 
  if (timeout) {
    timeout = parseInt(timeout, 10);
  }
  timeout = timeout || 0;
  mm(mod, method, function* () {
    yield sleep(timeout);
    return datas;
  });
  return this;
};
 
const mockData = mm.data;
mm.data = function(mod, method, data, timeout) {
  if (!is.generatorFunction(mod[method])) {
    return mockData.call(mm, mod, method, data, timeout);
  }
 
  return mm.datas(mod, method, data, timeout);
};
 
const mockError = mm.error;
mm.error = function(mod, method, error, props, timeout) {
  if (!is.generatorFunction(mod[method])) {
    return mockError.call(mm, mod, method, error, props, timeout);
  }
 
  error = mm._createError(error, props);
 
  if (timeout) {
    timeout = parseInt(timeout, 10);
  }
  timeout = timeout || 0;
  mm(mod, method, function* () {
    yield sleep(timeout);
    throw error;
  });
  return this;
};
 
const mockErrorOnce = mm.errorOnce;
mm.errorOnce = function(mod, method, error, props, timeout) {
  if (!is.generatorFunction(mod[method])) {
    return mockErrorOnce.call(mm, mod, method, error, props, timeout);
  }
 
  error = mm._createError(error, props);
 
  if (timeout) {
    timeout = parseInt(timeout, 10);
  }
  timeout = timeout || 0;
  mm(mod, method, function* () {
    yield sleep(timeout);
    mm.restore();
    throw error;
  });
  return this;
};