Skip to content

Commit 7a4f668

Browse files
committed
fix: Update mock imports to use unittest.mock for consistency
1 parent 896f1f0 commit 7a4f668

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

.github/workflows/check-format-and-typing.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,4 +30,4 @@ jobs:
3030
run: flake8 ukbot
3131

3232
- name: mypy static type checker
33-
run: mypy --config-file mypy.ini ukbot
33+
run: mypy ukbot

test/test_filters.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# encoding=utf-8
22
from collections import OrderedDict
33
import unittest
4-
from mock import Mock
4+
import unittest.mock as mock
55
from unittest import TestCase
66

77
from faker import Faker
@@ -19,31 +19,31 @@ class DummyDataProvider:
1919

2020
def page_mock(self, name=None, site=None, prefix=''):
2121
""" Create a MWClient Page instance mock """
22-
page = Mock(Page)
22+
page = mock.Mock(Page)
2323
page.site = site or self.site
2424
page.name = '%s%s' % (prefix, name or fake.name())
2525
return page
2626

2727
def article_mock(self, name=None, site=None):
2828
""" Create an UKBOt Article instance mock """
29-
article = Mock(Article)
30-
article.site = Mock(return_value=site or self.site)
29+
article = mock.Mock(Article)
30+
article.site = mock.Mock(return_value=site or self.site)
3131
article.name = name or fake.name()
3232
article.key = article.site().key + ':' + article.name
3333
return article
3434

3535
def __init__(self, articles: int, categories: int):
3636

3737
# Define a dummy site
38-
self.site = Mock(Site)
38+
self.site = mock.Mock(Site)
3939
self.site.key = 'dummy.wikipedia.org'
4040
# self.site.redirect_regexp = re.compile(u'(?:%s)' % u'|'.join(['redirect']), re.I)
4141
self.site.rights = ['bot']
4242

4343
# And a site manager
44-
self.sites = Mock(SiteManager)
45-
self.sites.keys = Mock(return_value=[self.site.key])
46-
self.sites.items = Mock(return_value=[(self.site.key, self.site)])
44+
self.sites = mock.Mock(SiteManager)
45+
self.sites.keys = mock.Mock(return_value=[self.site.key])
46+
self.sites.items = mock.Mock(return_value=[(self.site.key, self.site)])
4747

4848
# Create some articles and categories
4949
self.articles = [self.article_mock() for n in range(articles)]

test/test_rules.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# encoding=utf-8
22
import re
33
from datetime import datetime
4-
import mock
4+
import unittest.mock as mock
55
from unittest import TestCase
66

77
import pytz

0 commit comments

Comments
 (0)