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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
require "helper"
require "train/transports/mock"
require "securerandom" unless defined?(SecureRandom)
class TestFile
def initialize(string)
@string = string
end
def exist?
true
end
def size
@string.length
end
def content
@string
end
end
describe "uuid" do
def mock_platform(name, commands = {}, files = {}, plat_options = {})
Train::Platforms.list[name] = nil
mock = Train::Transports::Mock::Connection.new
commands.each do |command, data|
mock.mock_command(command, data)
end
file_objects = {}
files.each do |path, content|
file_objects[path] = TestFile.new(content)
end
mock.files = file_objects
mock.force_platform!(name, plat_options)
end
it "finds a linux uuid from chef entity_uuid" do
files = { "/var/chef/cache/data_collector_metadata.json" => '{"node_uuid":"d400073f-0920-41aa-8dd3-2ea59b18f5ce"}' }
plat = mock_platform("linux", {}, files)
_(plat.uuid).must_equal "d400073f-0920-41aa-8dd3-2ea59b18f5ce"
end
it "finds a windows uuid from chef entity_uuid" do
ENV["SYSTEMDRIVE"] = "C:"
files = { 'C:\chef\cache\data_collector_metadata.json' => '{"node_uuid":"d400073f-0920-41aa-8dd3-2ea59b18f5ce"}' }
plat = mock_platform("windows", {}, files)
_(plat.uuid).must_equal "d400073f-0920-41aa-8dd3-2ea59b18f5ce"
end
it "finds a linux uuid from /etc/chef/chef_guid" do
files = { "/etc/chef/chef_guid" => "5e430326-b5aa-56f8-975f-c3ca1c21df91" }
plat = mock_platform("linux", {}, files)
_(plat.uuid).must_equal "5e430326-b5aa-56f8-975f-c3ca1c21df91"
end
it "finds a linux uuid from /home/testuser/.chef/chef_guid" do
ENV["HOME"] = "/home/testuser"
files = { "/home/testuser/.chef/chef_guid" => "5e430326-b5aa-56f8-975f-c3ca1c21df91" }
plat = mock_platform("linux", {}, files)
_(plat.uuid).must_equal "5e430326-b5aa-56f8-975f-c3ca1c21df91"
end
it "finds a linux uuid from /etc/machine-id" do
files = { "/etc/machine-id" => "123141dsfadf" }
plat = mock_platform("linux", {}, files)
_(plat.uuid).must_equal "5e430326-b5aa-56f8-975f-c3ca1c21df91"
end
it "finds a linux uuid from /var/lib/dbus/machine-id" do
files = {
"/etc/machine-id" => "",
"/var/lib/dbus/machine-id" => "123141dsfadf",
}
plat = mock_platform("linux", {}, files)
_(plat.uuid).must_equal "5e430326-b5aa-56f8-975f-c3ca1c21df91"
end
it "finds a linux uuid from /etc/machine-id" do
files = { "/etc/machine-id" => "123141dsfadf" }
plat = mock_platform("linux", {}, files)
_(plat.uuid).must_equal "5e430326-b5aa-56f8-975f-c3ca1c21df91"
end
it "finds a windows uuid from wmic" do
commands = { "wmic csproduct get UUID /value" => "\r\r\n\r\r\nUUID=d400073f-0920-41aa-8dd3-2ea59b18f5ce\r\r\n\r\r\n\r\r\n\r\r\n", "wmic /?" => "" } # Mocking wmic command to simulate its availability
plat = mock_platform("windows", commands)
_(plat.uuid).must_equal "d400073f-0920-41aa-8dd3-2ea59b18f5ce"
end
it "finds a windows uuid from registry" do
commands = { '(Get-ItemProperty "Registry::HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Cryptography" -Name "MachineGuid")."MachineGuid"' => "d400073f-0920-41aa-8dd3-2ea59b18f5ce\r\n" }
plat = mock_platform("windows", commands)
_(plat.uuid).must_equal "d400073f-0920-41aa-8dd3-2ea59b18f5ce"
end
it 'finds a windows uuid from C:\chef\chef_guid' do
ENV["SYSTEMDRIVE"] = "C:"
files = { 'C:\chef\chef_guid' => "5e430326-b5aa-56f8-975f-c3ca1c21df91" }
plat = mock_platform("windows", {}, files)
_(plat.uuid).must_equal "5e430326-b5aa-56f8-975f-c3ca1c21df91"
end
it 'finds a windows uuid from C:\Users\test\.chef\chef_guid' do
ENV["HOMEDRIVE"] = "C:\\"
ENV["HOMEPATH"] = 'Users\test'
files = { 'C:\Users\test\.chef\chef_guid' => "5e430326-b5aa-56f8-975f-c3ca1c21df91" }
plat = mock_platform("windows", {}, files)
_(plat.uuid).must_equal "5e430326-b5aa-56f8-975f-c3ca1c21df91"
end
it "generates a uuid from a string" do
plat = mock_platform("linux")
uuid = Train::Platforms::Detect::UUID.new(plat)
_(uuid.uuid_from_string("123141dsfadf")).must_equal "5e430326-b5aa-56f8-975f-c3ca1c21df91"
end
it "finds a aws uuid" do
plat = mock_platform("aws")
plat.backend.stubs(:unique_identifier).returns("158551926027")
_(plat.uuid).must_equal "1d74ce61-ac15-5c48-9ee3-5aa8207ac37f"
end
it "finds an azure uuid" do
plat = mock_platform("azure")
plat.backend.stubs(:unique_identifier).returns("1d74ce61-ac15-5c48-9ee3-5aa8207ac37f")
_(plat.uuid).must_equal "2c2e4fa9-7287-5dee-85a3-6527face7b7b"
end
it "finds an docker uuid" do
plat = mock_platform("docker")
plat.backend.stubs(:unique_identifier).returns("343423425657")
_(plat.uuid).must_equal "a865c2dd-8877-5e06-9e58-69cbf82a8b06"
end
it "raises error if uuid not found" do
plat = mock_platform("foo")
err = _ { plat.uuid }.must_raise Train::PlatformUuidDetectionFailed
_(err.message).must_match("Could not find platform uuid! Please set a uuid_command for your platform.")
end
end
|
