def test_run():
"""Test a full run."""
src_path = "./test/testsource"
dst_path = "./test/testdestination"
argv = ("main.py", src_path, dst_path)
main = Main(argv)
main.run()
destfile1 = "./test/testdestination/rosa (814837)/"
destfile1 += "(814837) ろさ - 空と私.jpg"
print(destfile1)
assert os.path.exists(destfile1)
destfile2 = "./test/testdestination/cherrypin (206921)/"
destfile2 += "(206921) cherrypin - イラスト集め"
assert os.path.exists(destfile2)
# Copy files back
shutil.copy2(destfile1, src_path)
os.remove(destfile1)
shutil.copytree(destfile2, "./test/testsource/(206921) cherrypin - イラスト集め")
shutil.rmtree(destfile2)
srcfile1 = "./test/testsource/(814837) ろさ - 空と私.jpg"
assert os.path.exists(srcfile1)
srcfile2 = "./test/testsource/(206921) cherrypin - イラスト集め"
assert os.path.exists(srcfile2)
def test_run():
"""Test a full run."""
src_path = "./test/testsource"
dst_path = "./test/testdestination"
argv = ("main.py", src_path, dst_path)
main = Main(argv)
main.run()
destfile1 = "./test/testdestination/rosa (814837)/"
destfile1 += "(814837) ろさ - 空と私.jpg"
print(destfile1)
assert os.path.exists(destfile1)
destfile2 = "./test/testdestination/cherrypin (206921)/"
destfile2 += "(206921) cherrypin - イラスト集め"
assert os.path.exists(destfile2)
# Copy files back
shutil.copy2(destfile1, src_path)
os.remove(destfile1)
shutil.copytree(destfile2,
"./test/testsource/(206921) cherrypin - イラスト集め")
shutil.rmtree(destfile2)
srcfile1 = "./test/testsource/(814837) ろさ - 空と私.jpg"
assert os.path.exists(srcfile1)
srcfile2 = "./test/testsource/(206921) cherrypin - イラスト集め"
assert os.path.exists(srcfile2)
def test_wrong_dst_argument():
"""Test if script exits when destination argument is a file"""
src_path = "./test/testsource/(814837) ろさ - 空と私.jpg"
dst_path = "./test/testdestination"
argv = ("main.py", dst_path, src_path)
main = Main(argv)
def test_additional_arguments():
"""Test if script exits on too many arguments."""
src_path = "./test/testsource"
dst_path = "./test/testdestination"
argv = ("main.py", src_path, dst_path, dst_path)
main = Main(argv)
def test_arguments():
"""Test if arguments are read correctly."""
src_path = "./test/testsource"
dst_path = "./test/testdestination"
argv = ("main.py", src_path, dst_path)
main = Main(argv)
assert (main.src_path == src_path)
assert (main.dst_path == dst_path)
#!/usr/bin/env python # -*- coding: utf-8 -*- from pixivsort.main import Main from sys import argv main = Main(argv) main.run()
def test_missing_arguments():
"""Test if script exits on too few arguments."""
src_path = "./test/testsource"
argv = ("main.py", src_path)
main = Main(argv)