Skip to content

Commit cd6b3bf

Browse files
SG-38267 Replace mock (#1010)
* Replace mock * Clean-up
1 parent 438a3fa commit cd6b3bf

File tree

3 files changed

+17
-22
lines changed

3 files changed

+17
-22
lines changed

tests/platform_tests/test_application.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -10,20 +10,18 @@
1010

1111
from __future__ import with_statement
1212

13-
import sys
13+
import inspect
14+
import logging
1415
import os
1516
import shutil
16-
import logging
1717
import tempfile
18-
import mock
19-
import inspect
18+
from unittest import mock
2019

21-
from tank_test.tank_test_base import *
2220
import tank
2321
from tank.errors import TankError, TankHookMethodDoesNotExistError
24-
from tank.platform import application, constants, validation
22+
from tank.platform import application, validation
2523
from tank.template import Template
26-
from tank.deploy import descriptor
24+
from tank_test.tank_test_base import *
2725
from tank_vendor import six
2826
from tank_vendor.six import StringIO
2927

tests/platform_tests/test_engine.py

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -12,27 +12,26 @@
1212
Engine-related unit tests.
1313
"""
1414

15-
from __future__ import with_statement, print_function
15+
from __future__ import print_function, with_statement
1616

17+
import contextlib
1718
import os
19+
import random
1820
import sys
1921
import threading
20-
import random
2122
import time
23+
from unittest import mock
2224

25+
import sgtk
26+
import tank
27+
from sgtk.platform import engine
28+
from tank.errors import TankError
29+
from tank_test.tank_test_base import setUpModule # noqa
2330
from tank_test.tank_test_base import (
2431
TankTestBase,
2532
skip_if_pyside_missing,
2633
suppress_generated_code_qt_warnings,
2734
)
28-
from tank_test.tank_test_base import setUpModule # noqa
29-
30-
import contextlib
31-
import tank
32-
import sgtk
33-
from sgtk.platform import engine
34-
from tank.errors import TankError
35-
import mock
3635

3736

3837
class TestEngineBase(TankTestBase):

tests/python/tank_test/mock_appstore.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,17 +16,15 @@
1616

1717
import functools
1818
import sys
19+
from unittest import mock
1920

20-
import mock
21-
22-
from sgtk.descriptor import Descriptor
21+
from sgtk.descriptor import Descriptor, create_descriptor
2322
from sgtk.descriptor.io_descriptor.base import IODescriptorBase
24-
from sgtk.descriptor import create_descriptor
2523
from sgtk.util import sgre as re
26-
2724
from tank import TankError
2825
from tank.platform.environment import InstalledEnvironment
2926
from tank.util import suppress_known_deprecation
27+
3028
if sys.version_info[0:2] >= (3, 10):
3129
from setuptools._distutils.version import LooseVersion
3230
else:

0 commit comments

Comments
 (0)