def test_archive_returned(_):
c = InsightsConfig()
r = {} # rm_conf
d = DataCollector(c)
ret = d.done(c, r)
d.archive.create_tar_file.assert_called_once()
assert ret == d.archive.create_tar_file.return_value
def test_dir_returned(_):
c = InsightsConfig(output_dir='test')
r = {} # rm_conf
d = DataCollector(c)
ret = d.done(c, r)
d.archive.create_tar_file.assert_not_called()
assert ret == d.archive.archive_dir
def test_soscleaner_dir_returned(_, soscleaner):
'''
Test that SOSCleaner returns a directory when
output_dir is specified.
'''
c = InsightsConfig(obfuscate=True, output_dir='test')
r = {'keywords': ['test']}
d = DataCollector(c)
ret = d.done(c, r)
soscleaner.assert_called_once()
soscleaner.return_value.clean_report.assert_called_once()
assert ret == soscleaner.return_value.dir_path
def test_soscleaner_archive_returned(_, soscleaner):
'''
Test that SOSCleaner is enabled when obfuscate=True,
and returns an archive by default
'''
c = InsightsConfig(obfuscate=True)
r = {'keywords': ['test']}
d = DataCollector(c)
ret = d.done(c, r)
soscleaner.assert_called_once()
soscleaner.return_value.clean_report.assert_called_once()
assert ret == soscleaner.return_value.archive_path
