def test_can_reach_metadata_server(self, mock_client):
        """Verifies can_reach_metadata_server returns correctly."""
        mock_http_resp = mock.Mock(spec=http.client.HTTPResponse)
        mock_http_resp.return_value.status = http.client.OK
        mock_http_resp.return_value.getheader.return_value = (
            metadata_server._METADATA_FLAVOR_VALUE)
        mock_client.return_value.getresponse.side_effect = mock_http_resp

        self.assertTrue(metadata_server.can_reach_metadata_server())
Example #2
0
def _get_projectid_from_metadata():
    """Get the current project id from the metadata server, if reachable.

    Returns:
        str: The current project id or None if the metadata server is
            unreachable.
    """
    if metadata_server.can_reach_metadata_server():
        return metadata_server.get_project_id()
    return None
 def test_can_reach_metadata_server_timeout(self, mock_req):
     """Verifies can_reach_metadata_server returns correctly."""
     mock_req.side_effect = http.client.HTTPException()
     self.assertFalse(metadata_server.can_reach_metadata_server())