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_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_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()