Add session to all Session cmds, if it is not explicitly set

Author: askedrelic

libtmux/session.py

@@ -72,8 +72,13 @@ def cmd(self, *args, **kwargs):
             Renamed from ``.tmux`` to ``.cmd``.
 
         """
-        if '-t' not in kwargs:
-            kwargs['-t'] = self.id
+        # if -t is not set in any arg yet
+        if not any('-t' in unicode(x) for x in args):
+            # insert -t immediately after 1st arg, as per tmux format
+            new_args = [args[0]]
+            new_args += ['-t', self.id]
+            new_args += args[1:]
+            args = new_args
         return self.server.cmd(*args, **kwargs)
 
     def attach_session(self):

tests/test_session.py

@@ -235,3 +235,11 @@ def test_periods_raise_badsessionname(server, session, session_name, raises):
         session.rename_session(new_name)
         with pytest.raises(exc.LibTmuxException):
             server.switch_client(new_name)
+
+def test_cmd_inserts_sesion_id(session):
+    current_session_id = session.id
+    last_arg = 'last-arg'
+    cmd = session.cmd('not-a-command', last_arg)
+    assert '-t' in cmd.cmd
+    assert current_session_id in cmd.cmd
+    assert cmd.cmd[-1] == last_arg