def mock_app(self, app): expected_response = json.dumps({ "name": app, "ip": "{}.tsuru.io".format(app), "platform": "php", "units": [ { "Address": { "Host": "10.10.10.10" }, "ID": "app1/0", "Status": "stopped" }, ], "pool": "mypool", }) httpretty.register_uri(httpretty.GET, TsuruClientUrls.get_app_url(app), body=expected_response, content_type="application/json", status=200)
def test_get_app_with_success(self): app_name = 'morfeu' expected_response = self.__expected_unit(name=app_name) httpretty.register_uri(httpretty.GET, TsuruClientUrls.get_app_url(app_name), body=expected_response, content_type="application/json", status=200) self.assertEqual(self.tsuru_client.get_app(app_name=app_name), json.loads(expected_response))
def mock_list_apps(self, pool="", processname="web", platform="python", status=200): expected_response = self.__expected_unit(processname=processname, platform=platform) httpretty.register_uri( httpretty.GET, TsuruClientUrls.list_apps_url(pool=pool), body=expected_response, content_type="application/json", status=status, )
def test_get_app_with_timeout(self): def raiseTimeout(request, uri, headers): raise requests.Timeout('Connection timed out.') app_name = 'morfeu' httpretty.register_uri(httpretty.GET, TsuruClientUrls.get_app_url(app_name), body=raiseTimeout, content_type="application/json", status=500) self.assertEqual(self.tsuru_client.get_app(app_name=app_name), json.loads("{}"))
def test_sleep_app(self): url = TsuruClientUrls.get_sleep_url("myapp", "web", TSURU_APP_PROXY_URL) httpretty.register_uri(httpretty.POST, url, content_type="application/json", status=200) self.mock_app("myapp") app = TsuruApp(name="myapp") app.sleep() path = re.search('^http://[^/]+(.*)$', url).group(1) self.assertEqual(httpretty.last_request().path, path)
def test_get_app_with_failure(self): app_name = "morfeu" expected_response = self.__expected_unit(name=app_name) httpretty.register_uri( httpretty.GET, TsuruClientUrls.get_app_url(app_name), body=expected_response, content_type="application/json", status=500, ) self.assertEqual(self.tsuru_client.get_app(app_name=app_name), json.loads("{}"))
def test_list_apps_with_timeout(self): def raiseTimeout(request, uri, headers): raise requests.Timeout('Connection timed out.') httpretty.register_uri(method=httpretty.GET, uri=TsuruClientUrls.list_apps_url(), body=raiseTimeout, content_type="application/json", status=500) self.assertEqual(self.tsuru_client.list_apps(), [])
def test_list_apps_with_timeout(self): def raiseTimeout(request, uri, headers): raise requests.Timeout("Connection timed out.") httpretty.register_uri( method=httpretty.GET, uri=TsuruClientUrls.list_apps_url(), body=raiseTimeout, content_type="application/json", status=500, ) self.assertEqual(self.tsuru_client.list_apps(), [])
def mock_app(self, app): expected_response = json.dumps({ "name": app, "ip": "{}.tsuru.io".format(app), "platform": "php", "units": [ {"Address": {"Host": "10.10.10.10"}, "ID": "app1/0", "Status": "stopped"}, ], "pool": "mypool", }) httpretty.register_uri(httpretty.GET, TsuruClientUrls.get_app_url(app), body=expected_response, content_type="application/json", status=200)
def test_get_app_with_timeout(self): def raiseTimeout(request, uri, headers): raise requests.Timeout("Connection timed out.") app_name = "morfeu" httpretty.register_uri( httpretty.GET, TsuruClientUrls.get_app_url(app_name), body=raiseTimeout, content_type="application/json", status=500, ) self.assertEqual(self.tsuru_client.get_app(app_name=app_name), json.loads("{}"))
def test_sleep_app_with_failure(self): app_name = "morfeu" process_name = "web" proxy_url = "http://fake:123" expected_response = json.dumps("") httpretty.register_uri( httpretty.POST, TsuruClientUrls.get_sleep_url(app_name, process_name, proxy_url), body=expected_response, content_type="application/json", status=500, ) self.assertFalse(self.tsuru_client.sleep_app(app_name=app_name, process_name=process_name, proxy_url=proxy_url))
def test_stop_app(self): url = TsuruClientUrls.get_stop_url("myapp", "web") httpretty.register_uri(httpretty.POST, url, content_type="application/json", status=200) self.mock_app("myapp") app = TsuruApp(name="myapp") app.stop() path = re.search('^http://[^/]+(.*)$', url).group(1) self.assertEqual(httpretty.last_request().path, path)
def test_sleep_app_with_failure(self): app_name = 'morfeu' process_name = "web" proxy_url = 'http://fake:123' expected_response = json.dumps("") httpretty.register_uri(httpretty.POST, TsuruClientUrls.get_sleep_url(app_name, process_name, proxy_url), body=expected_response, content_type="application/json", status=500) self.assertFalse(self.tsuru_client.sleep_app(app_name=app_name, process_name=process_name, proxy_url=proxy_url))
def test_sleep_app_with_timeout(self): def raiseTimeout(request, uri, headers): raise requests.Timeout("Connection timed out.") app_name = "morfeu" process_name = "web" proxy_url = "http://fake:123" httpretty.register_uri( httpretty.POST, TsuruClientUrls.get_sleep_url(app_name, process_name, proxy_url), body=raiseTimeout, content_type="application/json", status=500, ) self.assertFalse(self.tsuru_client.sleep_app(app_name=app_name, process_name=process_name, proxy_url=proxy_url))
def test_sleep_app_with_timeout(self): def raiseTimeout(request, uri, headers): raise requests.Timeout('Connection timed out.') app_name = 'morfeu' process_name = "web" proxy_url = 'http://fake:123' httpretty.register_uri(httpretty.POST, TsuruClientUrls.get_sleep_url(app_name, process_name, proxy_url), body=raiseTimeout, content_type="application/json", status=500) self.assertFalse(self.tsuru_client.sleep_app(app_name=app_name, process_name=process_name, proxy_url=proxy_url))
def mock_list_apps(self, pool="", processname="web", platform="python", status=200): expected_response = self.__expected_unit(processname=processname, platform=platform) httpretty.register_uri(httpretty.GET, TsuruClientUrls.list_apps_url(pool=pool), body=expected_response, content_type="application/json", status=status)