python/refit/test_librefit.py

from src.modules.librefit import (
    get_current_path,
    get_standard_folder_name,
    get_standard_name_number,
    get_standard_file_name,
)


def test_get_default_file_name():
    """Tests if the function returns the correct value"""
    default_name = get_standard_file_name(None)
    assert "file" in default_name


def test_get_default_folder_name():
    """Test if the default directory name gets returned"""
    default_name = get_standard_folder_name(None)
    assert "directory" in default_name


def test_get_standard_name_number():
    """Tests if the number function returns the correctly formatted string."""
    name_number = get_standard_name_number(20, 3)
    assert "020" in name_number


def test_get_filename():
    """Tests if a passed filename is returned properly"""
    filename = get_standard_file_name(["testname"])
    assert "testname" in filename


def test_get_folder_name():
    """Tests if the function returns the passed folder name correctly"""
    folder_name = ["folder"]
    return_folder_name = get_standard_folder_name(folder_name)
    assert "folder" in return_folder_name


def test_folder_name_list():
    """The function is supposed to only return the first name of the
    passed list"""
    folder_names = ["folder1", "folder2", "folder3"]
    return_folder_name = get_standard_folder_name(folder_names)
    assert "folder1" in return_folder_name


def test_get_current_directory():
    """Tests if the directory is set to the current directory, if None
    is passed with the argument"""
    path = None
    directory = get_current_path(path)
    assert "." in directory


def test_for_existing_path():
    """Tests if the function returns the correct path."""
    path = "/home/cerberus/Documents/books/"
    directory = get_current_path(path)
    assert "/home/cerberus/Documents/books/" in directory