class TestPackage(unittest.TestCase):
-
def test_upload_fail(self):
mock = Mock()
- mock.post_cmd.return_value = 'foo'
- self.assertRaises(ClientException,
- package.Package(upload_http=mock).upload, 'bar')
+ mock.post_cmd.return_value = "foo"
+ self.assertRaises(
+ ClientException, package.Package(upload_http=mock).upload, "bar"
+ )
mock.post_cmd.return_value = None
- self.assertRaises(ClientException,
- package.Package(upload_http=mock).upload, 'bar')
+ self.assertRaises(
+ ClientException, package.Package(upload_http=mock).upload, "bar"
+ )
def test_wait_for_upload_bad_file(self):
mock = Mock()
- mock.post_cmd.return_value = 'foo'
- self.assertRaises(IOError,
- package.Package(upload_http=mock).wait_for_upload,
- 'invalidfile')
+ mock.post_cmd.return_value = "foo"
+ self.assertRaises(
+ IOError, package.Package(upload_http=mock).wait_for_upload, "invalidfile"
+ )