def setup(self):
super().setup()
index = 'localities'
if self.config_delete_indexes:
self.es_client.delete_index(index, True)
self.es_client.create_index(index)
journal = geomodel.wrap_journal(self.es_client)
def state(username, locs):
return geomodel.State('locality', username, locs)
def locality(cfg):
return geomodel.Locality(**cfg)
test_state = state('tester1', [
locality({
'sourceipaddress': '1.2.3.4',
'city': 'Portland',
'country': 'US',
'lastaction': toUTC(datetime.now()),
'latitude': 45.5234,
'longitude': -122.6762,
'radius': 50,
})
])
journal(geomodel.Entry.new(test_state), index)
self.refresh(index)
def test_journaling(self):
journal = locality.wrap_journal(self.es_client)
test_entry = locality.Entry(
'testingid1',
locality.State('locality', 't1', [
locality.Locality(
sourceipaddress='1.2.3.4',
city='Toronto',
country='CA',
lastaction=toUTC(datetime.now()) - timedelta(days=3),
latitude=43.6529,
longitude=-79.3849,
radius=50),
locality.Locality(
sourceipaddress='32.64.128.255',
city='Berlin',
country='DE',
lastaction=toUTC(datetime.now()) - timedelta(minutes=30),
latitude=52.520008,
longitude=13.404954,
radius=50)
]))
journal(test_entry, self.event_index_name)
self.refresh(self.event_index_name)
query_iface = locality.wrap_query(self.es_client)
loc_cfg = config.Localities(self.event_index_name, 30, 50.0)
entry = locality.find(query_iface, 't1', loc_cfg.es_index)
assert entry is not None
assert entry.state.username == 't1'
def setup(self):
super().setup()
index = 'localities'
if self.config_delete_indexes:
self.es_client.delete_index(index, True)
self.es_client.create_index(index)
journal = geomodel.wrap_journal(self.es_client)
def state(username, locs):
return geomodel.State('locality', username, locs)
def locality(cfg):
return geomodel.Locality(**cfg)
test_states = [
state(
'tester1',
[
locality({
'sourceipaddress':
'1.2.3.4',
'city':
'San Francisco',
'country':
'US',
'lastaction':
toUTC(datetime.now()) - timedelta(minutes=3),
'latitude':
37.773972,
'longitude':
-122.431297,
'radius':
50,
}),
locality({
'sourceipaddress':
'9.8.7.6',
'city':
'Toronto',
'country':
'CA',
'lastaction':
toUTC(datetime.now()) - timedelta(minutes=5),
'latitude':
43.6529,
'longitude':
-79.3849,
'radius':
50,
}),
],
)
]
for state in test_states:
journal(geomodel.Entry.new(state), index)
self.refresh(index)
def setup(self):
super().setup()
if self.config_delete_indexes:
self.es_client.delete_index(GeoModelTest.localities_index, True)
self.es_client.create_index(GeoModelTest.localities_index)
journal = geomodel.wrap_journal(self.es_client)
states = self.test_states if hasattr(self, 'test_states') else []
for state in states:
journal(geomodel.Entry.new(state), GeoModelTest.localities_index)
self.refresh(GeoModelTest.localities_index)