Skip to content

Commit 7716527

Browse files
committed
!squash multiple objects test
1 parent d83685c commit 7716527

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

tests/test_dataclasses.py

+14-2
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,11 @@
33

44
import pytest
55

6-
from libtmux._internal.query_list import QueryList
6+
from libtmux._internal.query_list import (
7+
MultipleObjectsReturned,
8+
ObjectDoesNotExist,
9+
QueryList,
10+
)
711
from libtmux.pane import Pane
812
from libtmux.server import Server
913
from libtmux.session import Session
@@ -161,8 +165,16 @@ def test_querylist(
161165
assert isinstance(w, Window)
162166
assert w.window_name == "test_2"
163167

164-
with pytest.raises(Exception):
168+
with pytest.raises(ObjectDoesNotExist):
165169
w = qs.get(window_name="non_existant")
166170

167171
result = qs.get(window_name="non_existant", default="default_value")
168172
assert result == "default_value"
173+
174+
# Test for multiple objects
175+
server = session.server
176+
second_session = server.new_session("second session")
177+
second_session.new_window(window_name="test_2")
178+
assert len(server.windows.filter(window_name="test_2")) == 2
179+
with pytest.raises(MultipleObjectsReturned):
180+
server.windows.get(window_name="test_2")

0 commit comments

Comments
 (0)