| Index: tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/machinelearning/test_machinelearning.py
|
| diff --git a/tools/telemetry/third_party/gsutilz/third_party/gcs-oauth2-boto-plugin/gcs_oauth2_boto_plugin/__init__.py b/tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/machinelearning/test_machinelearning.py
|
| similarity index 50%
|
| copy from tools/telemetry/third_party/gsutilz/third_party/gcs-oauth2-boto-plugin/gcs_oauth2_boto_plugin/__init__.py
|
| copy to tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/machinelearning/test_machinelearning.py
|
| index a70e6515e01f13e404e77ef814accaa5ba5c157e..6a4d97c09c2f3a5d6e02eb8b231e9aee330a4e56 100644
|
| --- a/tools/telemetry/third_party/gsutilz/third_party/gcs-oauth2-boto-plugin/gcs_oauth2_boto_plugin/__init__.py
|
| +++ b/tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/machinelearning/test_machinelearning.py
|
| @@ -1,4 +1,4 @@
|
| -# Copyright 2014 Google Inc. All Rights Reserved.
|
| +# Copyright (c) 2015 Amazon.com, Inc. or its affiliates. All Rights Reserved
|
| #
|
| # Permission is hereby granted, free of charge, to any person obtaining a
|
| # copy of this software and associated documentation files (the
|
| @@ -18,13 +18,28 @@
|
| # WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
| # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
| # IN THE SOFTWARE.
|
| +#
|
| +
|
| +from boto.compat import json
|
| +from boto.machinelearning.layer1 import MachineLearningConnection
|
| +from tests.unit import AWSMockServiceTestCase
|
|
|
| -from __future__ import absolute_import
|
|
|
| -# Import this module so that users can simply import the top-level module. Once
|
| -# imported, boto will pick up our subclasses of AuthHandler.
|
| -from gcs_oauth2_boto_plugin import oauth2_plugin
|
| +class TestMachineLearning(AWSMockServiceTestCase):
|
| + connection_class = MachineLearningConnection
|
|
|
| -from gcs_oauth2_boto_plugin.oauth2_helper import SetFallbackClientIdAndSecret
|
| -from gcs_oauth2_boto_plugin.oauth2_helper import SetLock
|
| + def test_predict(self):
|
| + ml_endpoint = 'mymlmodel.amazonaws.com'
|
| + self.set_http_response(status_code=200, body=b'')
|
| + self.service_connection.predict(
|
| + ml_model_id='foo', record={'Foo': 'bar'},
|
| + predict_endpoint=ml_endpoint)
|
| + self.assertEqual(self.actual_request.host, ml_endpoint)
|
|
|
| + def test_predict_with_scheme_in_endpoint(self):
|
| + ml_endpoint = 'mymlmodel.amazonaws.com'
|
| + self.set_http_response(status_code=200, body=b'')
|
| + self.service_connection.predict(
|
| + ml_model_id='foo', record={'Foo': 'bar'},
|
| + predict_endpoint='https://' + ml_endpoint)
|
| + self.assertEqual(self.actual_request.host, ml_endpoint)
|
|
|