def test_get_keys_gateway(self, dynamo, tables, db):
db.return_value = self.db
get_keys(ENVIRONMENT, REGION, None, True)
self.db.get_bastion_users.assert_called_with()
def test_get_keys_scheduler(self, dynamo, tables, db):
db.return_value = self.db
get_keys(ENVIRONMENT, REGION, None, False)
self.db.get_region_admins.assert_called_with()
def test_get_keys_service(self, dynamo, tables, db):
db.return_value = self.db
get_keys(ENVIRONMENT, REGION, SERVICE, False)
self.db.get_service_admins.assert_called_with(SERVICE)
