Skip to content

Commit

Permalink
added more tests for server interface
Browse files Browse the repository at this point in the history
  • Loading branch information
amirreza8002 committed Dec 27, 2024
1 parent 0047135 commit ce38036
Showing 1 changed file with 31 additions and 0 deletions.
31 changes: 31 additions & 0 deletions tests/test_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@

from django_valkey.cache import ValkeyCache
from django_valkey.client import DefaultClient, ShardClient
from django_valkey.server.sync_server.client import DefaultClient as ServerClient


@pytest.fixture
Expand Down Expand Up @@ -58,6 +59,36 @@ def test_close_disconnect_client_options(
assert mock.called


class TestServerClient:
@patch("tests.test_client.ServerClient.get_client")
@patch("tests.test_client.ServerClient.__init__", return_value=None)
def test_delete_pattern_calls_get_client_given_no_client(
self, init_mock, get_client_mock
):
client = ServerClient()
client._backend = Mock()
client._backend.key_prefix = ""

client.delete_pattern(pattern="foo*")
get_client_mock.assert_called_once_with(write=True, tried=None)

@patch("tests.test_client.ServerClient.make_pattern")
@patch("tests.test_client.ServerClient.get_client")
@patch("tests.test_client.ServerClient.__init__", return_value=None)
def test_delete_pattern_calls_make_pattern(
self, init_mock, get_client_mock, make_pattern_mock
):
client = DefaultClient()
client._backend = Mock()
client._backend.key_prefix = ""
get_client_mock.return_value.scan_iter.return_value = []

client.delete_pattern(pattern="foo*")

kwargs = {"version": None, "prefix": None}
make_pattern_mock.assert_called_once_with("foo*", **kwargs)


class TestDefaultClient:
@patch("tests.test_client.DefaultClient.get_client")
@patch("tests.test_client.DefaultClient.__init__", return_value=None)
Expand Down

0 comments on commit ce38036

Please sign in to comment.