import os

import testinfra.utils.ansible_runner


testinfra_hosts = testinfra.utils.ansible_runner.AnsibleRunner(
    os.environ['MOLECULE_INVENTORY_FILE']).get_hosts('parameters-optional')


def test_backup_directories(host):
    """
    Tests if backup directories have been set-up correctly.
    """

    with host.sudo():
        directories = ["/srv/backup", "/srv/backup/mariadb"]

        for path in directories:

            directory = host.file(path)

            assert directory.is_directory
            assert directory.user == 'root'
            assert directory.group == 'root'
            assert directory.mode == 0o700


def test_backup_script_file(host):
    """
    Tests if backup script file for dumping database has been set-up correctly.
    """

    with host.sudo():
        script = host.file('/etc/duply/main/pre.d/dump_testdb.sh')

        assert script.is_file
        assert script.user == 'root'
        assert script.group == 'root'
        assert script.mode == 0o700
        assert "/usr/bin/mysqldump \"testdb\" > \"/srv/backup/mariadb/testdb.sql\"" in script.content_string


def test_backup_run(host):
    """
    Tests if backup runs correctly, and if restore will included the backed-up
    database.
    """

    with host.sudo():

        # Remove restore directory in order to make sure restore has worked
        # correctly.
        host.run("rm -rf /root/restore")

        backup_run = host.run('duply main backup')
        assert backup_run.rc == 0

        database_dump = host.file('/srv/backup/mariadb/testdb.sql')
        assert database_dump.is_file
        assert 'Database: testdb' in database_dump.content_string

        restore_run = host.run('duply main restore /root/restore')
        assert restore_run.rc == 0

        restored_database_dump = host.file('/root/restore/srv/backup/mariadb/testdb.sql')
        assert restored_database_dump.is_file
        assert restored_database_dump.content_string == database_dump.content_string