-
Notifications
You must be signed in to change notification settings - Fork 676
/
splitter_spec.rb
119 lines (92 loc) · 3.48 KB
/
splitter_spec.rb
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
# encoding: utf-8
require File.expand_path('../spec_helper.rb', __FILE__)
module Backup
describe Splitter do
let(:model) { Model.new(:test_trigger, 'test label') }
let(:package) { model.package }
let(:splitter) { Splitter.new(model, 250, 2) }
let(:splitter_long_suffix) { Splitter.new(model, 250, 3) }
let(:s) { sequence '' }
before do
Splitter.any_instance.stubs(:utility).with(:split).returns('split')
end
# Note: BSD split will not accept a 'M' suffix for the byte size
# e.g. split -a 2 -b 250M
describe '#initialize' do
it 'sets instance variables' do
expect( splitter.package ).to be package
expect( splitter.chunk_size ).to be 250
expect( splitter.suffix_length ).to be 2
expect( splitter_long_suffix.package ).to be package
expect( splitter_long_suffix.chunk_size ).to be 250
expect( splitter_long_suffix.suffix_length ).to be 3
end
end
describe '#split_with' do
let(:given_block) { mock }
let(:block) { lambda {|arg| given_block.got(arg) } }
shared_examples 'split suffix handling' do
context 'when final package was larger than chunk_size' do
it 'updates chunk_suffixes for the package' do
suffixes = ['a' * splitter.suffix_length] * 2
suffixes.last.next!
splitter.stubs(:chunks).returns(
suffixes.map {|s| "/tmp/test_trigger.tar-#{ s }" }
)
given_block.expects(:got).in_sequence(s).with(
"split -a #{ splitter.suffix_length } -b 250m - " +
"'#{ File.join(Config.tmp_path, 'test_trigger.tar-') }'"
)
FileUtils.expects(:mv).never
splitter.split_with(&block)
expect( package.chunk_suffixes ).to eq suffixes
end
end
context 'when final package was not larger than chunk_size' do
it 'removes the suffix from the single file output by split' do
suffix = 'a' * splitter.suffix_length
splitter.stubs(:chunks).returns(["/tmp/test_trigger.tar-#{ suffix }"])
given_block.expects(:got).in_sequence(s).with(
"split -a #{ splitter.suffix_length } -b 250m - " +
"'#{ File.join(Config.tmp_path, 'test_trigger.tar-') }'"
)
FileUtils.expects(:mv).in_sequence(s).with(
File.join(Config.tmp_path, "test_trigger.tar-#{ suffix }"),
File.join(Config.tmp_path, 'test_trigger.tar')
)
splitter.split_with(&block)
expect( package.chunk_suffixes ).to eq []
end
end
end
context 'with suffix_length of 2' do
let(:splitter) { Splitter.new(model, 250, 2) }
include_examples 'split suffix handling'
end
context 'with suffix_length of 3' do
let(:splitter) { Splitter.new(model, 250, 3) }
include_examples 'split suffix handling'
end
end # describe '#split_with'
describe '#chunks' do
before do
@tmpdir = Dir.mktmpdir('backup_spec')
SandboxFileUtils.activate!(@tmpdir)
Config.send(:update, :root_path => @tmpdir)
FileUtils.mkdir_p(Config.tmp_path)
end
after do
FileUtils.rm_r(@tmpdir, :force => true, :secure => true)
end
it 'should return a sorted array of chunked file paths' do
files = [
'test_trigger.tar-aa',
'test_trigger.tar-ab',
'other_trigger.tar-aa'
].map {|name| File.join(Config.tmp_path, name) }
FileUtils.touch(files)
expect( splitter.send(:chunks) ).to eq files[0..1]
end
end
end
end