def test_missing_auto_upload_dir(self):
"""Test for AUTO_UPLOAD_DIR missing. It should exit returning False."""
# None case.
settings.AUTO_UPLOAD_DIR = None
self.assertFalse(Command.create_auto_upload_dirs())
# Empty case.
settings.AUTO_UPLOAD_DIR = ""
self.assertFalse(Command.create_auto_upload_dirs())
def test_missing_auto_upload_dir(self):
"""Test for AUTO_UPLOAD_DIR missing. It should exit returning False."""
# None case.
settings.AUTO_UPLOAD_DIR = None
self.assertFalse(Command.create_auto_upload_dirs())
# Empty case.
settings.AUTO_UPLOAD_DIR = ""
self.assertFalse(Command.create_auto_upload_dirs())
def test_auto_upload_sync_creation(self):
"""Tests automated case folder creation."""
# Create temporary directory to store everything.
tmp_path = tempfile.mkdtemp()
# Build the ghiro path for auto upload.
ghiro_path = os.path.join(tmp_path, "ghiro-test")
# Set path and test.
settings.AUTO_UPLOAD_DIR = ghiro_path
# Create base dir.
Command.create_auto_upload_dirs()
# Create case.
case = Case.objects.create(name="a", owner=self.user)
dir_name = os.path.join(settings.AUTO_UPLOAD_DIR, case.directory_name)
self.assertTrue(os.path.exists(dir_name))
# Cleanup.
shutil.rmtree(tmp_path)
def test_case_folders_creation(self):
# Create cases.
case1 = Case.objects.create(name="aaa", owner=self.user)
case2 = Case.objects.create(name="aab", owner=self.user)
# Create temporary directory to store everything.
tmp_path = tempfile.mkdtemp()
# Build the ghiro path for auto upload.
ghiro_path = os.path.join(tmp_path, "ghiro-test")
# Set path and create folders.
settings.AUTO_UPLOAD_DIR = ghiro_path
Command.create_auto_upload_dirs()
# Test.
for case in [case1, case2]:
case_path = os.path.join(ghiro_path, case.directory_name)
self.assertTrue(os.path.exists(case_path))
# Cleanup.
shutil.rmtree(tmp_path)
def test_case_folders_creation(self):
# Create cases.
case1 = Case.objects.create(name="aaa", owner=self.user)
case2 = Case.objects.create(name="aab", owner=self.user)
# Create temporary directory to store everything.
tmp_path = tempfile.mkdtemp()
# Build the ghiro path for auto upload.
ghiro_path = os.path.join(tmp_path, "ghiro-test")
# Set path and create folders.
settings.AUTO_UPLOAD_DIR = ghiro_path
Command.create_auto_upload_dirs()
# Test.
for case in [case1, case2]:
case_path = os.path.join(ghiro_path, case.directory_name)
self.assertTrue(os.path.exists(case_path))
# Cleanup.
shutil.rmtree(tmp_path)
def test_create_auto_upload_dir(self):
"""Test for AUTO_UPLOAD_DIR creation when missing."""
# Create temporary directory to store everything.
tmp_path = tempfile.mkdtemp()
# Build the ghiro path for auto upload.
ghiro_path = os.path.join(tmp_path, "ghiro-test")
# Set path and test.
settings.AUTO_UPLOAD_DIR = ghiro_path
self.assertNotEqual(Command.create_auto_upload_dirs(), False)
self.assertTrue(os.path.exists(ghiro_path))
# Cleanup.
shutil.rmtree(tmp_path)
def test_create_auto_upload_dir(self):
"""Test for AUTO_UPLOAD_DIR creation when missing."""
# Create temporary directory to store everything.
tmp_path = tempfile.mkdtemp()
# Build the ghiro path for auto upload.
ghiro_path = os.path.join(tmp_path, "ghiro-test")
# Set path and test.
settings.AUTO_UPLOAD_DIR = ghiro_path
self.assertNotEqual(Command.create_auto_upload_dirs(), False)
self.assertTrue(os.path.exists(ghiro_path))
# Cleanup.
shutil.rmtree(tmp_path)
def test_cleanup_auto_upload_dir(self):
"""Test for AUTO_UPLOAD_STARTUP_CLEANUP."""
# Create temporary directory to store everything.
tmp_path = tempfile.mkdtemp()
# Build the ghiro path for auto upload.
ghiro_path = os.path.join(tmp_path, "ghiro-test")
# Set path and create folders.
settings.AUTO_UPLOAD_DIR = ghiro_path
os.mkdir(ghiro_path)
# Test folder.
test_path = os.path.join(ghiro_path, "test")
os.mkdir(test_path)
# Test 1: not cleaning.
settings.AUTO_UPLOAD_STARTUP_CLEANUP = False
Command.create_auto_upload_dirs()
self.assertTrue(os.path.exists(test_path))
# Test 2: cleaning.
settings.AUTO_UPLOAD_STARTUP_CLEANUP = True
Command.create_auto_upload_dirs()
self.assertFalse(os.path.exists(test_path))
# Cleanup.
shutil.rmtree(tmp_path)
def test_cleanup_auto_upload_dir(self):
"""Test for AUTO_UPLOAD_STARTUP_CLEANUP."""
# Create temporary directory to store everything.
tmp_path = tempfile.mkdtemp()
# Build the ghiro path for auto upload.
ghiro_path = os.path.join(tmp_path, "ghiro-test")
# Set path and create folders.
settings.AUTO_UPLOAD_DIR = ghiro_path
os.mkdir(ghiro_path)
# Test folder.
test_path = os.path.join(ghiro_path, "test")
os.mkdir(test_path)
# Test 1: not cleaning.
settings.AUTO_UPLOAD_STARTUP_CLEANUP = False
Command.create_auto_upload_dirs()
self.assertTrue(os.path.exists(test_path))
# Test 2: cleaning.
settings.AUTO_UPLOAD_STARTUP_CLEANUP = True
Command.create_auto_upload_dirs()
self.assertFalse(os.path.exists(test_path))
# Cleanup.
shutil.rmtree(tmp_path)
def test_existent_auto_upload_dir(self):
"""Test for AUTO_UPLOAD_DIR creation when it already exist."""
# Create cases.
case1 = Case.objects.create(name="aaa", owner=self.user)
case2 = Case.objects.create(name="aab", owner=self.user)
# Create temporary directory to store everything.
tmp_path = tempfile.mkdtemp()
# Build the ghiro path for auto upload.
ghiro_path = os.path.join(tmp_path, "ghiro-test")
# Create AUTO_UPLOAD_DIR.
os.mkdir(ghiro_path)
# Set path and test.
settings.AUTO_UPLOAD_DIR = ghiro_path
self.assertNotEqual(Command.create_auto_upload_dirs(), False)
self.assertTrue(os.path.exists(ghiro_path))
# Cleanup.
shutil.rmtree(tmp_path)
def test_existent_auto_upload_dir(self):
"""Test for AUTO_UPLOAD_DIR creation when it already exist."""
# Create cases.
case1 = Case.objects.create(name="aaa", owner=self.user)
case2 = Case.objects.create(name="aab", owner=self.user)
# Create temporary directory to store everything.
tmp_path = tempfile.mkdtemp()
# Build the ghiro path for auto upload.
ghiro_path = os.path.join(tmp_path, "ghiro-test")
# Create AUTO_UPLOAD_DIR.
os.mkdir(ghiro_path)
# Set path and test.
settings.AUTO_UPLOAD_DIR = ghiro_path
self.assertNotEqual(Command.create_auto_upload_dirs(), False)
self.assertTrue(os.path.exists(ghiro_path))
# Cleanup.
shutil.rmtree(tmp_path)
