def teardown_method(self, _test_method):
for region in REGIONS:
docket.purge_query_docket(
ScrapeKey(region, constants.ScrapeType.BACKGROUND)
)
sessions.ds().delete_multi(self.sessions_to_delete)
self.project_id_patcher.stop()
def create_session(self, scrape_key):
session = sessions.ScrapeSession.new(
key=sessions.ds().key(sessions.SCRAPE_SESSION_KIND),
region=scrape_key.region_code,
scrape_type=scrape_key.scrape_type,
phase=scrape_phase.ScrapePhase.START)
sessions.ds().put(session.to_entity())
self.sessions_to_delete.append(session.to_entity().key)
def create_session(self, region_code, scrape_type, phase, start, end=None,
docket_ack_id=None):
session = ScrapeSession.new(
key=sessions.ds().key('ScrapeSession'), region=region_code,
scrape_type=scrape_type, phase=phase, docket_ack_id=docket_ack_id,
start=start, end=end)
sessions.ds().put(session.to_entity())
self.keys_to_delete.append(session.to_entity().key)
return session
def create_session(
self,
region_code: str,
scrape_type: constants.ScrapeType,
phase: scrape_phase.ScrapePhase,
start: datetime,
end: Optional[datetime] = None,
docket_ack_id: Optional[str] = None,
) -> ScrapeSession:
session = ScrapeSession.new(
key=sessions.ds().key("ScrapeSession"),
region=region_code,
scrape_type=scrape_type,
phase=phase,
docket_ack_id=docket_ack_id,
start=start,
end=end,
)
sessions.ds().put(session.to_entity())
self.keys_to_delete.append(session.to_entity().key)
return session
def teardown_method(self, _test_method):
sessions.ds().delete_multi(self.keys_to_delete)
def teardown_method(self, _test_method: Callable) -> None:
sessions.ds().delete_multi(self.keys_to_delete)
