compressing/test/gzip/file_stream.test.js at d688a0bbeaeb8d5fca86037fce17dd7c478413df · node-modules/compressing · GitHub

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
98
const fs = require('fs');
const os = require('os');
const path = require('path');
const uuid = require('uuid');
const { pipeline: pump } = require('stream');
const compressing = require('../..');
const assert = require('assert');

describe('test/gzip/file_stream.test.js', () => {
  it('should be a transform stream', done => {
    const sourceFile = path.join(__dirname, '..', 'fixtures', 'xx.log');
    const sourceStream = fs.createReadStream(sourceFile);
    const destFile = path.join(os.tmpdir(), uuid.v4() + '.log.gz');
    // console.log('destFile', destFile);
    const gzipStream = new compressing.gzip.FileStream();
    const destStream = fs.createWriteStream(destFile);
    pump(sourceStream, gzipStream, destStream, err => {
      assert(!err);
      assert(fs.existsSync(destFile));
      done();
    });
  });

  it('should compress according to file path', done => {
    const sourceFile = path.join(__dirname, '..', 'fixtures', 'xx.log');
    const destFile = path.join(os.tmpdir(), uuid.v4() + '.log.gz');
    // console.log('destFile', destFile);
    const gzipStream = new compressing.gzip.FileStream({ source: sourceFile });
    const destStream = fs.createWriteStream(destFile);
    pump(gzipStream, destStream, err => {
      assert(!err);
      assert(fs.existsSync(destFile));
      done();
    });
  });

  it('should compress file into Buffer', async () => {
    const sourceFile = path.join(__dirname, '..', 'fixtures', 'xx.log');
    const gzipStream = new compressing.gzip.FileStream({ source: sourceFile });
    const gzipChunks = [];
    for await (const chunk of gzipStream) {
      gzipChunks.push(chunk);
    }

    const destFile = path.join(os.tmpdir(), uuid.v4() + '.log.gz');
    await fs.promises.writeFile(destFile, Buffer.concat(gzipChunks));
    // console.log(destFile);
  });

  it('should compress buffer', done => {
    const sourceFile = path.join(__dirname, '..', 'fixtures', 'xx.log');
    const sourceBuffer = fs.readFileSync(sourceFile);
    const destFile = path.join(os.tmpdir(), uuid.v4() + '.log.gz');
    // console.log('destFile', destFile);
    const destStream = fs.createWriteStream(destFile);
    const gzipStream = new compressing.gzip.FileStream({ source: sourceBuffer });
    pump(gzipStream, destStream, err => {
      assert(!err);
      assert(fs.existsSync(destFile));
      done();
    });

  });

  it('should compress stream', done => {
    const sourceFile = path.join(__dirname, '..', 'fixtures', 'xx.log');
    const sourceStream = fs.createReadStream(sourceFile);
    const destFile = path.join(os.tmpdir(), uuid.v4() + '.log.gz');
    // console.log('destFile', destFile);
    const destStream = fs.createWriteStream(destFile);
    const gzipStream = new compressing.gzip.FileStream({ source: sourceStream });
    pump(gzipStream, destStream, err => {
      assert(!err);
      assert(fs.existsSync(destFile));
      done();
    });
  });

  it('should emit error if sourceFile does not exit', done => {
    const sourceFile = 'file-not-exist';
    const gzipStream = new compressing.gzip.FileStream({ source: sourceFile });
    gzipStream.on('error', err => {
      assert(err);
      done();
    });
  });

  it('should emit error if sourceStream emit error', done => {
    const sourceFile = 'file-not-exist';
    const sourceStream = fs.createReadStream(sourceFile);
    const gzipStream = new compressing.gzip.FileStream({ source: sourceStream });
    gzipStream.on('error', err => {
      assert(err && err.code === 'ENOENT');
      done();
    });
  });

});