Changeset 40962 for branches/ipp-259_genericise_backups/tools/backups/confluence_backup_test.py

Timestamp:

Oct 22, 2019, 2:30:49 PM (7 years ago)

Author:

fairlamb

Message:

fixed calling as main + tests, removed argsparse from most

File:

• branches/ipp-259_genericise_backups/tools/backups/confluence_backup_test.py (modified) (7 diffs)

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

@@ -6 +6 @@
 from pathlib import Path
 
-import backups.backup as bckup
-import backups.confluence_backup as jbck_mod
-import backups.utils.errors as errs
-from backups.confluence_backup import ConfluenceBackup
-from backups.utils.errors import ValidationError
-from backups.utils.config_parse_helper import ConfigSchema
+import backup as bckup
+import confluence_backup as cbck_mod
+
+from confluence_backup import ConfluenceBackup
+from utils.errors import ValidationError
+from utils.config_parse_helper import ConfigSchema
 
 CONF_TEST_CONFIG = os.path.join(os.path.dirname(__file__), './testing/confluence_test.config')
@@ -67 +67 @@
 
     # Create backup files that will be copied
-    confluence_backup_name = ConfluenceBackup.get_confluence_backup_date_format() + "_confluence_test_file.zip"
+    confluence_backup_name = f"confluence-xml-backup-{ConfluenceBackup.get_confluence_backup_date_format()}.zip"
     confluence_backup_filepath = os.path.join(confluence_backup_path, confluence_backup_name)
     Path(confluence_backup_filepath).touch()
@@ -96 +96 @@
             }
         assert jb.copy_dict == \
-            { 'sources' : ['/var/atlassian/application-data/confluence/export']
+            { 'sources' : ['/var/atlassian/application-data/confluence/backups']
             , 'additional_args' : ['-v']
             }
         assert jb.tar_dict == \
-            { 'sources' : ['/var/atlassian/application-data/confluence/data']
+            { 'sources' : ['/var/atlassian/application-data/confluence/attachments']
             , 'additional_args' : ['--verbose', '-z', '-p', '-cf']
             , 'target_filename' : 'confluence.tar.gz'
@@ -118 +118 @@
 
     def test_default_config_fails_to_load_if_the_file_does_not_exist(self):
-        default_config = jbck_mod.EXPECTED_CONFIG_FILE
+        default_config = cbck_mod.EXPECTED_CONFIG_FILE
         assert not os.path.exists(default_config)
 
-        with pytest.raises(errs.ValidationError) as e:
+        with pytest.raises(ValidationError) as e:
             ConfluenceBackup()
         assert "ValidationError: Config filepath does not exist" in str(e)
@@ -174 +174 @@
 
     def test_nonexistant_config_file_raises_error(self):
-        with pytest.raises(errs.ValidationError) as e:
+        with pytest.raises(ValidationError) as e:
             ConfluenceBackup("not a real file path")
         assert "ValidationError: Config filepath does not exist" in str(e)
 
     def test_main_loads_default_and_fails_due_to_missing_paths(self):
-        with pytest.raises(errs.ValidationError) as e:
-            jbck_mod.main()
+        with pytest.raises(ValidationError) as e:
+            cbck_mod.main()
         assert "ValidationError: Config filepath does not exist" in str(e)
 
 
+class TestFullRunFromMain(object):
+
+    def test_main_loads_and_runs_fine_when_given_config(self, tmpdir):
+        config = configparser.ConfigParser()
+        bpath1 = os.path.join(tmpdir, "path/1")
+        bpath2 = os.path.join(tmpdir, "path/2")
+        config['DEFAULT'] = \
+            { 'backup_paths' :  f"{bpath1}, {bpath2}"
+            , 'source_machine' : 'rocket'
+            , 'verbose' : 'True'
+            }
+        copy_source = os.path.join(tmpdir, "copy/source")
+        conf_backup_name = ConfluenceBackup.get_confluence_backup_date_format() + "_conf_test_file.zip"
+        config['COPY'] = \
+            { 'sources' : f'{copy_source}'
+            }
+        tar_source =  os.path.join(tmpdir, "tar/source")
+        config['TAR'] = \
+            { 'sources' : f"{tar_source}"
+            , 'additional_args' : "--verbose -z -p -cf"
+            , 'target_filename' : 'conf.tar.gz'
+            , 'prefix_date' : 'False'
+            }
+        os.makedirs(bpath1)
+        os.makedirs(bpath2)
+        os.makedirs(copy_source)
+        os.makedirs(tar_source)
+        Path(os.path.join(copy_source, conf_backup_name)).touch()
+
+        schema = ConfigSchema(None)
+        schema.add_section(bckup.DEFAULT_SCHEMA_SECTION)
+        schema.add_section(bckup.COPY_SCHEMA_SECTION)
+        schema.add_section(bckup.TAR_SCHEMA_SECTION)
+
+        cbck_mod.main(config, schema)
+        assert True is True
+
+
 class TestCopyingBackups(object):
 
@@ -189 +227 @@
 
     def test_get_confluence_backup_date_format(self):
-        jb = ConfluenceBackup(config_file=CONF_TEST_CONFIG)
+        cb = ConfluenceBackup(config_file=CONF_TEST_CONFIG)
         today = datetime.date.today()
         expected_format = today.strftime("%Y_%m_%d")
-        assert jb.get_confluence_backup_date_format() == expected_format
+        assert cb.get_confluence_backup_date_format() == expected_format
 
         another_date = datetime.date(1988, 11, 23)
         expected_format = another_date.strftime("%Y_%m_%d")
-        assert jb.get_confluence_backup_date_format(another_date) == expected_format
+        assert cb.get_confluence_backup_date_format(another_date) == expected_format
 
     def test_copying_fails_if_paths_not_present(self):
-        jb = ConfluenceBackup(config_file=CONF_TEST_CONFIG)
-        # return_value = jb._copy_confluence_backup()
-        with pytest.raises(ValidationError) as e:
-            jb._run_copy()
+        cb = ConfluenceBackup(config_file=CONF_TEST_CONFIG)
+        # return_value = cb._copy_confluence_backup()
+        with pytest.raises(ValidationError) as e:
+            cb._run_copy()
         assert "ValidationError: target path does not exist:" in str(e)
 
@@ -208 +246 @@
         create_items_for_full_confluence_backup_test(tmpdir)
         config = make_config(tmpdir)
-        jb = ConfluenceBackup(config_file=config)
+        cb = ConfluenceBackup(config_file=config)
 
         today = ConfluenceBackup.get_confluence_backup_date_format()
-        expected_host_backup_copy = os.path.join(tmpdir, "host_backup_path", f"{today}_confluence_test_file.zip" )
-        expected_backup_1_copy    = os.path.join(tmpdir, "backup_1_path",    f"{today}_confluence_test_file.zip" )
-        expected_backup_2_copy    = os.path.join(tmpdir, "backup_2_path",    f"{today}_confluence_test_file.zip" )
+        expected_host_backup_copy = os.path.join(tmpdir, "host_backup_path", f"confluence-xml-backup-{today}.zip" )
+        expected_backup_1_copy    = os.path.join(tmpdir, "backup_1_path",    f"confluence-xml-backup-{today}.zip" )
+        expected_backup_2_copy    = os.path.join(tmpdir, "backup_2_path",    f"confluence-xml-backup-{today}.zip" )
         assert not os.path.exists(expected_host_backup_copy)
         assert not os.path.exists(expected_backup_1_copy)
         assert not os.path.exists(expected_backup_2_copy)
 
-        return_value = jb._run_copy()
+        return_value = cb._run_copy()
         assert return_value == 0