1# Copyright (c) 2015 Amazon.com, Inc. or its affiliates.  All Rights Reserved
2#
3# Permission is hereby granted, free of charge, to any person obtaining a
4# copy of this software and associated documentation files (the
5# "Software"), to deal in the Software without restriction, including
6# without limitation the rights to use, copy, modify, merge, publish, dis-
7# tribute, sublicense, and/or sell copies of the Software, and to permit
8# persons to whom the Software is furnished to do so, subject to the fol-
9# lowing conditions:
10#
11# The above copyright notice and this permission notice shall be included
12# in all copies or substantial portions of the Software.
13#
14# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
15# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
16# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
17# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
18# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
20# IN THE SOFTWARE.
21#
22
23from boto.compat import json
24from boto.machinelearning.layer1 import MachineLearningConnection
25from tests.unit import AWSMockServiceTestCase
26
27
28class TestMachineLearning(AWSMockServiceTestCase):
29    connection_class = MachineLearningConnection
30
31    def test_predict(self):
32        ml_endpoint = 'mymlmodel.amazonaws.com'
33        self.set_http_response(status_code=200, body=b'')
34        self.service_connection.predict(
35            ml_model_id='foo', record={'Foo': 'bar'},
36            predict_endpoint=ml_endpoint)
37        self.assertEqual(self.actual_request.host, ml_endpoint)
38
39    def test_predict_with_scheme_in_endpoint(self):
40        ml_endpoint = 'mymlmodel.amazonaws.com'
41        self.set_http_response(status_code=200, body=b'')
42        self.service_connection.predict(
43            ml_model_id='foo', record={'Foo': 'bar'},
44            predict_endpoint='https://' + ml_endpoint)
45        self.assertEqual(self.actual_request.host, ml_endpoint)
46