def start():
client = crane_ec2.Client()
signal.signal(signal.SIGHUP, huphandler)
signal.signal(signal.SIGTERM, termhandler)
creator.set_model(Instance)
creator.build_queue()
creator.start_creator(DatabaseManager, client)
def start():
client = crane_ec2.Client()
signal.signal(signal.SIGHUP, huphandler)
signal.signal(signal.SIGTERM, termhandler)
creator.set_model(Instance)
creator.build_queue()
creator.start_creator(DatabaseManager, client)
def test_create_database_terminates_the_instance_when_cant_create_db(self):
exc_msg = u"I've failed to create your database, sorry! :("
module = "mysqlapi.api.models.DatabaseManager.create_database"
with mock.patch(module) as c_database:
c_database.side_effect = Exception(exc_msg)
instance = Instance(
ec2_id="i-00009",
name="home",
host="unknown.host",
state="running",
)
ec2_client = mocks.FakeEC2Client()
try:
t = start_creator(DatabaseManager, ec2_client)
create_database(instance, ec2_client)
t.stop()
self.assertIn("unauthorize instance home", ec2_client.actions)
self.assertIn("terminate instance home", ec2_client.actions)
index_unauthorize = ec2_client.actions.index(
"unauthorize instance home"
)
index_terminate = ec2_client.actions.index(
"terminate instance home"
)
msg = "Should unauthorize before terminate."
assert index_unauthorize < index_terminate, msg
self.assertIsNotNone(instance.pk)
self.assertEqual("error", instance.state)
self.assertEqual(exc_msg, instance.reason)
finally:
instance.delete()
def test_create_database_terminates_the_instance_when_cant_create_db(self):
exc_msg = u"I've failed to create your database, sorry! :("
module = "mysqlapi.api.models.DatabaseManager.create_database"
with mock.patch(module) as c_database:
c_database.side_effect = Exception(exc_msg)
instance = Instance(
ec2_id="i-00009",
name="home",
host="unknown.host",
state="running",
)
ec2_client = mocks.FakeEC2Client()
try:
t = start_creator(DatabaseManager, ec2_client)
create_database(instance, ec2_client)
t.stop()
self.assertIn("unauthorize instance home", ec2_client.actions)
self.assertIn("terminate instance home", ec2_client.actions)
index_unauthorize = ec2_client.actions.index(
"unauthorize instance home")
index_terminate = ec2_client.actions.index(
"terminate instance home")
msg = "Should unauthorize before terminate."
assert index_unauthorize < index_terminate, msg
self.assertIsNotNone(instance.pk)
self.assertEqual("error", instance.state)
self.assertEqual(exc_msg, instance.reason)
finally:
instance.delete()
def test_create_database_should_call_run_from_client(self):
try:
cli = mocks.FakeEC2Client()
t = start_creator(DatabaseManager, cli)
data = {"name": "bowl", "service_host": "127.0.0.1"}
request = RequestFactory().post("/", data)
view = CreateDatabase()
view._client = cli
response = view.post(request)
t.stop()
self.assertEqual(201, response.status_code)
self.assertIn("run instance bowl", cli.actions)
finally:
self.cursor.execute("DROP DATABASE IF EXISTS bowl")
def test_create_database_should_authorize_access_to_the_instance(self):
try:
cli = mocks.FakeEC2Client()
t = start_creator(DatabaseManager, cli)
data = {"name": "entre_nous", "service_host": "127.0.0.1"}
request = RequestFactory().post("/", data)
view = CreateDatabase()
view._client = cli
response = view.post(request)
t.stop()
self.assertEqual(201, response.status_code)
self.assertIn("authorize instance entre_nous", cli.actions)
finally:
self.cursor.execute("DROP DATABASE IF EXISTS entre_nous")
def test_create_database_should_call_run_from_client(self):
try:
cli = mocks.FakeEC2Client()
t = start_creator(DatabaseManager, cli)
data = {"name": "bowl", "service_host": "127.0.0.1"}
request = RequestFactory().post("/", data)
view = CreateDatabase()
view._client = cli
response = view.post(request)
t.stop()
self.assertEqual(201, response.status_code)
self.assertIn("run instance bowl", cli.actions)
finally:
self.cursor.execute("DROP DATABASE IF EXISTS bowl")
def test_create_database_should_authorize_access_to_the_instance(self):
try:
cli = mocks.FakeEC2Client()
t = start_creator(DatabaseManager, cli)
data = {"name": "entre_nous", "service_host": "127.0.0.1"}
request = RequestFactory().post("/", data)
view = CreateDatabase()
view._client = cli
response = view.post(request)
t.stop()
self.assertEqual(201, response.status_code)
self.assertIn("authorize instance entre_nous", cli.actions)
finally:
self.cursor.execute("DROP DATABASE IF EXISTS entre_nous")
def test_create_database_ec2(self):
try:
client = mocks.FakeEC2Client()
t = start_creator(DatabaseManager, client)
request = RequestFactory().post("/", {"name": "ciclops"})
view = CreateDatabase()
view._client = client
response = view.post(request)
self.assertEqual(201, response.status_code)
self.assertEqual("", response.content)
t.stop()
sql = "select SCHEMA_NAME from information_schema.SCHEMATA " + \
"where SCHEMA_NAME = 'ciclops'"
self.cursor.execute(sql)
row = self.cursor.fetchone()
self.assertEqual("ciclops", row[0])
finally:
db = DatabaseManager("ciclops")
db.drop_database()
def test_create_database_ec2(self):
try:
client = mocks.FakeEC2Client()
t = start_creator(DatabaseManager, client)
request = RequestFactory().post("/", {"name": "ciclops"})
view = CreateDatabase()
view._client = client
response = view.post(request)
self.assertEqual(201, response.status_code)
self.assertEqual("", response.content)
t.stop()
sql = "select SCHEMA_NAME from information_schema.SCHEMATA " + \
"where SCHEMA_NAME = 'ciclops'"
self.cursor.execute(sql)
row = self.cursor.fetchone()
self.assertEqual("ciclops", row[0])
finally:
db = DatabaseManager("ciclops")
db.drop_database()
def test_create_database_terminates_the_instance_when_cant_authorize(self):
instance = Instance(
ec2_id="i-00009",
name="home",
host="unknown.host",
state="running",
)
ec2_client = mocks.FakeEC2Client()
ec2_client.authorize = lambda *args, **kwargs: False
try:
t = start_creator(DatabaseManager, ec2_client)
create_database(instance, ec2_client)
t.stop()
self.assertIn("terminate instance home", ec2_client.actions)
self.assertIsNotNone(instance.pk)
self.assertEqual("error", instance.state)
reason = "Failed to authorize access to the instance."
self.assertEqual(reason, instance.reason)
finally:
instance.delete()
def test_create_database_terminates_the_instance_when_cant_authorize(self):
instance = Instance(
ec2_id="i-00009",
name="home",
host="unknown.host",
state="running",
)
ec2_client = mocks.FakeEC2Client()
ec2_client.authorize = lambda *args, **kwargs: False
try:
t = start_creator(DatabaseManager, ec2_client)
create_database(instance, ec2_client)
t.stop()
self.assertIn("terminate instance home", ec2_client.actions)
self.assertIsNotNone(instance.pk)
self.assertEqual("error", instance.state)
reason = "Failed to authorize access to the instance."
self.assertEqual(reason, instance.reason)
finally:
instance.delete()
def test_create_database_sends_the_instance_to_the_queue(self):
instance = Instance(
ec2_id="i-00009",
name="der_trommler",
host="127.0.0.1",
state="running",
)
ec2_client = mocks.MultipleFailureEC2Client(times=0)
try:
t = start_creator(DatabaseManager, ec2_client)
create_database(instance, ec2_client)
t.stop()
sql = "select SCHEMA_NAME from information_schema.SCHEMATA " +\
"where SCHEMA_NAME = 'der_trommler'"
self.cursor.execute(sql)
row = self.cursor.fetchone()
self.assertIsNotNone(row)
self.assertEqual("der_trommler", row[0])
self.assertIsNotNone(instance.pk)
finally:
self.cursor.execute("DROP DATABASE IF EXISTS der_trommler")
instance.delete()
def test_create_database_sends_the_instance_to_the_queue(self):
instance = Instance(
ec2_id="i-00009",
name="der_trommler",
host="127.0.0.1",
state="running",
)
ec2_client = mocks.MultipleFailureEC2Client(times=0)
try:
t = start_creator(DatabaseManager, ec2_client)
create_database(instance, ec2_client)
t.stop()
sql = "select SCHEMA_NAME from information_schema.SCHEMATA " +\
"where SCHEMA_NAME = 'der_trommler'"
self.cursor.execute(sql)
row = self.cursor.fetchone()
self.assertIsNotNone(row)
self.assertEqual("der_trommler", row[0])
self.assertIsNotNone(instance.pk)
finally:
self.cursor.execute("DROP DATABASE IF EXISTS der_trommler")
instance.delete()
