def test_unthrottle_wifi_measures(self):
session = self.db_master_session
block = MeasureBlock()
block.measure_type = MEASURE_TYPE_CODE['wifi']
block.start_id = 120
block.end_id = 140
block.s3_key = 'fake_key'
block.archive_sha = 'fake_sha'
block.archive_date = None
session.add(block)
k = dict(lat=1.0, lon=1.0)
for i in range(100, 150):
session.add(WifiMeasure(id=i, key=str(i), created=self.really_old))
session.add(Wifi(total_measures=11000, key=str(i), **k))
session.commit()
with patch.object(S3Backend, 'check_archive', lambda x, y, z: True):
delete_wifimeasure_records.delay(batch=7).get()
wifi_unthrottle_measures.delay(10000, 1000).get()
wifis = session.query(Wifi).all()
self.assertEquals(len(wifis), 50)
for wifi in wifis:
if 120 <= int(wifi.key) and int(wifi.key) < 140:
self.assertEquals(wifi.total_measures, 0)
else:
self.assertEquals(wifi.total_measures, 1)
self.check_stats(counter=['items.wifi_unthrottled'])
def test_unthrottle_wifi_measures(self):
session = self.db_master_session
block = MeasureBlock()
block.measure_type = MEASURE_TYPE_CODE['wifi']
block.start_id = 120
block.end_id = 140
block.s3_key = 'fake_key'
block.archive_sha = 'fake_sha'
block.archive_date = None
session.add(block)
k = dict(lat=1.0, lon=1.0)
for i in range(100, 150):
session.add(WifiMeasure(id=i, key=str(i), created=self.really_old))
session.add(Wifi(total_measures=11000, key=str(i), **k))
session.commit()
with patch.object(S3Backend, 'check_archive', lambda x, y, z: True):
delete_wifimeasure_records.delay(batch=7).get()
wifi_unthrottle_measures.delay(10000, 1000).get()
wifis = session.query(Wifi).all()
self.assertEquals(len(wifis), 50)
for wifi in wifis:
if 120 <= int(wifi.key) and int(wifi.key) < 140:
self.assertEquals(wifi.total_measures, 0)
else:
self.assertEquals(wifi.total_measures, 1)
self.check_stats(counter=['items.wifi_unthrottled'])
def test_delete_wifi_observations(self):
obs = WifiObservationFactory.create_batch(50, created=self.old)
self.session.flush()
start_id = obs[0].id + 20
block = ObservationBlockFactory(
measure_type=ObservationType.wifi,
start_id=start_id, end_id=start_id + 20, archive_date=None)
self.session.commit()
delete_wifimeasure_records.delay(batch=7).get()
self.assertEquals(self.session.query(WifiObservation).count(), 30)
self.assertTrue(block.archive_date is not None)
def test_delete_wifi_observations(self):
obs = WifiObservationFactory.create_batch(50, created=self.old)
self.session.flush()
start_id = obs[0].id + 20
block = ObservationBlockFactory(measure_type=ObservationType.wifi,
start_id=start_id,
end_id=start_id + 20,
archive_date=None)
self.session.commit()
with patch.object(S3Backend, 'check_archive', lambda x, y, z: True):
delete_wifimeasure_records.delay(batch=7).get()
self.assertEquals(self.session.query(WifiObservation).count(), 30)
self.assertTrue(block.archive_date is not None)
def test_delete_wifi_measures(self):
session = self.db_master_session
block = MeasureBlock()
block.measure_type = MEASURE_TYPE_CODE['wifi']
block.start_id = 120
block.end_id = 140
block.s3_key = 'fake_key'
block.archive_sha = 'fake_sha'
block.archive_date = None
session.add(block)
for i in range(100, 150):
session.add(WifiMeasure(id=i, created=self.really_old))
session.commit()
with patch.object(S3Backend, 'check_archive', lambda x, y, z: True):
delete_wifimeasure_records.delay(batch=7).get()
self.assertEquals(session.query(WifiMeasure).count(), 30)
self.assertTrue(block.archive_date is not None)
def test_delete_wifi_measures(self):
session = self.db_master_session
block = MeasureBlock()
block.measure_type = MEASURE_TYPE_CODE['wifi']
block.start_id = 120
block.end_id = 140
block.s3_key = 'fake_key'
block.archive_sha = 'fake_sha'
block.archive_date = None
session.add(block)
for i in range(100, 150):
session.add(WifiMeasure(id=i, created=self.really_old))
session.commit()
with patch.object(S3Backend, 'check_archive', lambda x, y, z: True):
delete_wifimeasure_records.delay(batch=7).get()
self.assertEquals(session.query(WifiMeasure).count(), 30)
self.assertTrue(block.archive_date is not None)