def test_save_processed(self):
config = self._get_config()
config.tarfile_module = tarfile
config.gzip_module = gzip
tar_store = TarFileWritingCrashStore(config)
processed_crash = {
'crash_id': '091204bd-87c0-42ba-8f58-554492141212',
'payload': 'nothing to see here',
'some_date': datetime(1960, 5, 4, 15, 10)
}
processed_crash_as_str = json.dumps(
processed_crash,
default=dates_to_strings_for_json
)
# the call to be tested
tar_store.save_processed(processed_crash)
# this try to get it back
tar_store.close()
result_tar_fp = tarfile.TarFile(join(TEMP_DIR, 'a_tarball_name.tar'))
result_gzip_fp = gzip.GzipFile(
fileobj=result_tar_fp.extractfile(
'091204bd-87c0-42ba-8f58-554492141212.jsonz'
)
)
reconstituted_processed_crash_as_str = result_gzip_fp.read().strip()
eq_(
processed_crash_as_str,
reconstituted_processed_crash_as_str
)
def test_get_processed(self):
config = self._get_config()
config.tarfile_module = tarfile
config.gzip_module = gzip
writing_tar_store = TarFileWritingCrashStore(config)
processed_crash_1 = {
'crash_id': '091204bd-87c0-42ba-8f58-554492141212',
'payload': 'nothing to see here',
'some_date': "1960-05-04T15:10:00"
}
processed_crash_2 = {
'crash_id': '666884bd-87c0-42ba-8f58-554492141212',
'payload': 'nothing to see here, ether',
'some_date': "1974-05-04T15:10:00"
}
writing_tar_store.save_processed(processed_crash_1)
writing_tar_store.save_processed(processed_crash_2)
writing_tar_store.close()
reading_tar_store = TarFileSequentialReadingCrashStore(config)
reconstituted_processed_crash_1 = reading_tar_store.get_processed(
'it does not matter what is said here'
)
eq_(processed_crash_1, reconstituted_processed_crash_1)
reconstituted_processed_crash_2 = reading_tar_store.get_processed(
'it does not matter what is said here'
)
eq_(processed_crash_2, reconstituted_processed_crash_2)
def test_init(self):
config = self._get_config()
# the call to be tested
tar_store = TarFileWritingCrashStore(config)
# this is what should have happened
ok_(not hasattr(tar_store, 'tar_file'))
ok_(isinstance(tar_store.tarfile_module, Mock))
ok_(isinstance(tar_store.gzip_module, Mock))
tar_store.tarfile_module.open.assert_called_once_with(
config.tarball_name, 'w')
def test_get_processed(self):
config = self._get_config()
config.tarfile_module = tarfile
config.gzip_module = gzip
writing_tar_store = TarFileWritingCrashStore(config)
processed_crash_1 = {
'crash_id': '091204bd-87c0-42ba-8f58-554492141212',
'payload': 'nothing to see here',
'some_date': "1960-05-04T15:10:00"
}
processed_crash_2 = {
'crash_id': '666884bd-87c0-42ba-8f58-554492141212',
'payload': 'nothing to see here, ether',
'some_date': "1974-05-04T15:10:00"
}
writing_tar_store.save_processed(processed_crash_1)
writing_tar_store.save_processed(processed_crash_2)
writing_tar_store.close()
reading_tar_store = TarFileSequentialReadingCrashStore(config)
reconstituted_processed_crash_1 = reading_tar_store.get_processed(
'it does not matter what is said here')
eq_(processed_crash_1, reconstituted_processed_crash_1)
reconstituted_processed_crash_2 = reading_tar_store.get_processed(
'it does not matter what is said here')
eq_(processed_crash_2, reconstituted_processed_crash_2)
def test_save_processed(self):
config = self._get_config()
config.tarfile_module = tarfile
config.gzip_module = gzip
tar_store = TarFileWritingCrashStore(config)
processed_crash = {
'crash_id': '091204bd-87c0-42ba-8f58-554492141212',
'payload': 'nothing to see here',
'some_date': datetime(1960, 5, 4, 15, 10)
}
processed_crash_as_str = json.dumps(processed_crash,
default=dates_to_strings_for_json)
# the call to be tested
tar_store.save_processed(processed_crash)
# this try to get it back
tar_store.close()
result_tar_fp = tarfile.TarFile(join(TEMP_DIR, 'a_tarball_name.tar'))
result_gzip_fp = gzip.GzipFile(fileobj=result_tar_fp.extractfile(
'091204bd-87c0-42ba-8f58-554492141212.jsonz'))
reconstituted_processed_crash_as_str = result_gzip_fp.read().strip()
eq_(processed_crash_as_str, reconstituted_processed_crash_as_str)
