@@ -157,8 +157,9 @@ class ApplicationCreatorTests(unittest.TestCase):
157
157
@patch ('application_creator.os' )
158
158
@patch ('application_creator.tarfile' )
159
159
@patch ('shutil.copy' )
160
+ @patch ('platform.dist' )
160
161
# pylint: disable=unused-argument
161
- def test_create_application(self, copy_mock, tar_mock, os_mock, shutil_mock, spur_ssh,
162
+ def test_create_application (self , dist_mock , copy_mock , tar_mock , os_mock , shutil_mock , spur_ssh ,
162
163
hdfs_client_mock , post_mock , put_mock , exec_ssh_mock ,
163
164
os_sys_mock , dt_mock , hive_mock , hbase_mock ):
164
165
dt_mock .utcnow .return_value = (datetime (2013 , 01 , 01 ))
@@ -170,6 +171,7 @@ def json(self):
170
171
return {'id' : 'someid' }
171
172
172
173
post_mock .return_value = Resp ()
174
+ dist_mock .return_value = 'ubuntu'
173
175
with patch ("__builtin__.open" , mock_open (read_data = "[]" )):
174
176
creator = ApplicationCreator (self .config , self .environment , self .service )
175
177
creator .create_application ('abcd' , self .package_metadata , 'aname' , self .property_overrides )
0 commit comments