1
1
import distutils .dist
2
+ import unittest .mock
2
3
3
- import mock
4
4
import pytest
5
5
import setuptools .dist
6
6
@@ -11,7 +11,7 @@ def test_valid_rst():
11
11
dist = distutils .dist .Distribution (attrs = dict (
12
12
long_description = "Hello, I am some text." ))
13
13
checker = readme_renderer .integration .distutils .Check (dist )
14
- checker .warn = mock .Mock ()
14
+ checker .warn = unittest . mock .Mock ()
15
15
16
16
checker .check_restructuredtext ()
17
17
@@ -22,14 +22,14 @@ def test_invalid_rst():
22
22
dist = distutils .dist .Distribution (attrs = dict (
23
23
long_description = "Hello, I am some `totally borked< text." ))
24
24
checker = readme_renderer .integration .distutils .Check (dist )
25
- checker .warn = mock .Mock ()
26
- checker .announce = mock .Mock ()
25
+ checker .warn = unittest . mock .Mock ()
26
+ checker .announce = unittest . mock .Mock ()
27
27
28
28
checker .check_restructuredtext ()
29
29
30
30
# Should warn once for the syntax error, and finally to warn that the
31
31
# overall syntax is invalid
32
- checker .warn .assert_called_once_with (mock .ANY )
32
+ checker .warn .assert_called_once_with (unittest . mock .ANY )
33
33
message = checker .warn .call_args [0 ][0 ]
34
34
assert 'invalid markup' in message
35
35
assert 'line 1: Warning:' in message
@@ -47,14 +47,14 @@ def test_malicious_rst():
47
47
dist = distutils .dist .Distribution (attrs = dict (
48
48
long_description = description ))
49
49
checker = readme_renderer .integration .distutils .Check (dist )
50
- checker .warn = mock .Mock ()
51
- checker .announce = mock .Mock ()
50
+ checker .warn = unittest . mock .Mock ()
51
+ checker .announce = unittest . mock .Mock ()
52
52
53
53
checker .check_restructuredtext ()
54
54
55
55
# Should warn once for the syntax error, and finally to warn that the
56
56
# overall syntax is invalid
57
- checker .warn .assert_called_once_with (mock .ANY )
57
+ checker .warn .assert_called_once_with (unittest . mock .ANY )
58
58
message = checker .warn .call_args [0 ][0 ]
59
59
assert 'directive disabled' in message
60
60
@@ -68,7 +68,7 @@ def test_markdown():
68
68
long_description = "Hello, I am some text." ,
69
69
long_description_content_type = "text/markdown" ))
70
70
checker = readme_renderer .integration .distutils .Check (dist )
71
- checker .warn = mock .Mock ()
71
+ checker .warn = unittest . mock .Mock ()
72
72
73
73
checker .check_restructuredtext ()
74
74
@@ -79,21 +79,21 @@ def test_markdown():
79
79
def test_invalid_missing ():
80
80
dist = distutils .dist .Distribution (attrs = dict ())
81
81
checker = readme_renderer .integration .distutils .Check (dist )
82
- checker .warn = mock .Mock ()
82
+ checker .warn = unittest . mock .Mock ()
83
83
84
84
checker .check_restructuredtext ()
85
85
86
- checker .warn .assert_called_once_with (mock .ANY )
86
+ checker .warn .assert_called_once_with (unittest . mock .ANY )
87
87
assert 'missing' in checker .warn .call_args [0 ][0 ]
88
88
89
89
90
90
def test_invalid_empty ():
91
91
dist = distutils .dist .Distribution (attrs = dict (
92
92
long_description = "" ))
93
93
checker = readme_renderer .integration .distutils .Check (dist )
94
- checker .warn = mock .Mock ()
94
+ checker .warn = unittest . mock .Mock ()
95
95
96
96
checker .check_restructuredtext ()
97
97
98
- checker .warn .assert_called_once_with (mock .ANY )
98
+ checker .warn .assert_called_once_with (unittest . mock .ANY )
99
99
assert 'missing' in checker .warn .call_args [0 ][0 ]
0 commit comments