def setUp(self):
self.tsuru_client = TsuruClient()
import requests
import logging
from morfeu.tsuru.client import TsuruClient
from morfeu.settings import TIME_RANGE_IN_HOURS, ESEARCH_HOST, TIMEOUT
LOG = logging.getLogger(__name__)
tsuru_client = TsuruClient()
class TsuruApp(object):
def __init__(self, name=None, dry=False):
self.dry = dry
self.name = name
self.timeout = TIMEOUT
self.ip = None
self.pool = None
self.__load_info()
def __unicode__(self):
return u"{0}".format(self.name)
def __load_info(self):
app_info = tsuru_client.get_app(self.name)
self.ip = app_info.get("ip")
self.pool = app_info.get("pool")
def sleep(self):
if self.dry:
class MorfeuTsuruClientTestCase(unittest.TestCase):
def setUp(self):
self.tsuru_client = TsuruClient()
def tearDown(self):
self.tsuru_client = None
def raiseTimeout(request, uri, headers):
raise requests.Timeout("Connection timed out.")
def __expected_unit(self, name="app1", platform="python", processname="web"):
expected_response = json.dumps(
[
{
"ip": "10.10.10.10",
"name": name,
"platform": platform,
"units": [{"ID": "app1/0", "Status": "started", "ProcessName": processname}],
}
]
)
return 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,
)
@httpretty.activate
def test_list_apps_with_success(self):
self.mock_list_apps(processname="web")
self.assertEqual(self.tsuru_client.list_apps(), [{u"app1": [u"app1/0"]}])
self.mock_list_apps(processname="web")
self.assertEqual(self.tsuru_client.list_apps(), [{u"app1": [u"app1/0"]}])
@httpretty.activate
def test_list_apps_by_domain(self):
self.mock_list_apps()
self.assertEqual(self.tsuru_client.list_apps(domain="11"), [])
@httpretty.activate
def test_list_static_apps(self):
"""
static apps should not be put slepp. it should return an empty list
"""
self.mock_list_apps(platform="static")
self.assertEqual(self.tsuru_client.list_apps(), [{u"app1": [u"app1/0"]}])
@httpretty.activate
def test_list_apps_by_pool(self):
os.environ["POOLS"] = "green"
self.mock_list_apps(pool="green")
self.assertEqual(self.tsuru_client.list_apps(), [{u"app1": [u"app1/0"]}])
@httpretty.activate
def test_list_apps_no_web_apps(self):
self.mock_list_apps(processname="worker")
self.assertEqual(self.tsuru_client.list_apps(), [])
@httpretty.activate
def test_list_apps_with_failure(self):
self.mock_list_apps(processname="web", status=500)
self.assertEqual(self.tsuru_client.list_apps(), [])
@httpretty.activate
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(), [])
@httpretty.activate
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))
@httpretty.activate
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("{}"))
@httpretty.activate
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("{}"))
@httpretty.activate
def test_get_app_without_app_name(self):
self.assertEqual(self.tsuru_client.get_app(), json.loads("{}"))
@httpretty.activate
def test_sleep_app_with_success(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=200,
)
self.assertTrue(self.tsuru_client.sleep_app(app_name=app_name, process_name=process_name, proxy_url=proxy_url))
@httpretty.activate
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))
@httpretty.activate
def test_sleep_app_without_app_name(self):
self.assertFalse(self.tsuru_client.sleep_app())
@httpretty.activate
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))
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Morfeu.. put apps to sleep')
parser.add_argument('--dry', action='store_true',
help='Just pretending...')
parser.add_argument('--daemon', action='store_true',
help='Should be daemonized?')
args = parser.parse_args()
parser = SafeConfigParser()
dry = args.dry
daemon = args.daemon
tsuru_client = TsuruClient()
while True:
try:
LOG.info("Running morfeu...")
apps_to_sleep = []
apps = tsuru_client.list_apps(type="web", domain=DOMAIN)
never_sleep = [TSURU_APP_PROXY] + SKIP_APPS
apps = [app for app in apps if set(app.keys()).intersection(never_sleep) == set([])]
threads = []
for app in apps:
app_name = app.keys()[0]
parser = argparse.ArgumentParser(description='Morfeu.. put apps to sleep')
parser.add_argument('--dry',
action='store_true',
help='Just pretending...')
parser.add_argument('--daemon',
action='store_true',
help='Should be daemonized?')
args = parser.parse_args()
parser = SafeConfigParser()
dry = args.dry
daemon = args.daemon
tsuru_client = TsuruClient()
while True:
try:
LOG.info("Running morfeu...")
apps_to_sleep = []
apps = tsuru_client.list_apps(type="web", domain=DOMAIN)
never_sleep = [TSURU_APP_PROXY] + SKIP_APPS
apps = [
app for app in apps
if set(app.keys()).intersection(never_sleep) == set([])
]
threads = []
class MorfeuTsuruClientTestCase(unittest.TestCase):
def setUp(self):
self.tsuru_client = TsuruClient()
def tearDown(self):
self.tsuru_client = None
def raiseTimeout(request, uri, headers):
raise requests.Timeout('Connection timed out.')
def __expected_unit(self, name="app1", platform="python", processname="web"):
expected_response = json.dumps([{
"ip": "10.10.10.10",
"name": name,
"platform": platform,
"units": [{"ID": "app1/0", "Status": "started", "ProcessName": processname}]
}])
return 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)
@httpretty.activate
def test_list_apps_with_success(self):
self.mock_list_apps(processname="web")
self.assertEqual(self.tsuru_client.list_apps(), [{u'app1': [u'app1/0']}])
self.mock_list_apps(processname="web")
self.assertEqual(self.tsuru_client.list_apps(), [{u'app1': [u'app1/0']}])
@httpretty.activate
def test_list_apps_by_domain(self):
self.mock_list_apps()
self.assertEqual(self.tsuru_client.list_apps(domain="11"), [])
@httpretty.activate
def test_list_static_apps(self):
"""
static apps should not be put slepp. it should return an empty list
"""
self.mock_list_apps(platform="static")
self.assertEqual(self.tsuru_client.list_apps(), [{u'app1': [u'app1/0']}])
@httpretty.activate
def test_list_apps_by_pool(self):
os.environ["POOLS"] = "green"
self.mock_list_apps(pool="green")
self.assertEqual(self.tsuru_client.list_apps(), [{u'app1': [u'app1/0']}])
@httpretty.activate
def test_list_apps_no_web_apps(self):
self.mock_list_apps(processname="worker")
self.assertEqual(self.tsuru_client.list_apps(), [])
@httpretty.activate
def test_list_apps_with_failure(self):
self.mock_list_apps(processname="web", status=500)
self.assertEqual(self.tsuru_client.list_apps(), [])
@httpretty.activate
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(), [])
@httpretty.activate
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))
@httpretty.activate
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("{}"))
@httpretty.activate
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("{}"))
@httpretty.activate
def test_get_app_without_app_name(self):
self.assertEqual(self.tsuru_client.get_app(), json.loads("{}"))
@httpretty.activate
def test_sleep_app_with_success(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=200)
self.assertTrue(self.tsuru_client.sleep_app(app_name=app_name,
process_name=process_name,
proxy_url=proxy_url))
@httpretty.activate
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))
@httpretty.activate
def test_sleep_app_without_app_name(self):
self.assertFalse(self.tsuru_client.sleep_app())
@httpretty.activate
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 setUp(self):
self.tsuru_client = TsuruClient()
