60 lines
2.7 KiB
Python
60 lines
2.7 KiB
Python
import unittest
|
|
import os
|
|
from pathlib import Path
|
|
|
|
repo_cleanup = __import__('repo-cleanup')
|
|
repo_path = 'repo/'
|
|
|
|
|
|
def touch(path):
|
|
Path(path).touch()
|
|
|
|
|
|
def create_test_data():
|
|
try:
|
|
os.mkdir(repo_path)
|
|
except FileExistsError:
|
|
pass
|
|
|
|
files = ['test-r87.e176baf-1-x86_64.pkg.tar.zst', 'test-debug-r87.e176baf-1-x86_64.pkg.tar.zst', 'test-debug-r100.ab937ef-1-x86_64.pkg.tar.zst', 'test-r100.ab937ef-1-x86_64.pkg.tar.zst', 'repo.db',
|
|
'repo.db.tar.zst', 'repo.files', 'repo.files.tar.zst', 'example-program-10.2.1-2-x86_64.pkg.tar.zst', 'example-program-8.4.3-5-x86_64.pkg.tar.zst', 'example-program-debug-10.2.1-2-x86_64.pkg.tar.zst']
|
|
for f in files:
|
|
touch(f'{repo_path}/{f}')
|
|
|
|
|
|
class Tests(unittest.TestCase):
|
|
def test_list_old(self):
|
|
create_test_data()
|
|
result = repo_cleanup.get_old_packages(repo_path)
|
|
expected = ['example-program-8.4.3-5-x86_64.pkg.tar.zst',
|
|
'test-debug-r87.e176baf-1-x86_64.pkg.tar.zst', 'test-r87.e176baf-1-x86_64.pkg.tar.zst']
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_list_new(self):
|
|
create_test_data()
|
|
result = repo_cleanup.get_new_packages(repo_path)
|
|
expected = ['example-program-10.2.1-2-x86_64.pkg.tar.zst', 'example-program-debug-10.2.1-2-x86_64.pkg.tar.zst',
|
|
'test-debug-r100.ab937ef-1-x86_64.pkg.tar.zst', 'test-r100.ab937ef-1-x86_64.pkg.tar.zst']
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_package_map(self):
|
|
create_test_data()
|
|
result = repo_cleanup.HelperFunctions.package_map(repo_path)
|
|
expected = {'example-program': ['example-program-8.4.3-5-x86_64.pkg.tar.zst', 'example-program-10.2.1-2-x86_64.pkg.tar.zst', 'example-program-debug-10.2.1-2-x86_64.pkg.tar.zst'], 'test': [
|
|
'test-debug-r87.e176baf-1-x86_64.pkg.tar.zst', 'test-debug-r100.ab937ef-1-x86_64.pkg.tar.zst', 'test-r87.e176baf-1-x86_64.pkg.tar.zst', 'test-r100.ab937ef-1-x86_64.pkg.tar.zst']}
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_repo_name(self):
|
|
create_test_data()
|
|
result = repo_cleanup.HelperFunctions.get_repo_name(repo_path)
|
|
expected = 'repo'
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_package_list(self):
|
|
create_test_data()
|
|
result = repo_cleanup.HelperFunctions.get_package_list(repo_path)
|
|
expected = ['test-r87.e176baf-1-x86_64.pkg.tar.zst', 'test-debug-r100.ab937ef-1-x86_64.pkg.tar.zst', 'test-r100.ab937ef-1-x86_64.pkg.tar.zst', 'example-program-10.2.1-2-x86_64.pkg.tar.zst',
|
|
'example-program-8.4.3-5-x86_64.pkg.tar.zst', 'test-debug-r87.e176baf-1-x86_64.pkg.tar.zst', 'example-program-debug-10.2.1-2-x86_64.pkg.tar.zst']
|
|
|
|
|
|
create_test_data()
|