1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
#!/usr/bin/env python
# coding: utf-8
from __future__ import unicode_literals
import mock
import os
import subprocess
import tempfile
import unittest
import shutil
from mkdocs.utils import ghp_import
class UtilsTests(unittest.TestCase):
@mock.patch('subprocess.call', auto_spec=True)
@mock.patch('subprocess.Popen', auto_spec=True)
def test_try_rebase(self, mock_popen, mock_call):
popen = mock.Mock()
mock_popen.return_value = popen
popen.communicate.return_value = (
'4c82346e4b1b816be89dd709d35a6b169aa3df61\n', '')
popen.wait.return_value = 0
ghp_import.try_rebase('origin', 'gh-pages')
mock_popen.assert_called_once_with(
['git', 'rev-list', '--max-count=1', 'origin/gh-pages'],
stdin=subprocess.PIPE, stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
mock_call.assert_called_once_with(
['git', 'update-ref', 'refs/heads/gh-pages',
'4c82346e4b1b816be89dd709d35a6b169aa3df61'])
@mock.patch('subprocess.Popen', auto_spec=True)
def test_get_prev_commit(self, mock_popen):
popen = mock.Mock()
mock_popen.return_value = popen
popen.communicate.return_value = (
b'4c82346e4b1b816be89dd709d35a6b169aa3df61\n', '')
popen.wait.return_value = 0
result = ghp_import.get_prev_commit('test-branch')
self.assertEqual(result, u'4c82346e4b1b816be89dd709d35a6b169aa3df61')
mock_popen.assert_called_once_with(
['git', 'rev-list', '--max-count=1', 'test-branch', '--'],
stdin=subprocess.PIPE, stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
@mock.patch('subprocess.Popen', auto_spec=True)
def test_get_config(self, mock_popen):
popen = mock.Mock()
mock_popen.return_value = popen
popen.communicate.return_value = (
b'Dougal Matthews\n', '')
result = ghp_import.get_config('user.name')
self.assertEqual(result, u'Dougal Matthews')
mock_popen.assert_called_once_with(
['git', 'config', 'user.name'],
stdout=subprocess.PIPE, stdin=subprocess.PIPE)
@mock.patch('mkdocs.utils.ghp_import.get_prev_commit')
@mock.patch('mkdocs.utils.ghp_import.get_config')
def test_start_commit(self, mock_get_config, mock_get_prev_commit):
pipe = mock.Mock()
mock_get_config.side_effect = ['username', 'email']
mock_get_prev_commit.return_value = 'SHA'
ghp_import.start_commit(pipe, 'test-branch', 'test-message')
mock_get_prev_commit.assert_called_once_with('test-branch')
self.assertEqual(pipe.stdin.write.call_count, 5)
@mock.patch('mkdocs.utils.ghp_import.try_rebase', return_value=True)
@mock.patch('mkdocs.utils.ghp_import.get_prev_commit', return_value='sha')
@mock.patch('mkdocs.utils.ghp_import.get_config', return_value='config')
@mock.patch('subprocess.call', auto_spec=True)
@mock.patch('subprocess.Popen', auto_spec=True)
def test_ghp_import(self, mock_popen, mock_call, mock_get_config,
mock_get_prev_commit, mock_try_rebase):
directory = tempfile.mkdtemp()
open(os.path.join(directory, 'file'), 'a').close()
try:
popen = mock.Mock()
mock_popen.return_value = popen
popen.communicate.return_value = ('', '')
popen.wait.return_value = 0
ghp_import.ghp_import(directory, "test message",
remote='fake-remote-name',
branch='fake-branch-name')
self.assertEqual(mock_popen.call_count, 2)
self.assertEqual(mock_call.call_count, 0)
finally:
shutil.rmtree(directory)
|
