зеркало из https://github.com/mozilla/MozDef.git
Fixup dashboard geomodel alert action
This commit is contained in:
Родитель
e30f3f1d69
Коммит
3345587f32
|
@ -107,8 +107,8 @@ class message(object):
|
|||
previous_location_str = u'{0}, {1}'.format(previous_city, previous_country)
|
||||
|
||||
alert_record = {
|
||||
'alert_id': b2a_hex(os.urandom(15)),
|
||||
'alert_code': b2a_hex(self.alert_classname),
|
||||
'alert_id': b2a_hex(os.urandom(15)).decode(),
|
||||
'alert_code': b2a_hex(self.alert_classname.encode()).decode(),
|
||||
'user_id': auth_full_username,
|
||||
'risk': self.config['risk'],
|
||||
'summary': summary,
|
||||
|
|
|
@ -103,7 +103,7 @@ class TestDashboardGeomodel(object):
|
|||
assert self.test_connect_called is True
|
||||
assert self.test_result_record is None
|
||||
|
||||
def test_unicode_location(self):
|
||||
def test_str_location(self):
|
||||
self.good_message_dict['summary'] = u"ttesterson@mozilla.com NEWCOUNTRY \u0107abcd, \xe4Spain access from 1.2.3.4 (duo) [deviation:12.07010770457331] last activity was from Ottawa, Canada (3763 km away) approx 23.43 hours before"
|
||||
self.good_message_dict['details']['locality_details']['city'] = u'\u0107abcd'
|
||||
self.good_message_dict['details']['locality_details']['country'] = u'\xe4Spain'
|
||||
|
@ -112,19 +112,19 @@ class TestDashboardGeomodel(object):
|
|||
assert result_message == self.good_message_dict
|
||||
assert self.test_connect_called is True
|
||||
assert self.test_result_record is not None
|
||||
assert type(result_message['summary']) is unicode
|
||||
assert type(result_message['details']['locality_details']['city']) is unicode
|
||||
assert type(result_message['details']['locality_details']['country']) is unicode
|
||||
assert type(result_message['summary']) is str
|
||||
assert type(result_message['details']['locality_details']['city']) is str
|
||||
assert type(result_message['details']['locality_details']['country']) is str
|
||||
|
||||
def test_unicode_username(self):
|
||||
def test_str_username(self):
|
||||
self.good_message_dict['details']['principal'] = u'\xfcttesterson@mozilla.com'
|
||||
assert self.test_result_record is None
|
||||
result_message = self.plugin.onMessage(self.good_message_dict)
|
||||
assert result_message == self.good_message_dict
|
||||
assert self.test_connect_called is True
|
||||
assert self.test_result_record is not None
|
||||
assert type(result_message['summary']) is unicode
|
||||
assert type(result_message['details']['principal']) is unicode
|
||||
assert type(result_message['summary']) is str
|
||||
assert type(result_message['details']['principal']) is str
|
||||
|
||||
def test_written_details(self):
|
||||
assert self.test_result_record is None
|
||||
|
|
Загрузка…
Ссылка в новой задаче