def test_mark_invalid_shards(self, mock_rm):
merge_results.mark_invalid_shards(['123abc'], [
'/tmp/123abc/dummy.json', '/tmp/123abc/dummy2.json',
'/tmp/1234abc/dummy.json'
])
self.assertEqual([
mock.call('/tmp/123abc/dummy.json'),
mock.call('/tmp/123abc/dummy2.json')
], mock_rm.call_args_list)
def test_mark_invalid_shards_bad_summary(self, _mo):
# Should not raise an exception.
raised_exception = False
try:
merge_results.mark_invalid_shards(['1234567890abcdeff'],
'dummy.json', 'o.json')
except: # pylint: disable=bare-except
raised_exception = True
self.assertFalse(raised_exception, 'unexpected exception')
def test_mark_invalid_shards(self, mo):
mock_result = mock.mock_open()
mock_write = mock.MagicMock()
mock_result.return_value.write = mock_write
mo.side_effect = (
mo.return_value,
mock.mock_open(read_data='{}').return_value,
mock_result.return_value,
)
merge_results.mark_invalid_shards(['1234567890abcdeff'], 'dummy.json',
'o.json')
written = json.loads(''.join(c[0][0] for c in mock_write.call_args_list))
self.assertIn('missing_shards', written)
self.assertEqual(written['missing_shards'], [0])
