diff -Naur pgcli-3.1.0-original/tests/metadata.py pgcli-3.1.0/tests/metadata.py
--- pgcli-3.1.0-original/tests/metadata.py 2020-08-14 18:08:15.000000000 -0400
+++ pgcli-3.1.0/tests/metadata.py 2021-03-29 12:48:22.105962073 -0400
@@ -3,7 +3,7 @@
from pgcli.packages.parseutils.meta import FunctionMetadata, ForeignKey
from prompt_toolkit.completion import Completion
from prompt_toolkit.document import Document
-from mock import Mock
+from unittest.mock import Mock
import pytest
parametrize = pytest.mark.parametrize
diff -Naur pgcli-3.1.0-original/tests/test_completion_refresher.py pgcli-3.1.0/tests/test_completion_refresher.py
--- pgcli-3.1.0-original/tests/test_completion_refresher.py 2020-08-14 18:08:15.000000000 -0400
+++ pgcli-3.1.0/tests/test_completion_refresher.py 2021-03-29 12:48:25.523982262 -0400
@@ -1,6 +1,6 @@
import time
import pytest
-from mock import Mock, patch
+from unittest.mock import Mock, patch
@pytest.fixture
diff -Naur pgcli-3.1.0-original/tests/test_main.py pgcli-3.1.0/tests/test_main.py
--- pgcli-3.1.0-original/tests/test_main.py 2020-08-14 18:08:15.000000000 -0400
+++ pgcli-3.1.0/tests/test_main.py 2021-03-29 12:48:01.263838960 -0400
@@ -1,6 +1,6 @@
import os
import platform
-import mock
+from unittest import mock
import pytest
diff -Naur pgcli-3.1.0-original/tests/test_naive_completion.py pgcli-3.1.0/tests/test_naive_completion.py
--- pgcli-3.1.0-original/tests/test_naive_completion.py 2020-08-14 18:08:15.000000000 -0400
+++ pgcli-3.1.0/tests/test_naive_completion.py 2021-03-29 12:48:28.710001082 -0400
@@ -13,7 +13,7 @@
@pytest.fixture
def complete_event():
- from mock import Mock
+ from unittest.mock import Mock
return Mock()
diff -Naur pgcli-3.1.0-original/tests/test_pgexecute.py pgcli-3.1.0/tests/test_pgexecute.py
--- pgcli-3.1.0-original/tests/test_pgexecute.py 2020-08-14 18:08:15.000000000 -0400
+++ pgcli-3.1.0/tests/test_pgexecute.py 2021-03-29 12:47:49.041766766 -0400
@@ -2,7 +2,7 @@
import psycopg2
import pytest
-from mock import patch, MagicMock
+from unittest.mock import patch, MagicMock
from pgspecial.main import PGSpecial, NO_QUERY
from utils import run, dbtest, requires_json, requires_jsonb
diff -Naur pgcli-3.1.0-original/tests/test_rowlimit.py pgcli-3.1.0/tests/test_rowlimit.py
--- pgcli-3.1.0-original/tests/test_rowlimit.py 2020-08-14 18:08:15.000000000 -0400
+++ pgcli-3.1.0/tests/test_rowlimit.py 2021-03-29 12:48:06.210868182 -0400
@@ -1,5 +1,5 @@
import pytest
-from mock import Mock
+from unittest.mock import Mock
from pgcli.main import PGCli