def testMultipleBIOSMultipleURLs(self):
ami = mock.Mock(text="ami-12345678")
arg = rdf_cloud.CloudMetadataRequests(requests=[
rdf_cloud.CloudMetadataRequest(
bios_version_regex=".*amazon",
instance_type="AMAZON",
url="http://169.254.169.254/latest/meta-data/ami-id",
label="amazon-ami"),
rdf_cloud.CloudMetadataRequest(
bios_version_regex="Google",
instance_type="GOOGLE",
url=self.PROJ_URL,
label="Google-project-id",
headers={"Metadata-Flavor": "Google"})
])
with mock.patch.object(
cloud.GetCloudVMMetadata,
"BIOS_VERSION_COMMAND",
new=["/bin/echo", "4.2.amazon"]):
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.assertEqual(len(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.object(cloud.GetCloudVMMetadata,
"BIOS_VERSION_COMMAND",
new="/bin/false"):
with self.assertRaises(subprocess.CalledProcessError):
self.RunAction(cloud.GetCloudVMMetadata,
arg=rdf_cloud.CloudMetadataRequest())
def testWindowsServiceQuery(self):
project = mock.Mock(text="myproject")
sc_query_output = open(
os.path.join(config.CONFIG["Test.data_dir"],
"scquery_output.txt")).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.assertEqual(len(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)
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,
"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)