|
|
@@ -16,7 +16,6 @@
|
|
|
|
|
|
import sys
|
|
|
|
|
|
-
|
|
|
from nose.tools import assert_equal, assert_false, assert_true
|
|
|
|
|
|
from desktop.lib.rest.raz_http_client import RazHttpClient
|
|
|
@@ -31,26 +30,25 @@ else:
|
|
|
class TestRazHttpClient():
|
|
|
|
|
|
def test_get_file(self):
|
|
|
- with patch('desktop.lib.rest.raz_http_client.AdlsRazClient.get_url') as get_url:
|
|
|
- with patch('desktop.lib.rest.raz_http_client.HttpClient.execute') as execute:
|
|
|
+ with patch('desktop.lib.rest.raz_http_client.AdlsRazClient.get_url') as raz_get_url:
|
|
|
+ with patch('desktop.lib.rest.raz_http_client.HttpClient.execute') as http_execute:
|
|
|
|
|
|
- get_url.return_value = 'https://gethue.blob.core.windows.net/hue/data/customer.csv?sv=2014-02-14&sr=b&' + \
|
|
|
- 'sig=pJL%2FWyed41tptiwBM5ymYre4qF8wzrO05tS5MCjkutc%3D&st=2015-01-02T01%3A40%3A51Z&se=2015-01-02T02%3A00%3A51Z&sp=r'
|
|
|
- execute.return_value = 'my file'
|
|
|
+ raz_get_url.return_value = 'sv=2014-02-14&sr=b&sig=pJL%2FWyed41tptiwBM5ymYre4qF8wzrO05tS5MCjkutc%3D&st=2015-01-02T01%3A40%3A51Z&se=2015-01-02T02%3A00%3A51Z&sp=r'
|
|
|
+ http_execute.return_value = 'my file content'
|
|
|
|
|
|
client = RazHttpClient(username='test', base_url='https://gethue.blob.core.windows.net')
|
|
|
f = client.execute(http_method='GET', path='/gethue/data/customer.csv')
|
|
|
|
|
|
- assert_equal('my file', f)
|
|
|
- get_url.assert_called_with(action='GET', path='https://gethue.blob.core.windows.net/gethue/data/customer.csv', headers=None)
|
|
|
- execute.assert_called_with(
|
|
|
+ assert_equal('my file content', f)
|
|
|
+ raz_get_url.assert_called_with(action='GET', path='https://gethue.blob.core.windows.net/gethue/data/customer.csv', headers=None)
|
|
|
+ http_execute.assert_called_with(
|
|
|
http_method='GET',
|
|
|
- path='/gethue/data/customer.csv&https://gethue.blob.core.windows.net/hue/data/customer.csv?sv=2014-02-14&sr=b&sig=pJL%2FWyed41tptiwBM5ymYre4qF8wzrO05tS5MCjkutc%3D&st=2015-01-02T01%3A40%3A51Z&se=2015-01-02T02%3A00%3A51Z&sp=r',
|
|
|
+ path='/gethue/data/customer.csv&sv=2014-02-14&sr=b&sig=pJL%2FWyed41tptiwBM5ymYre4qF8wzrO05tS5MCjkutc%3D&st=2015-01-02T01%3A40%3A51Z&se=2015-01-02T02%3A00%3A51Z&sp=r',
|
|
|
params=None,
|
|
|
data=None,
|
|
|
headers=None,
|
|
|
allow_redirects=False,
|
|
|
- urlencode=True,
|
|
|
+ urlencode=False,
|
|
|
files=None,
|
|
|
stream=False,
|
|
|
clear_cookies=False,
|