Changeset 40923 for branches/ipp-259_genericise_backups/tools/backups/backup_test.py

Timestamp:

Oct 8, 2019, 3:17:21 PM (7 years ago)

Author:

fairlamb

Message:

Fleshed out basic copy functionality; based off of atlassian backups

File:

• branches/ipp-259_genericise_backups/tools/backups/backup_test.py (modified) (4 diffs)

branches/ipp-259_genericise_backups/tools/backups/backup_test.py

@@ -3 +3 @@
 import socket
 from configparser import ConfigParser
-
+from pathlib import Path
+
+import backups.utils.errors as errs
 from backups.backup import Backup
-from backups.utils.errors import ConfigError
 
 
@@ -17 +18 @@
     config = ConfigParser()
 
+    # make the paths:
+    for b in backup_targs:
+        os.makedirs(os.path.join(tmpdir, b))
+
+    # Get the correct formats for the config
     backup_paths = ','.join([os.path.join(tmpdir, b) for b in backup_targs])
     source_machine = socket.gethostname()
-
     config['DEFAULT'] = \
     {
@@ -98 +103 @@
     def test_failure_if_no_defaults_provided(self):
         config = ConfigParser()
-        with pytest.raises(ConfigError):
+        with pytest.raises(errs.ConfigError):
             Backup(config_file=config)
 
-        with pytest.raises(ConfigError):
+        with pytest.raises(errs.ConfigError):
             Backup(config_file=None)
 
-        with pytest.raises(ConfigError):
+        with pytest.raises(errs.ConfigError):
             Backup(config_file="/not/a/real/file.config")
 
@@ -130 +135 @@
         backy.default_dict['source_machine'] = actual_hostname
         assert backy.hostname_is_valid() is True
+
+
+class TestRunningCopy():
+
+    def test_simple_copy(self, tmpdir):
+        config = make_default_config(tmpdir, ['bck1', 'bck2'])
+        source = os.path.join(tmpdir, "and_specific_file.jpeg")
+        config['COPY'] = {'sources' : f"{source}"}
+
+        # make file to be copied
+        Path(source).touch()
+
+        backy = Backup(config)
+
+        expected1file = os.path.join(tmpdir, 'bck1', 'and_specific_file.jpeg')
+        expected2file = os.path.join(tmpdir, 'bck2', 'and_specific_file.jpeg')
+        assert not os.path.exists(expected1file)
+        assert not os.path.exists(expected2file)
+        backy._run_copy()
+        assert os.path.exists(expected1file)
+        assert os.path.exists(expected2file)
+
+    def test_copy_with_additional_args(self, tmpdir):
+        config = make_default_config(tmpdir, ['bck1', 'bck2'])
+        source1 = os.path.join(tmpdir, "some_dir")
+        source2 = os.path.join(tmpdir, "and_specific_file.jpeg")
+        config['COPY'] = \
+        { 'sources' : f"{source1}, {source2}"
+        , 'additional_args' : "-v --preserve -r"
+        }
+
+        os.makedirs(source1)
+        Path(source2).touch()
+
+        backy = Backup(config)
+
+        expected1dir  = os.path.join(tmpdir, 'bck1', 'some_dir')
+        expected1file = os.path.join(tmpdir, 'bck1', 'and_specific_file.jpeg')
+        expected2dir  = os.path.join(tmpdir, 'bck2', 'some_dir')
+        expected2file = os.path.join(tmpdir, 'bck2', 'and_specific_file.jpeg')
+        assert not os.path.exists(expected1dir)
+        assert not os.path.exists(expected1file)
+        assert not os.path.exists(expected2dir)
+        assert not os.path.exists(expected2file)
+        backy._run_copy()
+        assert os.path.exists(expected1dir)
+        assert os.path.exists(expected1file)
+        assert os.path.exists(expected2dir)
+        assert os.path.exists(expected2file)
+
+    def copy_fails_if_source_file_does_not_exist(self, tmpdir):
+        config = make_default_config(tmpdir, ['bck1', 'bck2'])
+        source = os.path.join(tmpdir, "and_specific_file.jpeg")
+        config['COPY'] = \
+        { 'sources' : f"{source}"
+        , 'additional_args' : '-r'
+        }
+
+        assert not os.path.exists(source)
+        backy = Backup(config)
+        with pytest.raises(errs.SubprocessError):
+            backy._run_copy()
+
+    def copy_fails_if_destination_does_not_exist(self, tmpdir):
+        config = ConfigParser()
+
+        backup_does_not_exist = os.path.join(tmpdir, 'does_not_exist')
+
+        config['DEFAULT'] = \
+        { 'backup_paths' : backup_does_not_exist
+        , 'source_machine': socket.gethostname()
+        }
+        source = os.path.join(tmpdir, "and_specific_file.jpeg")
+        Path(source).touch()
+        config['COPY'] = {'sources' : f"{source}"}
+
+        assert not os.path.exists(backup_does_not_exist)
+        backy = Backup(config)
+        with pytest.raises(errs.SubprocessError):
+            backy._run_copy()
+
+    def test_copy_requires_dash_r_arg_for_copying_dirs(self, tmpdir):
+        config = make_default_config(tmpdir, ['bck1', 'bck2'])
+        source = os.path.join(tmpdir, "some_dir")
+        config['COPY'] = \
+        { 'sources' : f"{source}"
+        }
+        os.makedirs(source)
+
+        with pytest.raises(errs.SubprocessError):
+            backy = Backup(config)
+            backy._run_copy()
+
+        config['COPY'] = \
+        { 'sources' : f"{source}"
+        , 'additional_args' : '-r'
+        }
+        backy = Backup(config)
+
+        expected1dir  = os.path.join(tmpdir, 'bck1', 'some_dir')
+        expected2dir  = os.path.join(tmpdir, 'bck2', 'some_dir')
+        assert not os.path.exists(expected1dir)
+        assert not os.path.exists(expected2dir)
+        backy._run_copy()
+        assert os.path.exists(expected1dir)
+        assert os.path.exists(expected2dir)
+
+
+# class TestRunningTar():
+
+#     def test_simply_tar(self):
+#         assert True is False
+
+#     def test_tar_with_additional_args(self):
+#         assert True is False
+
+
+# class TestRunningRsync():
+
+#     def test_simply_rsync(self):
+#         assert True is False
+
+#     def test_rsync_with_additional_args(self):
+#         assert True is False
+
+
+# class TestRunningMysqldump():
+
+#     def test_simply_mysqldump(self):
+#         assert True is False
+
+#     def test_mysqldump_with_additional_args(self):
+#         assert True is False