def test_save_result(self):
with self.client:
user_id = self.user_data["user1"]["user_id"]
request_data = {
"first_name": "John",
"last_name": "Doe",
"middle_name": "Test",
"birth_date": "01/01/1980",
}
record = CrawlerFactory.create(CrawlerFactory.setup())
self.login(self.user_data["user1"]["email"],
self.user_data["user1"]["password"])
g.database = get_database()
expungeservice.stats.save_result(request_data, record)
g.database.connection.commit()
self.check_search_result_saved(self.user_data["user1"]["user_id"],
request_data,
num_eligible_charges=6,
num_charges=9)
def test_it_creates_json(self):
crawler = CrawlerFactory.setup()
record = CrawlerFactory.create(crawler)
jsonified_dictionary = json.loads(
json.dumps(record, cls=ExpungeModelEncoder))
assert type(jsonified_dictionary) == dict
def setUp(self):
EndpointShared.setUp(self)
""" Save these actual function refs to reinstate after we're done mocking them."""
self.crawler_login = search.Crawler.login
self.search = search.Crawler.search
self.crawler = CrawlerFactory.setup()
self.crawler.logged_in = True
self.mock_record = {"john_doe": CrawlerFactory.create(self.crawler)}
def setup_and_teardown(service):
service.setup()
service.crawler = CrawlerFactory.setup()
service.crawler.logged_in = True
service.mock_record = {"john_doe": CrawlerFactory.create(service.crawler)}
service.search_request_data = {
"names": [{"first_name": "John", "last_name": "Doe", "middle_name": "", "birth_date": "02/02/1990"}]
}
yield
service.teardown()
def setUp(self):
EndpointShared.setUp(self)
""" Save these actual function refs to reinstate after we're done mocking them."""
self.crawler_login = search.Crawler.login
self.search = search.Crawler.search
self.save_result = search.save_result
self.crawler = CrawlerFactory.setup()
self.crawler.logged_in = True
self.mock_record = {"john_doe": CrawlerFactory.create(self.crawler)}
self.search_request_data = {
"first_name": "John",
"last_name": "Doe",
"middle_name": "",
"birth_date": "02/02/1990"
}
def setUp(self):
self.crawler = CrawlerFactory.setup()
def crawler():
return CrawlerFactory.setup()
def test_case_without_dispos(self):
crawler = CrawlerFactory.setup()
record = CrawlerFactory.create(crawler, JohnDoe.SINGLE_CASE_RECORD, {'CASEJD1': CaseDetails.CASE_WITHOUT_DISPOS})
expunger = Expunger(record)
expunger.run()
def test_partial_dispos(self):
crawler = CrawlerFactory.setup()
record = CrawlerFactory.create(crawler, JohnDoe.SINGLE_CASE_RECORD, {'CASEJD1': CaseDetails.CASE_WITH_PARTIAL_DISPOS})
expunger = Expunger(record)
expunger.run()