def test_find_by_date_interval_given_no_events_on_time_range(self):
db.session.add(self._create_event(date=datetime.now() + timedelta(days=10)))
db.session.add(self._create_event(date=datetime.now() - timedelta(days=10)))
start_date = datetime.now() - timedelta(days=1)
end_date = datetime.now() + timedelta(days=1)
self.assertEquals(0, len(Event.find_all_by({'start_date': start_date, 'end_date': end_date}).items))
def test_find_by_account(self):
db.session.add(self._create_event(account='account_a'))
db.session.add(self._create_event(account='account_b'))
self.assertEquals(1,
len(Event.find_all_by({
'account': 'account_b'
}).items))
def get(self, region):
args = self._parse_args()
result = Event.find_all_by(dict(args, **{'region': region}), args.page,
args.page_size)
return {
'count': result.total,
'events': EventSchema(many=True).dump(result.items).data
}
def _create_event(self,
region='reg',
account='adm',
resource_id='1',
event_key='USER.LOGIN',
date=datetime.now()):
return Event(event_key,
resource_id=resource_id,
description='User has logged',
username='******',
account=account,
date=date,
region=region,
original_event='{"action":"USER.LOGIN"}')
def test_find_by_date_interval(self):
db.session.add(self._create_event(date=datetime.now() +
timedelta(days=10)))
db.session.add(self._create_event(date=datetime.now()))
start_date = datetime.now() - timedelta(days=1)
end_date = datetime.now() + timedelta(days=1)
self.assertEquals(
1,
len(
Event.find_all_by({
'start_date': start_date,
'end_date': end_date
}).items))
def test_find_by_type(self):
db.session.add(self._create_event(event_key='LB.CREATE'))
db.session.add(self._create_event(event_key='VM.CREATE'))
self.assertEquals(1, len(Event.find_all_by({'type': 'VM'}).items))
def test_find_by_action(self):
db.session.add(self._create_event(event_key='LB.CREATE'))
db.session.add(self._create_event(event_key='LB.DELETE'))
self.assertEquals(1, len(Event.find_all_by({'action': 'DELETE'}).items))
def test_find_by_resource_id(self):
db.session.add(self._create_event(resource_id='1'))
db.session.add(self._create_event(resource_id='2'))
self.assertEquals(1, len(Event.find_all_by({'resource_id': '2'}).items))
def test_find_by_account(self):
db.session.add(self._create_event(account='account_a'))
db.session.add(self._create_event(account='account_b'))
self.assertEquals(1, len(Event.find_all_by({'account': 'account_b'}).items))
def test_find_by_region(self):
db.session.add(self._create_event(region='reg1'))
db.session.add(self._create_event(region='reg2'))
self.assertEquals(1, len(Event.find_all_by({'region': 'reg1'}).items))
def get(self, region):
args = self._parse_args()
result = Event.find_all_by(dict(args, **{'region': region}), args.page, args.page_size)
return {'count': result.total, 'events': EventSchema(many=True).dump(result.items).data}
def test_find_by_username(self):
db.session.add(self._create_event(username='******'))
db.session.add(self._create_event(username='******'))
self.assertEquals(1, len(Event.find_all_by({'username': '******'}).items))
def test_find_by_type(self):
db.session.add(self._create_event(event_key='LB.CREATE'))
db.session.add(self._create_event(event_key='VM.CREATE'))
self.assertEquals(1, len(Event.find_all_by({'type': 'VM'}).items))
def test_find_by_action(self):
db.session.add(self._create_event(event_key='LB.CREATE'))
db.session.add(self._create_event(event_key='LB.DELETE'))
self.assertEquals(1, len(Event.find_all_by({'action': 'DELETE'}).items))
def test_find_by_resource_id(self):
db.session.add(self._create_event(resource_id='1'))
db.session.add(self._create_event(resource_id='2'))
self.assertEquals(1, len(Event.find_all_by({'resource_id': '2'}).items))
def test_find_by_region(self):
db.session.add(self._create_event(region='reg1'))
db.session.add(self._create_event(region='reg2'))
self.assertEquals(1, len(Event.find_all_by({'region': 'reg1'}).items))
