def test_file_checksum_matcher_read_failed(self, mock_match): mock_match.side_effect = IOError('No file found.') matcher = verifiers.FileChecksumMatcher('dummy/path', Mock()) with self.assertRaises(IOError): hc_assert_that(self._mock_result, matcher) self.assertTrue(mock_match.called) self.assertEqual(verifiers.MAX_RETRIES + 1, mock_match.call_count)
def test_file_checksum_matchcer_invalid_sleep_time(self): with self.assertRaises(ValueError) as cm: verifiers.FileChecksumMatcher('file_path', 'expected_checksum', 'invalid_sleep_time') self.assertEqual( cm.exception.args[0], 'Sleep seconds, if received, must be int. ' 'But received: \'invalid_sleep_time\', ' '<type \'str\'>')
def test_file_checksum_matcher_service_error(self, mock_match): mock_match.side_effect = HttpError( response={'status': '404'}, url='', content='Not Found', ) matcher = verifiers.FileChecksumMatcher('gs://dummy/path', Mock()) with self.assertRaises(HttpError): hc_assert_that(self._mock_result, matcher) self.assertTrue(mock_match.called) self.assertEqual(verifiers.MAX_RETRIES + 1, mock_match.call_count)