Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Side by Side Diff: update-copyright/tests/test_update_copyright.py

Issue 29459580: Issue 5250 - Add copyright update script (Closed) Base URL: https://hg.adblockplus.org/codingtools
Patch Set: Minor formatting fixes Created July 4, 2017, 3:13 p.m.
Left:
Right:
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 #!/usr/bin/env python3
2
3 import os
4 import re
5 import datetime
6 import subprocess
7 import shutil
8 import urllib.parse
9
10 import pytest
11
12 from update_copyright import extract_urls, text_replace, hg_commit, main
13
14 data_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'data')
15
16
17 def create_repo(path):
18 subprocess.check_call(['hg', 'init', path])
19 with open(os.path.join(path, '.hg', 'hgrc'), 'w+') as hgrc:
20 set_user = '[ui]\nusername = Test User <test@example.com>'
21 hgrc.write(set_user)
22 shutil.copy(os.path.join(data_path, 'sample_file.py'), path)
23 subprocess.check_call(['hg', 'commit', '-Am', 'Initial commit',
24 '--repository', path])
25
26
27 @pytest.fixture()
28 def temp_dir(tmpdir):
29 temp_dir = tmpdir.mkdir('temp_dir')
30 return temp_dir
31
32
33 @pytest.fixture()
34 def temp_repo(tmpdir):
35 """"Returns a path to a temporary repo containing one sample file"""
36 temp_repo = tmpdir.mkdir('tmp_dir')
37 create_repo(str(temp_repo))
38 return temp_repo
39
40
41 @pytest.fixture()
42 def base_dir(tmpdir):
43 """Returns a temporary directory that contains one html page and two
44 repositories (one with push access, the other without)"""
45 tmp_repo = tmpdir.mkdir('tmp_dir')
46 temp_dir = str(tmp_repo)
47 subprocess.check_call(['cp', os.path.join(data_path, 'hg_page.html'),
48 temp_dir])
49 repo_1 = os.path.join(temp_dir, 'repo_1')
50 repo_2 = os.path.join(temp_dir, 'repo_2')
51 os.mkdir(repo_1)
52 os.mkdir(repo_2)
53 create_repo(repo_1)
54 create_repo(repo_2)
55
56 # Make repo_2 read-only
57 with open(os.path.join(repo_2, '.hg/hgrc'), 'w') as hgrc:
58 hook = '[hooks]\npretxnchangegroup = return True'
59 hgrc.write(hook)
60 return temp_dir
61
62
63 def test_extract_urls():
64 data_url = urllib.parse.urljoin('file:///', data_path)
65 urls = [data_url + '/repo_1/',
66 data_url + '/repo_2/']
67 assert urls == extract_urls(os.path.join(data_url, 'hg_page.html'))
68
69
70 def test_text_replacement(temp_repo):
71 filename = temp_repo.join('sample_file.py').strpath
72 text_replace(temp_repo.strpath, filename)
73 with open(filename) as file:
74 text = file.read()
75 pattern = re.compile(r'(copyright.*?\d{4})(?:-\d{4})?\s+eyeo gmbh',
76 re.I)
77 for year in re.finditer(pattern, text):
78 dates = re.search(r'(\d{4})-(\d{4})', year.group(0))
79 assert dates.group(2) == str(datetime.datetime.now().year)
80
81
82 def test_hg_commit(temp_repo, temp_dir):
83 directory = str(temp_dir)
84 repo = str(temp_repo)
85 subprocess.check_call(['hg', 'clone', repo, directory])
86 open(os.path.join(directory, 'foo'), 'w').close()
87 subprocess.check_call(['hg', 'add', '--repository', directory])
88 hg_commit(directory, repo)
89
90 # Make sure both files contain the commmit message from hg log
91 log_1 = subprocess.run(['hg', 'log', '--repository', repo],
92 stdout=subprocess.PIPE)
93 assert 'Noissue - Updated copyright year' in str(log_1.stdout)
94
95
96 def test_all(base_dir):
97 main(urllib.parse.urljoin('file:///', os.path.join(
98 base_dir, 'hg_page.html')), None)
99
100 # assert hg log for repo_1
101 log_1 = subprocess.run(['hg', 'log', '--repository',
102 os.path.join(base_dir, 'repo_1')],
103 stdout=subprocess.PIPE)
104 assert 'Noissue - Updated copyright year' in str(log_1.stdout)
105
106 # assert the .patch file for repo_2
107 assert'Noissue - Updated copyright year' in open('repo_2.patch').read()
108 subprocess.call(['rm', 'repo_2.patch']) # cleanup
OLDNEW

Powered by Google App Engine
This is Rietveld