Source code for testing.unit.test_manifest

# -*- Mode:Python; indent-tabs-mode:nil; tab-width:4; encoding:utf-8 -*-
#
# Copyright 2002 Ben Escoto <ben@emerose.org>
# Copyright 2007 Kenneth Loafman <kenneth@loafman.com>
#
# This file is part of duplicity.
#
# Duplicity is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation; either version 2 of the License, or (at your
# option) any later version.
#
# Duplicity is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with duplicity; if not, write to the Free Software Foundation,
# Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA


import re
import unittest
from unittest.mock import patch

from duplicity import config
from duplicity import manifest
from duplicity import path

from . import UnitTestCase


[docs]class VolumeInfoTest(UnitTestCase): """Test VolumeInfo"""
[docs] def test_basic(self): """Basic VolumeInfoTest""" vi = manifest.VolumeInfo() vi.set_info(3, (b"hello", b"there"), None, (), None) vi.set_hash("MD5", "aoseutaohe") s = vi.to_string() assert isinstance(s, (bytes, str)) # print "---------\n%s\n---------" % s vi2 = manifest.VolumeInfo() vi2.from_string(s) assert vi == vi2
[docs] def test_special(self): """Test VolumeInfo with special characters""" vi = manifest.VolumeInfo() vi.set_info( 3234, ( b"\n eu \x233", b"heuo", b"\xd8\xab\xb1Wb\xae\xc5]\x8a\xbb\x15v*\xf4\x0f!\xf9>\xe2Y\x86\xbb\xab\xdbp\xb0\x84\x13k\x1d\xc2\xf1\xf5e\xa5U\x82\x9aUV\xa0\xf4\xdf4\xba\xfdX\x03\x82\x07s\xce\x9e\x8b\xb34\x04\x9f\x17 \xf4\x8f\xa6\xfa\x97\xab\xd8\xac\xda\x85\xdcKvC\xfa#\x94\x92\x9e\xc9\xb7\xc3_\x0f\x84g\x9aB\x11<=^\xdbM\x13\x96c\x8b\xa7|*\"\\'^$@#!(){}?+ ~` ", # noqa ), None, (b"\n",), None, ) s = vi.to_string() assert isinstance(s, (str, bytes)) # print "---------\n%s\n---------" % s vi2 = manifest.VolumeInfo() vi2.from_string(s) assert vi == vi2
[docs] def test_contains(self): """Test to see if contains() works""" vi = manifest.VolumeInfo() vi.set_info(1, ("1", "2"), None, ("1", "3"), None) assert vi.contains(("1",), recursive=1) assert not vi.contains(("1",), recursive=0) vi2 = manifest.VolumeInfo() vi2.set_info(1, ("A",), None, ("Z",), None) assert vi2.contains(("M",), recursive=1) assert vi2.contains(("M",), recursive=0) vi3 = manifest.VolumeInfo() vi3.set_info(1, ("A",), None, ("Z",), None) assert not vi3.contains(("3",), recursive=1) assert not vi3.contains(("3",), recursive=0)
[docs]class ManifestTest(UnitTestCase): """Test Manifest class"""
[docs] def setUp(self): UnitTestCase.setUp(self) self.old_files_changed = config.file_changed config.file_changed = "testing"
[docs] def tearDown(self): config.file_changed = self.old_files_changed
[docs] def test_basic(self): vi1 = manifest.VolumeInfo() vi1.set_info(3, (b"hello",), None, (), None) vi2 = manifest.VolumeInfo() vi2.set_info(4, (b"goodbye", b"there"), None, (b"aoeusht",), None) vi3 = manifest.VolumeInfo() vi3.set_info(34, (), None, (), None) m = manifest.Manifest() for vi in [vi1, vi2, vi3]: m.add_volume_info(vi) self.set_config("local_path", path.Path("Foobar")) m.set_dirinfo() m.set_files_changed_info([]) s = m.to_string() assert s.lower().startswith(b"hostname") assert s.endswith(b"\n") m2 = manifest.Manifest().from_string(s) assert m == m2
[docs] def test_corrupt_filelist(self): vi1 = manifest.VolumeInfo() vi1.set_info(3, (b"hello",), None, (), None) vi2 = manifest.VolumeInfo() vi2.set_info(4, (b"goodbye", b"there"), None, (b"aoeusht",), None) vi3 = manifest.VolumeInfo() vi3.set_info(34, (), None, (), None) m = manifest.Manifest() for vi in [vi1, vi2, vi3]: m.add_volume_info(vi) self.set_config("local_path", path.Path("Foobar")) m.set_dirinfo() m.set_files_changed_info( [ (b"one", b"new"), (b"two", b"changed"), (b"three", b"new"), ] ) # build manifest string s = m.to_string() # make filecount higher than files in list s2 = re.sub(b"Filelist 3", b"Filelist 5", s) m2 = manifest.Manifest().from_string(s2) assert hasattr(m2, "corrupt_filelist")
[docs] def test_hostname_checks(self): self.set_config("hostname", "hostname") self.set_config("fqdn", "fqdn") m = manifest.Manifest() # Matching hostname should work m.hostname = "hostname" m.check_dirinfo() # Matching fqdn should also work for backwards compatibility m.hostname = "fqdn" m.check_dirinfo() # Bad match should throw a fatal error and quit m.hostname = "foobar" self.assertRaises(SystemExit, m.check_dirinfo) # But not if we tell the system to ignore it self.set_config("allow_source_mismatch", True) m.check_dirinfo()
if __name__ == "__main__": unittest.main()