More tests for Server

Author: tony

tmuxp/server.py

@@ -39,10 +39,17 @@ class Server(TmuxRelationalObject):
     socket_name = None
     socket_path = None
     config_file = None
+    colors = None
     childIdAttribute = 'session_id'
 
-    def __init__(self, socket_name=None, socket_path=None, config_file=None,
-                 **kwargs):
+    def __init__(
+        self,
+        socket_name=None,
+        socket_path=None,
+        config_file=None,
+        colors=None,
+        **kwargs
+    ):
         self._windows = []
         self._panes = []
 
@@ -55,6 +62,9 @@ def __init__(self, socket_name=None, socket_path=None, config_file=None,
         if config_file:
             self.config_file = config_file
 
+        if colors:
+            self.colors = colors
+
     def tmux(self, *args, **kwargs):
         args = list(args)
         if self.socket_name:
@@ -63,6 +73,13 @@ def tmux(self, *args, **kwargs):
             args.insert(0, '-S{}'.format(self.socket_path))
         if self.config_file:
             args.insert(0, '-f{}'.format(self.config_file))
+        if self.colors:
+            if self.colors == 256:
+                args.insert(0, '-2')
+            elif self.colors == 88:
+                args.insert(0, '-8')
+            else:
+                raise ValueError('Server.colors must equal 88 or 256')
 
         return tmux(*args, **kwargs)
 
@@ -437,14 +454,14 @@ def new_session(self,
         :type kill_session: bool
         '''
 
-        # ToDo: Update below to work with attach_if_exists
         if self.has_session(session_name):
             if kill_session:
                 self.tmux('kill-session', '-t%s' % session_name)
                 logger.info('session %s exists. killed it.' % session_name)
             else:
                 raise TmuxSessionExists(
-                    'Session named %s exists' % session_name)
+                    'Session named %s exists' % session_name
+                )
 
         logger.debug('creating session %s' % session_name)
 
@@ -485,6 +502,4 @@ def new_session(self,
 
         session = Session(server=self, **session)
 
-        # self._sessions.append(session)
-
         return session

tmuxp/testsuite/test_server.py

@@ -44,5 +44,12 @@ def test_config(self):
         myserver = Server(config_file='test')
         self.assertEqual(myserver.config_file, 'test')
 
+    def test_256_colors(self):
+        myserver = Server(colors=256)
+        self.assertEqual(myserver.colors, 256)
+
+        # try to read the command in  util.tmux()
+        # get util.tmux  to have a self.cmd and create tests for it
+
 if __name__ == '__main__':
     unittest.main()