import os import pytest from unittest.mock import MagicMock, Mock from g2h.file import generate_breadcrumbs, sanitize_file_name, tree_join def test_sanitize_not_necessary(): assert(sanitize_file_name('foo', None) == 'foo') def test_sanitize_success(): assert(sanitize_file_name('.test', []) == '_test') def test_sanitize_collissions(): entries = [] for i in range(1, 10): entry = Mock() entry.name = '_' * i + 'test' entries.append(entry) with pytest.raises(RuntimeError): sanitize_file_name('.test', entries) def test_tree_join(): commit = Mock() commit.hexsha = '123' assert(tree_join(commit, 'test') == os.path.join('123', 't', 'test')) def test_generate_breadcrumbs(): commit = Mock() commit.hexsha = '123' assert(generate_breadcrumbs('', commit, '', None) == '[/](/123/t/)') assert(generate_breadcrumbs('', commit, 'sub', None) == '[/](/123/t/)/[sub](/123/t/sub)') assert(generate_breadcrumbs('', commit, 'sub/dir', None) == '[/](/123/t/)/[sub](/123/t/sub)/[dir](/123/t/sub/dir)') assert(generate_breadcrumbs('', commit, 'sub/dir/file.txt', None) == '[/](/123/t/)/[sub](/123/t/sub)/[dir](/123/t/sub/dir)/[file.txt](/123/t/sub/dir/file.txt)') assert(generate_breadcrumbs('content sub', commit, 'sub', None) == '[/](/content sub/123/t/)/[sub](/content sub/123/t/sub)') assert(generate_breadcrumbs('', commit, 'sub/.file', MagicMock()) == '[/](/123/t/)/[sub](/123/t/sub)/[.file](/123/t/sub/_file)')