def test_no_removals_for_no_repo_entitlements(self, m_os_unlink):
m_entitlements = mock.Mock()
m_entitlements.ENTITLEMENT_CLASSES = [RepoTestEntitlement]
clean_apt_files(_entitlements=m_entitlements)
assert 0 == m_os_unlink.call_count
def test_files_for_all_series_removed(self, mock_apt_entitlement, tmpdir):
m_entitlements = mock.Mock()
m_entitlements.ENTITLEMENT_CLASSES = [mock_apt_entitlement]
clean_apt_files(_entitlements=m_entitlements)
if mock_apt_entitlement.is_repo:
assert [] == tmpdir.listdir()
else:
assert sorted(
[tmpdir.join("source-test_ent"), tmpdir.join("pref-test_ent")]
) == sorted(tmpdir.listdir())
def test_other_files_not_removed(self, mock_apt_entitlement, tmpdir):
other_filename = "other_file-acidic"
tmpdir.join(other_filename).ensure()
m_entitlements = mock.Mock()
m_entitlements.ENTITLEMENT_CLASSES = [mock_apt_entitlement]
clean_apt_files(_entitlements=m_entitlements)
if mock_apt_entitlement.is_repo:
assert [tmpdir.join(other_filename)] == tmpdir.listdir()
else:
assert sorted([
tmpdir.join("source-test_ent"),
tmpdir.join("pref-test_ent"),
tmpdir.join(other_filename),
]) == sorted(tmpdir.listdir())