def testAWSRequestWithMetadataResponse(self):
instanceid = mock.Mock(text="i-001d78bb6472d9d3b")
arg = rdf_cloud.CloudMetadataRequests(requests=[
rdf_cloud.CloudMetadataRequest(
bios_version_regex=".*amazon",
service_name_regex="SERVICE_NAME: AWSLiteAgent",
instance_type="AMAZON",
url="http://169.254.169.254/latest/meta-data/instance-id",
label="instance-id")
])
with mock.patch.multiple(
cloud.GetCloudVMMetadata,
LINUX_BIOS_VERSION_COMMAND=["/bin/echo", "4.2.amazon"],
WINDOWS_SERVICES_COMMAND=[
"cmd.exe", "/C", "echo SERVICE_NAME: AWSLiteAgent"
],
GetAWSMetadataToken=mock.Mock(
return_value="metadataaccesstoken"),
):
with mock.patch.object(requests, "request") as mock_requests:
mock_requests.side_effect = [instanceid]
results = self.RunAction(cloud.GetCloudVMMetadata, arg=arg)
result_responses = list(results[0].responses)
self.assertLen(result_responses, 1)
self.assertEqual(results[0].instance_type, "AMAZON")
def testMultipleBIOSMultipleURLs(self):
ami = mock.Mock(text="ami-12345678")
arg = rdf_cloud.CloudMetadataRequests(requests=[
rdf_cloud.CloudMetadataRequest(
bios_version_regex=".*amazon",
service_name_regex="SERVICE_NAME: AWSLiteAgent",
instance_type="AMAZON",
url="http://169.254.169.254/latest/meta-data/ami-id",
label="amazon-ami"),
rdf_cloud.CloudMetadataRequest(
bios_version_regex="Google",
service_name_regex="SERVICE_NAME: GCEAgent",
instance_type="GOOGLE",
url=self.PROJ_URL,
label="Google-project-id",
headers={"Metadata-Flavor": "Google"})
])
with mock.patch.multiple(
cloud.GetCloudVMMetadata,
LINUX_BIOS_VERSION_COMMAND=["/bin/echo", "4.2.amazon"],
WINDOWS_SERVICES_COMMAND=[
"cmd.exe", "/C", "echo SERVICE_NAME: AWSLiteAgent"
]):
with mock.patch.object(requests, "request") as mock_requests:
mock_requests.side_effect = [ami]
results = self.RunAction(cloud.GetCloudVMMetadata, arg=arg)
responses = list(results[0].responses)
self.assertLen(responses, 1)
self.assertEqual(results[0].instance_type, "AMAZON")
for response in responses:
if response.label == "amazon-ami":
self.assertEqual(response.text, ami.text)
else:
self.fail("Bad response.label: %s" % response.label)
def testMatchingBIOSMultipleURLs(self):
zone = mock.Mock(text="projects/123456789733/zones/us-central1-a")
project = mock.Mock(text="myproject")
arg = rdf_cloud.CloudMetadataRequests(requests=[
rdf_cloud.CloudMetadataRequest(
bios_version_regex="Google",
instance_type="GOOGLE",
url=self.ZONE_URL,
label="zone",
headers={"Metadata-Flavor": "Google"}),
rdf_cloud.CloudMetadataRequest(
bios_version_regex="Google",
instance_type="GOOGLE",
url=self.PROJ_URL,
label="project-id",
headers={"Metadata-Flavor": "Google"})
])
with mock.patch.object(cloud.GetCloudVMMetadata,
"BIOS_VERSION_COMMAND",
new=["/bin/echo", "Google"]):
with mock.patch.object(requests, "request") as mock_requests:
mock_requests.side_effect = [zone, project]
results = self.RunAction(cloud.GetCloudVMMetadata, arg=arg)
responses = list(results[0].responses)
self.assertEqual(len(responses), 2)
self.assertEqual(results[0].instance_type, "GOOGLE")
for response in responses:
if response.label == "zone":
self.assertEqual(response.text, zone.text)
elif response.label == "project-id":
self.assertEqual(response.text, project.text)
else:
self.fail("Bad response.label: %s" % response.label)
def testBIOSCommandRaises(self):
with mock.patch.multiple(
cloud.GetCloudVMMetadata,
LINUX_BIOS_VERSION_COMMAND="/bin/false",
WINDOWS_SERVICES_COMMAND=["cmd", "/C", "exit 1"]):
with self.assertRaises(subprocess.CalledProcessError):
self.RunAction(cloud.GetCloudVMMetadata,
arg=rdf_cloud.CloudMetadataRequests())
def testNonMatchingBIOS(self):
zone = mock.Mock(text="projects/123456789733/zones/us-central1-a")
arg = rdf_cloud.CloudMetadataRequests(requests=[
rdf_cloud.CloudMetadataRequest(
bios_version_regex="Google",
instance_type="GOOGLE",
url=self.ZONE_URL,
label="zone",
headers={"Metadata-Flavor": "Google"})
])
with mock.patch.object(cloud.GetCloudVMMetadata,
"LINUX_BIOS_VERSION_COMMAND",
new=["/bin/echo", "Gaagle"]):
with mock.patch.object(requests, "request") as mock_requests:
mock_requests.side_effect = [zone]
results = self.RunAction(cloud.GetCloudVMMetadata, arg=arg)
self.assertFalse(results)
def testNonUnicodeOutput(self):
responses.add(responses.GET, "https://foo.bar/quux", status=200)
request = rdf_cloud.CloudMetadataRequest()
request.bios_version_regex = "foo"
request.service_name_regex = "foo"
request.url = "https://foo.bar/quux"
args = rdf_cloud.CloudMetadataRequests()
args.requests = [request]
subprocess_check_output = lambda _: b"foo\xffbar"
with mock.patch.object(subprocess, "check_output", subprocess_check_output):
results = self.RunAction(cloud.GetCloudVMMetadata, args)
self.assertLen(results, 1)
self.assertLen(results[0].responses, 1)
self.assertEqual(results[0].responses[0].label, "https://foo.bar/quux")
def testWindowsServiceQuery(self):
project = mock.Mock(text="myproject")
scquery_output_path = os.path.join(config.CONFIG["Test.data_dir"],
"scquery_output.txt")
with io.open(scquery_output_path, "rb") as filedesc:
sc_query_output = filedesc.read()
arg = rdf_cloud.CloudMetadataRequests(requests=[
rdf_cloud.CloudMetadataRequest(
bios_version_regex=".*amazon",
instance_type="AMAZON",
service_name_regex="SERVICE_NAME: AWSLiteAgent",
url="http://169.254.169.254/latest/meta-data/ami-id",
label="amazon-ami"),
rdf_cloud.CloudMetadataRequest(
bios_version_regex="Google",
instance_type="GOOGLE",
service_name_regex="SERVICE_NAME: GCEAgent",
url=self.PROJ_URL,
label="Google-project-id",
headers={"Metadata-Flavor": "Google"})
])
with mock.patch.object(platform, "system", return_value="Windows"):
with mock.patch.object(subprocess,
"check_output",
return_value=sc_query_output):
with mock.patch.object(requests, "request") as mock_requests:
mock_requests.side_effect = [project]
results = self.RunAction(cloud.GetCloudVMMetadata, arg=arg)
responses = list(results[0].responses)
self.assertLen(responses, 1)
self.assertEqual(results[0].instance_type, "GOOGLE")
for response in responses:
if response.label == "Google-project-id":
self.assertEqual(response.text, project.text)
else:
self.fail("Bad response.label: %s" % response.label)
