1 files changed, 88 insertions, 0 deletions
diff --git a/service/test/unit/test_runserver.py b/service/test/unit/test_runserver.py
new file mode 100644
index 00000000..99b502f1
--- /dev/null
+++ b/service/test/unit/test_runserver.py
@@ -0,0 +1,88 @@
+#
+# Copyright (c) 2014 ThoughtWorks, Inc.
+#
+# Pixelated is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Pixelated is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
+
+import unittest
+import sys
+import thread
+import json
+
+import os
+import pixelated.config.app_factory
+import pixelated.runserver
+from mockito import *
+import pixelated.config.app_factory as app_factory
+from leap.common.events import server as events_server
+
+
+class RunserverTest(unittest.TestCase):
+
+    def setUp(self):
+        events_server.ensure_server = lambda port=None: None
+        when(app_factory).create_app().thenReturn(None)
+
+    def tearDown(self):
+        unstub()
+
+    def test_that_config_file_can_be_specified_on_command_line(self):
+        self.config_file_loaded = None
+
+        def _mock_parse_config_from_file(config_file):
+            self.config_file_loaded = config_file
+            return 1, 2, 3
+
+        pixelated.runserver.parse_config_from_file = _mock_parse_config_from_file
+        sys.argv = ['pixelated-user-agent', '--config', 'pixelated.cfg']
+
+        pixelated.runserver.setup()
+
+        self.assertEquals('pixelated.cfg', self.config_file_loaded)
+
+    def test_that_organization_switch_reads_the_credentials_from_pipe(self):
+        fifo_path = '/tmp/credentials-pipe'
+        if os.path.exists(fifo_path):
+            os.remove(fifo_path)
+        test_fifo = os.mkfifo('/tmp/credentials-pipe')
+        thread.start_new_thread(self.spin_up_fifo, (fifo_path,))
+        sys.argv = ['tmp/does_not_exist', '--dispatcher', fifo_path]
+        pixelated.runserver.setup()
+
+    def test_that_organization_switch_reads_the_credentials_from_stdin(self):
+        data = json.dumps({'leap_provider_hostname': 'test_provider', 'user': 'test_user', 'password': 'test_password'})
+        orig_stdin = sys.stdin
+        try:
+            sys.stdin = Mock()
+            when(sys.stdin).read().thenReturn(data)
+
+            sys.argv = ['tmp/does_not_exist', '--dispatcher-stdin']
+            pixelated.runserver.setup()
+
+            self.assertEquals('test_provider', pixelated.runserver.app.config['LEAP_SERVER_NAME'])
+            self.assertEquals('test_user', pixelated.runserver.app.config['LEAP_USERNAME'])
+            self.assertEquals('test_password', pixelated.runserver.app.config['LEAP_PASSWORD'])
+        finally:
+            sys.stdin = orig_stdin
+
+    def test_start_services_pass_args_through(self):
+        args = {}
+        when(app_factory).create_app(any(), args).thenReturn(None)
+
+        pixelated.runserver.start_services(args)
+
+        verify(app_factory).create_app(any(), args)
+
+    def spin_up_fifo(self, test_fifo):
+        with open(test_fifo, 'w') as fifo:
+            fifo.write(json.dumps({'leap_provider_hostname': 'test_provider', 'user': 'test_user', 'password': 'test_password'}))