def setUp(self):
self.status = MagicMock(spec=Table)
self.assignments = MagicMock(spec=Table)
self.revisions = MagicMock(spec=Table)
self.units = MagicMock(spec=Table)
self.kms = MagicMock(spec=KMSConnection)
self.instance_id = 'i-123456'
self.service = 'mock-service'
self.db = FlotillaAgentDynamo(self.instance_id, self.service,
self.status, self.assignments,
self.revisions, self.units, self.kms)
self.assignments.batch_get.return_value = [{'assignment': ASSIGNED}]
self.revision = {
'units': [UNIT_1_HASH, UNIT_2_HASH, UNIT_3_HASH],
'rev_hash': ASSIGNED,
'label': 'test'
}
self.revisions.batch_get.return_value = [self.revision]
self.units.batch_get.return_value = [{
'name': '1',
'unit_file': '',
'environment': '',
'unit_hash': UNIT_1_HASH
}, {
'name': '2',
'unit_file': '',
'environment': '',
'unit_hash': UNIT_2_HASH
}, {
'name': '3',
'unit_file': '',
'environment': '',
'unit_hash': UNIT_3_HASH
}]
