def test_get_citations_statute(self, reclassify_statutory_citation):
reclassify_statutory_citation.return_value = ('99', '31999')
citation_text = [
'2 U.S.C. 23',
'2 U.S.C. 23a',
'2 U.S.C. 23a-1',
'4 U.S.C. 23a-1(a)(i)',
'4 U.S.C. 24ff',
]
expected_calls = [
mock.call('2', '23'),
mock.call('2', '23a'),
mock.call('2', '23a-1'),
mock.call('4', '23a-1'),
mock.call('4', '24ff'),
]
citations = load_legal_docs.get_citations(citation_text)
us_code = citations['us_code']
assert len(us_code) == len(expected_calls)
assert reclassify_statutory_citation.call_args_list == expected_calls
assert us_code[0]['text'] == '99 U.S.C. 31999'
assert us_code[1]['text'] == '99 U.S.C. 31999'
assert us_code[2]['text'] == '99 U.S.C. 31999'
assert us_code[3]['text'] == '99 U.S.C. 31999(a)(i)'
assert us_code[4]['text'] == '99 U.S.C. 31999'
url = us_code[0]['url']
assert url == 'https://www.govinfo.gov/link/uscode/99/31999'
def test_get_citations_regulation_no_section(self):
citation_text = [
'11 C.F.R. 19',
]
citations = load_legal_docs.get_citations(citation_text)
regulations = citations['regulations']
assert len(regulations) == 1
assert regulations[0]['url'] == '/regulations/19/CURRENT'
def test_get_citations_statute(self,
reclassify_archived_mur_statutory_citation):
reclassify_archived_mur_statutory_citation.return_value = ('99',
'31999')
citation_text = [
'2 U.S.C. 23',
'2 U.S.C. 23a',
'2 U.S.C. 23a-1',
'4 U.S.C. 23a-1(a)(i)',
'4 U.S.C. 24ff',
]
expected_calls = [
mock.call('2', '23'),
mock.call('2', '23a'),
mock.call('2', '23a-1'),
mock.call('4', '23a-1'),
mock.call('4', '24ff'),
]
citations = load_legal_docs.get_citations(citation_text)
us_code = citations['us_code']
assert len(us_code) == len(expected_calls)
assert reclassify_archived_mur_statutory_citation.call_args_list == expected_calls
assert us_code[0]['text'] == '99 U.S.C. 31999'
assert us_code[1]['text'] == '99 U.S.C. 31999'
assert us_code[2]['text'] == '99 U.S.C. 31999'
assert us_code[3]['text'] == '99 U.S.C. 31999(a)(i)'
assert us_code[4]['text'] == '99 U.S.C. 31999'
parsed_url = urllib.parse.urlparse(us_code[0]['url'])
query = urllib.parse.parse_qs(parsed_url.query)
assert query == dict([
('collection', ['uscode']),
('link-type', ['html']),
('year', ['mostrecent']),
('title', ['99']),
('section', ['31999']),
])
