Blame view

node_modules/fs-write-stream-atomic/test/basic.js 2.68 KB
2a09d1a4   liuqimichale   添加宜春 天水 宣化
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
  var fs = require('graceful-fs')
  var test = require('tap').test
  var path = require('path')
  var writeStream = require('../index.js')
  
  var rename = fs.rename
  fs.rename = function (from, to, cb) {
    setTimeout(function () {
      rename(from, to, cb)
    }, 100)
  }
  
  test('basic', function (t) {
    // open 10 write streams to the same file.
    // then write to each of them, and to the target
    // and verify at the end that each of them does their thing
    var target = path.resolve(__dirname, 'test.txt')
    var n = 10
  
    // We run all of our assertions twice:
    //   once for finish, once for close
    // There are 6 assertions, two fixed, plus 4 lines in the file.
    t.plan(n * 2 * 6)
  
    var streams = []
    for (var i = 0; i < n; i++) {
      var s = writeStream(target)
      s.on('finish', verifier('finish', i))
      s.on('close', verifier('close', i))
      streams.push(s)
    }
  
    function verifier (ev, num) {
      return function () {
        if (ev === 'close') {
          t.equal(this.__emittedFinish, true, num + '. closed only after finish')
        } else {
          this.__emittedFinish = true
          t.equal(ev, 'finish', num + '. finished')
        }
  
        // make sure that one of the atomic streams won.
        var res = fs.readFileSync(target, 'utf8')
        var lines = res.trim().split(/\n/)
        lines.forEach(function (line, lineno) {
          var first = lines[0].match(/\d+$/)[0]
          var cur = line.match(/\d+$/)[0]
          t.equal(cur, first, num + '. line ' + lineno + ' matches')
        })
  
        var resExpr = /^first write \d+\nsecond write \d+\nthird write \d+\nfinal write \d+\n$/
        t.similar(res, resExpr, num + '. content matches')
      }
    }
  
    // now write something to each stream.
    streams.forEach(function (stream, i) {
      stream.write('first write ' + i + '\n')
    })
  
    // wait a sec for those writes to go out.
    setTimeout(function () {
      // write something else to the target.
      fs.writeFileSync(target, 'brutality!\n')
  
      // write some more stuff.
      streams.forEach(function (stream, i) {
        stream.write('second write ' + i + '\n')
      })
  
      setTimeout(function () {
        // Oops!  Deleted the file!
        fs.unlinkSync(target)
  
        // write some more stuff.
        streams.forEach(function (stream, i) {
          stream.write('third write ' + i + '\n')
        })
  
        setTimeout(function () {
          fs.writeFileSync(target, 'brutality TWO!\n')
          streams.forEach(function (stream, i) {
            stream.end('final write ' + i + '\n')
          })
        }, 50)
      }, 50)
    }, 50)
  })
  
  test('cleanup', function (t) {
    fs.readdirSync(__dirname).filter(function (f) {
      return f.match(/^test.txt/)
    }).forEach(function (file) {
      fs.unlinkSync(path.resolve(__dirname, file))
    })
    t.end()
  })