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
|
require "helper"
require "train/transports/mock"
require "train/file/remote/windows"
describe Train::File::Remote::Windows do
let(:cls) { Train::File::Remote::Windows }
let(:backend) do
backend = Train::Transports::Mock.new.connection
backend.mock_os({ family: "windows" })
backend
end
describe "#md5sum" do
let(:md5_checksum) { "4ce0c733cdcf1d2f78532bbd9ce3441d" }
let(:filepath) { 'C:\Windows\explorer.exe' }
let(:ruby_md5_mock) do
checksum_mock = mock
checksum_mock.expects(:update).returns("")
checksum_mock.expects(:hexdigest).returns(md5_checksum)
checksum_mock
end
it "defaults to a Ruby based checksum if other methods fail" do
backend.mock_command("CertUtil -hashfile #{filepath} MD5", "", "", 1)
Digest::MD5.expects(:new).returns(ruby_md5_mock)
_(cls.new(backend, "/tmp/testfile").md5sum).must_equal md5_checksum
end
it "calculates the correct md5sum on the `windows` platform family" do
output = <<-EOC
MD5 hash of file C:\\Windows\\explorer.exe:\r
4c e0 c7 33 cd cf 1d 2f 78 53 2b bd 9c e3 44 1d\r
CertUtil: -hashfile command completed successfully.\r
EOC
backend.mock_command("CertUtil -hashfile #{filepath} MD5", output)
_(cls.new(backend, filepath).md5sum).must_equal md5_checksum
end
end
describe "#sha256sum" do
let(:sha256_checksum) do
"85270240a5fd51934f0627c92b2282749d071fdc9ac351b81039ced5b10f798b"
end
let(:filepath) { 'C:\Windows\explorer.exe' }
let(:ruby_sha256_mock) do
checksum_mock = mock
checksum_mock.expects(:update).returns("")
checksum_mock.expects(:hexdigest).returns(sha256_checksum)
checksum_mock
end
it "defaults to a Ruby based checksum if other methods fail" do
backend.mock_command("CertUtil -hashfile #{filepath} SHA256", "", "", 1)
Digest::SHA256.expects(:new).returns(ruby_sha256_mock)
_(cls.new(backend, "/tmp/testfile").sha256sum).must_equal sha256_checksum
end
it "calculates the correct sha256sum on the `windows` platform family" do
output = <<-EOC
SHA256 hash of file C:\\Windows\\explorer.exe:\r
85 27 02 40 a5 fd 51 93 4f 06 27 c9 2b 22 82 74 9d 07 1f dc 9a c3 51 b8 10 39 ce d5 b1 0f 79 8b\r
CertUtil: -hashfile command completed successfully.\r
EOC
backend.mock_command("CertUtil -hashfile #{filepath} SHA256", output)
_(cls.new(backend, filepath).sha256sum).must_equal sha256_checksum
end
end
end
|
