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
'use strict';
 
const Test = require('supertest').Test;
 
class EggTest extends Test {
  /**
   * Unexpectations:
   *
   *   .unexpectHeader('Content-Type')
   *   .unexpectHeader('Content-Type', fn)
   *
   * @return {EggTest}
   * @api public
   */
 
  unexpectHeader(name, b) {
    if (typeof b === 'function') {
      this.end(b);
    }
 
    // header
    if (typeof name === 'string') {
      this._asserts.push(this._unexpectHeader.bind(this, name));
    }
    return this;
  }
 
  /**
   * Expectations:
   *
   *   .expectHeader('Content-Type')
   *   .expectHeader('Content-Type', fn)
   *
   * @return {EggTest}
   * @api public
   */
 
  expectHeader(name, b) {
    if (typeof b === 'function') {
      this.end(b);
    }
 
    // header
    if (typeof name === 'string') {
      this._asserts.push(this._expectHeader.bind(this, name));
    }
    return this;
  }
 
  _unexpectHeader(name, res) {
    const actual = res.headers[name.toLowerCase()];
    if (actual) {
      return new Error('unexpected "' + name + '" header field, got "' + actual + '"');
    }
  }
 
  _expectHeader(name, res) {
    const actual = res.headers[name.toLowerCase()];
    if (!actual) {
      return new Error('expected "' + name + '" header field');
    }
  }
}
 
module.exports = EggTest;