def test_bad_terminal(self): stdout = mock.MagicMock() stdout.write = mock.MagicMock(side_effect=[ UnicodeEncodeError('codec', u'foo', 100, 105, 'artificial UnicodeEncodeError') ] + list(range(25))) stderr = mock.MagicMock() console_tool = ConsoleTool(self.b2_api, stdout, stderr) console_tool.run_command( ['b2', 'authorize_account', 'my-account', 'good-app-key'])
def test_bad_terminal(self): stdout = mock.MagicMock() stdout.write = mock.MagicMock( side_effect=[ UnicodeEncodeError('codec', u'foo', 100, 105, 'artificial UnicodeEncodeError') ] + list(range(25)) ) stderr = mock.MagicMock() console_tool = ConsoleTool(self.b2_api, stdout, stderr) console_tool.run_command(['b2', 'authorize_account', 'my-account', 'good-app-key'])
def _run_command(self, argv, expected_stdout='', expected_stderr='', expected_status=0): """ Runs one command using the ConsoleTool, checking stdout, stderr, and the returned status code. The ConsoleTool is stateless, so we can make a new one for each call, with a fresh stdout and stderr """ expected_stdout = self._trim_leading_spaces(expected_stdout) expected_stderr = self._trim_leading_spaces(expected_stderr) stdout = six.StringIO() stderr = six.StringIO() console_tool = ConsoleTool(self.b2_api, stdout, stderr) actual_status = console_tool.run_command(['b2'] + argv) # The json module in Python 2.6 includes trailing spaces. Later version of Python don't. actual_stdout = self._trim_trailing_spaces(stdout.getvalue()) actual_stderr = self._trim_trailing_spaces(stderr.getvalue()) if expected_stdout != actual_stdout: print(repr(expected_stdout)) print(repr(actual_stdout)) if expected_stderr != actual_stderr: print(repr(expected_stderr)) print(repr(actual_stderr)) self.assertEqual(expected_stdout, actual_stdout, 'stdout') self.assertEqual(expected_stderr, actual_stderr, 'stderr') self.assertEqual(expected_status, actual_status, 'exit status code')
def _run_command(self, argv, expected_stdout='', expected_stderr='', expected_status=0): """ Runs one command using the ConsoleTool, checking stdout, stderr, and the returned status code. The ConsoleTool is stateless, so we can make a new one for each call, with a fresh stdout and stderr """ expected_stdout = self._trim_leading_spaces(expected_stdout) expected_stderr = self._trim_leading_spaces(expected_stderr) stdout = six.StringIO() stderr = six.StringIO() console_tool = ConsoleTool(self.b2_api, stdout, stderr) actual_status = console_tool.run_command(['b2'] + argv) # The json module in Python 2.6 includes trailing spaces. Later version of Python don't. actual_stdout = self._trim_trailing_spaces(stdout.getvalue()) actual_stderr = self._trim_trailing_spaces(stderr.getvalue()) if expected_stdout != actual_stdout: print(repr(expected_stdout)) print(repr(actual_stdout)) if expected_stderr != actual_stderr: print(repr(expected_stderr)) print(repr(actual_stderr)) self.assertEqual(expected_stdout, actual_stdout, 'stdout') self.assertEqual(expected_stderr, actual_stderr, 'stderr') self.assertEqual(expected_status, actual_status, 'exit status code')
def test_get_bucket_with_hidden(self): self._authorize_account() self._create_my_bucket() # Put some files into the test bucket. Unroll the loop for convenience. bucket = self.b2_api.get_bucket_by_name('my-bucket') bucket.upload(UploadSourceBytes(b'test'), 'upload1') bucket.upload(UploadSourceBytes(b'test'), 'upload2') bucket.upload(UploadSourceBytes(b'test'), 'upload3') bucket.upload(UploadSourceBytes(b'test'), 'upload4') bucket.upload(UploadSourceBytes(b'test'), 'upload5') bucket.upload(UploadSourceBytes(b'test'), 'upload6') # Hide some new files. Don't check the results here; it will be clear enough that # something has failed if the output of 'get-bucket' does not match the canon. stdout, stderr = self._get_stdouterr() console_tool = ConsoleTool(self.b2_api, stdout, stderr) console_tool.run_command(['b2', 'hide_file', 'my-bucket', 'hidden1']) console_tool.run_command(['b2', 'hide_file', 'my-bucket', 'hidden2']) console_tool.run_command(['b2', 'hide_file', 'my-bucket', 'hidden3']) console_tool.run_command(['b2', 'hide_file', 'my-bucket', 'hidden4']) # Now check the output of get-bucket against the canon. expected_stdout = ''' { "accountId": "my-account", "bucketId": "bucket_0", "bucketInfo": {}, "bucketName": "my-bucket", "bucketType": "allPublic", "corsRules": [], "fileCount": 10, "lifecycleRules": [], "revision": 1, "totalSize": 24 } ''' self._run_command(['get-bucket', '--showSize', 'my-bucket'], expected_stdout, '', 0)