summaryrefslogtreecommitdiff
path: root/tests/unit/mturk/test_connection.py
blob: 05f6be259ec1279348476b87da12bed32d815c7d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# -*- coding: utf-8 -*-

from tests.unit import AWSMockServiceTestCase

from boto.mturk.connection import MTurkConnection


GET_FILE_UPLOAD_URL = b"""
<GetFileUploadURLResult>
  <Request>
    <IsValid>True</IsValid>
  </Request>
  <FileUploadURL>http://s3.amazonaws.com/myawsbucket/puppy.jpg</FileUploadURL>
</GetFileUploadURLResult>"""


class TestMTurkConnection(AWSMockServiceTestCase):
    connection_class = MTurkConnection

    def setUp(self):
        super(TestMTurkConnection, self).setUp()

    def test_get_file_upload_url_success(self):
        self.set_http_response(status_code=200, body=GET_FILE_UPLOAD_URL)
        rset = self.service_connection.get_file_upload_url('aid', 'qid')
        self.assertEquals(len(rset), 1)
        self.assertEquals(rset[0].FileUploadURL,
                          'http://s3.amazonaws.com/myawsbucket/puppy.jpg')