Skip to content

Commit a249eb1

Browse files
committed
Add specs for Disk#subsource and Disk#close
1 parent 23bf36b commit a249eb1

File tree

1 file changed

+20
-2
lines changed

1 file changed

+20
-2
lines changed

spec/lib/rex/image_source/disk_spec.rb

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -121,17 +121,35 @@
121121
end
122122

123123
describe "#subsource" do
124-
let(:offset) { 0 }
124+
let(:offset) { 2 }
125125
let(:len) { 512 }
126126

127127
it "returns a new Rex::ImageSource::Disk" do
128128
expect(subject.subsource(offset, len)).to be_kind_of(described_class)
129129
end
130130

131-
131+
it "returns a new Rex::ImageSource::Disk with same file" do
132+
expect(subject.subsource(offset, len).file).to eq(subject.file)
133+
end
134+
135+
it "returns a new Rex::ImageSource::Disk with provided size" do
136+
expect(subject.subsource(offset, len).size).to eq(len)
137+
end
138+
139+
it "returns a new Rex::ImageSource::Disk with file_offset added to the original" do
140+
expect(subject.subsource(offset, len).file_offset).to eq(offset + subject.file_offset)
141+
end
132142
end
133143

134144
describe "#close" do
145+
it "returns nil" do
146+
expect(subject.close).to be_nil
147+
end
135148

149+
it "closes the associated file" do
150+
expect(subject.file.closed?).to be_falsey
151+
subject.close
152+
expect(subject.file.closed?).to be_truthy
153+
end
136154
end
137155
end

0 commit comments

Comments
 (0)