def test_before_deploy_fails(self, mock_is_timedout):
"""
Test that the correct database entires are made by BeforeDeploy worker
when the worker fails on its actions
"""
# Worker receives a packet, most likely from the webapp
# Example packet:
#
# {
# 'application': 'staging',
# '....': '....',
# }
#
#
packet = {
'environment': 'staging',
'application': 'adsws',
'version': 'v1.0.0',
}
# Override the run test returned value. This means the logic of the test
# does not have to be mocked
mock_is_timedout.return_value = True
with MiniRabbit(RABBITMQ_URL) as w:
w.publish(route='in', exchange='test', payload=json.dumps(packet))
# Worker runs the tests
params = {
'RABBITMQ_URL': RABBITMQ_URL,
'exchange': 'test',
'subscribe': 'in',
'publish': 'ads.deploy.deploy',
'header_frame': None,
'error': 'error',
'status': 'database',
'TEST_RUN': True
}
before_deploy_worker = BeforeDeploy(params=params)
before_deploy_worker.run()
before_deploy_worker.connection.close()
# Worker sends a packet to the next worker
with MiniRabbit(RABBITMQ_URL) as w:
self.assertEqual(w.message_count('in'), 0)
self.assertEqual(w.message_count('ads.deploy.deploy'), 0)
self.assertEqual(w.message_count('database'), 1)
self.assertEqual(w.message_count('error'), 1)
# Start the DB Writer worker
params = {
'RABBITMQ_URL': RABBITMQ_URL,
'exchange': 'test',
'subscribe': 'database',
'TEST_RUN': True
}
db_worker = DatabaseWriterWorker(params=params)
db_worker.app = self.app
db_worker.run()
db_worker.connection.close()
with self.app.session_scope() as session:
all_deployments = session.query(Deployment).all()
self.assertEqual(
len(all_deployments),
1,
msg='More (or less) than 1 deployment entry: {}'
.format(all_deployments)
)
deployment = all_deployments[0]
for key in packet:
self.assertEqual(
packet[key],
getattr(deployment, key)
)
self.assertEqual(deployment.deployed, False)
self.assertEqual(
deployment.msg,
'BeforeDeploy: waiting too long for the environment to come up'
)
def test_before_deploy_fails(self, mock_is_timedout):
"""
Test that the correct database entires are made by BeforeDeploy worker
when the worker fails on its actions
"""
# Worker receives a packet, most likely from the webapp
# Example packet:
#
# {
# 'application': 'staging',
# '....': '....',
# }
#
#
packet = {"environment": "staging", "application": "adsws", "tag": "v1.0.0", "commit": "gf9gd8f"}
# Override the run test returned value. This means the logic of the test
# does not have to be mocked
mock_is_timedout.return_value = True
with MiniRabbit(RABBITMQ_URL) as w:
w.publish(route="in", exchange="test", payload=json.dumps(packet))
# Worker runs the tests
params = {
"RABBITMQ_URL": RABBITMQ_URL,
"exchange": "test",
"subscribe": "in",
"publish": "ads.deploy.deploy",
"header_frame": None,
"error": "error",
"status": "database",
"TEST_RUN": True,
}
before_deploy_worker = BeforeDeploy(params=params)
before_deploy_worker.run()
before_deploy_worker.connection.close()
# Worker sends a packet to the next worker
with MiniRabbit(RABBITMQ_URL) as w:
self.assertEqual(w.message_count("in"), 0)
self.assertEqual(w.message_count("ads.deploy.deploy"), 0)
self.assertEqual(w.message_count("database"), 1)
self.assertEqual(w.message_count("error"), 1)
# Start the DB Writer worker
params = {"RABBITMQ_URL": RABBITMQ_URL, "exchange": "test", "subscribe": "database", "TEST_RUN": True}
db_worker = DatabaseWriterWorker(params=params)
db_worker.app = self.app
db_worker.run()
db_worker.connection.close()
with self.app.session_scope() as session:
all_deployments = session.query(Deployment).all()
self.assertEqual(
len(all_deployments), 1, msg="More (or less) than 1 deployment entry: {}".format(all_deployments)
)
deployment = all_deployments[0]
for key in packet:
self.assertEqual(packet[key], getattr(deployment, key))
self.assertEqual(deployment.deployed, False)
self.assertEqual(deployment.msg, "BeforeDeploy: waiting too long for the environment to come up")
def test_before_deploy_receives_restart(self, mock_executioner):
"""
Test that the correct database entires are made by BeforeDeploy worker
when the worker succeeds on its actions
"""
# Worker receives a packet, most likely from the webapp
# Example packet:
#
# {
# 'application': 'staging',
# '....': '....',
# }a
#
#
packet = {
'environment': 'staging',
'application': 'adsws',
'action': 'restart'
}
# Override the run test returned value. This means the logic of the test
# does not have to be mocked
mock_r = mock.Mock(retcode=0)
mock_r.out.splitlines.return_value = []
mock_x = mock_executioner.return_value
mock_x.cmd.return_value = mock_r
with MiniRabbit(RABBITMQ_URL) as w:
w.publish(route='in', exchange='test', payload=json.dumps(packet))
# Worker runs the tests
params = {
'RABBITMQ_URL': RABBITMQ_URL,
'exchange': 'test',
'subscribe': 'in',
'publish': 'ads.deploy.deploy',
'header_frame': None,
'status': 'database',
'error': 'error',
'status': 'database',
'TEST_RUN': True
}
before_deploy_worker = BeforeDeploy(params=params)
before_deploy_worker.run()
before_deploy_worker.connection.close()
# Worker sends a packet to the next worker
with MiniRabbit(RABBITMQ_URL) as w:
self.assertEqual(w.message_count('in'), 0)
self.assertEqual(w.message_count('ads.deploy.deploy'), 0)
self.assertEqual(w.message_count('ads.deploy.restart'), 1)
self.assertEqual(w.message_count('database'), 1)
self.assertEqual(w.message_count('error'), 0)
# Start the DB Writer worker
params = {
'RABBITMQ_URL': RABBITMQ_URL,
'exchange': 'test',
'subscribe': 'database',
'TEST_RUN': True
}
db_worker = DatabaseWriterWorker(params=params)
db_worker.app = self.app
db_worker.run()
db_worker.connection.close()
# remove irrelevant keys before checking things
packet.pop('action')
# check database entries
with self.app.session_scope() as session:
all_deployments = session.query(Deployment).all()
self.assertEqual(
len(all_deployments),
1,
msg='More (or less) than 1 deployment entry: {}'
.format(all_deployments)
)
deployment = all_deployments[0]
for key in packet:
self.assertEqual(
packet[key],
getattr(deployment, key)
)
self.assertEqual(deployment.deployed, None)
self.assertEqual(
deployment.msg,
'Deploy to be restarted'
)
def test_before_deploy_receives_restart(self, mock_executioner):
"""
Test that the correct database entires are made by BeforeDeploy worker
when the worker succeeds on its actions
"""
# Worker receives a packet, most likely from the webapp
# Example packet:
#
# {
# 'application': 'staging',
# '....': '....',
# }a
#
#
packet = {
"environment": "staging",
"application": "adsws",
"tag": "v1.0.0",
"commit": "gf9gd8f",
"action": "restart",
}
# Override the run test returned value. This means the logic of the test
# does not have to be mocked
mock_r = mock.Mock(retcode=0)
mock_r.out.splitlines.return_value = []
mock_x = mock_executioner.return_value
mock_x.cmd.return_value = mock_r
with MiniRabbit(RABBITMQ_URL) as w:
w.publish(route="in", exchange="test", payload=json.dumps(packet))
# Worker runs the tests
params = {
"RABBITMQ_URL": RABBITMQ_URL,
"exchange": "test",
"subscribe": "in",
"publish": "ads.deploy.deploy",
"header_frame": None,
"status": "database",
"error": "error",
"status": "database",
"TEST_RUN": True,
}
before_deploy_worker = BeforeDeploy(params=params)
before_deploy_worker.run()
before_deploy_worker.connection.close()
# Worker sends a packet to the next worker
with MiniRabbit(RABBITMQ_URL) as w:
self.assertEqual(w.message_count("in"), 0)
self.assertEqual(w.message_count("ads.deploy.deploy"), 0)
self.assertEqual(w.message_count("ads.deploy.restart"), 1)
self.assertEqual(w.message_count("database"), 1)
self.assertEqual(w.message_count("error"), 0)
# Start the DB Writer worker
params = {"RABBITMQ_URL": RABBITMQ_URL, "exchange": "test", "subscribe": "database", "TEST_RUN": True}
db_worker = DatabaseWriterWorker(params=params)
db_worker.app = self.app
db_worker.run()
db_worker.connection.close()
# remove irrelevant keys before checking things
packet.pop("action")
# check database entries
with self.app.session_scope() as session:
all_deployments = session.query(Deployment).all()
self.assertEqual(
len(all_deployments), 1, msg="More (or less) than 1 deployment entry: {}".format(all_deployments)
)
deployment = all_deployments[0]
for key in packet:
self.assertEqual(packet[key], getattr(deployment, key))
self.assertEqual(deployment.deployed, None)
self.assertEqual(deployment.msg, "Deploy to be restarted")
