diff options
| -rw-r--r-- | test/test_manifestdiff.py | 21 | ||||
| -rw-r--r-- | ubuntu/cloudimage/mfdiff/manifest.py | 3 |
2 files changed, 24 insertions, 0 deletions
diff --git a/test/test_manifestdiff.py b/test/test_manifestdiff.py index 3f6cfec..3d33731 100644 --- a/test/test_manifestdiff.py +++ b/test/test_manifestdiff.py @@ -126,3 +126,24 @@ class TestManifestDiff(object): assert added_packages == {'package2': '1.0', 'package3': '1.0'} assert removed_packages == {'package7': '1.0', 'package8': '1.0'} assert changed_packages == ['package4', 'package5'] + + def test_kernel_fixups(self, tmpdir): + package_list1 = [('linux-image-4.13.0-16-generic', '4.13.0-16.19')] + package_list2 = [('linux-image-4.13.0-25-generic', '4.13.0-25.29')] + + manifest_file1 = tmpdir.join('manifest1') + manifest_file2 = tmpdir.join('manifest2') + write_manifest_file(manifest_file1, package_list1) + write_manifest_file(manifest_file2, package_list2) + + manifest1 = Manifest(str(manifest_file1), 'bionic', 'amd64') + manifest2 = Manifest(str(manifest_file2), 'bionic', 'amd64') + diff = ManifestDiff(manifest1, manifest2) + + added_packages = diff.get_added() + removed_packages = diff.get_removed() + changed_packages = diff.get_changed() + + assert added_packages == {} + assert removed_packages == {} + assert changed_packages == ['linux-image-4.13.0-25-generic'] diff --git a/ubuntu/cloudimage/mfdiff/manifest.py b/ubuntu/cloudimage/mfdiff/manifest.py index 2ac24ba..6907ad8 100644 --- a/ubuntu/cloudimage/mfdiff/manifest.py +++ b/ubuntu/cloudimage/mfdiff/manifest.py @@ -36,6 +36,9 @@ class Manifest(object): def __getitem__(self, key): return self.dict[key] + def __setitem__(self, key, value): + self.dict[key] = value + def __delitem__(self, key): del self.dict[key] |