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])