def setUp(self):
self.service_funnel = ServiceFunnel()
html_string = """
<article class="article " id="id-39334" data-tags="A,B,C"><h2>S1</h2></article>
<article class="article " id="id-39335" data-tags="B,C,D,E"><h2>S2</h2></article>
"""
self.service_funnel.scrape_html(html_string)
def test_scrape_html(self):
html_text_class = HTMLText()
text_input = html_text_class.get_string()
service_funnel = ServiceFunnel()
result = service_funnel.scrape_html(text_input)
assert isinstance(result, dict)
assert len(result) > 1
def test_handle_request(self, mock_response_class, mock_request_class):
request = {
"selected_tags": [{
"name": "Kündigung"
}, {
"name": "Mobilfunkvertrag"
}]
}
service_funnel = ServiceFunnel()
service_funnel.handle_request(request)
mock_request_class.return_value.get_request_type.assert_called_once()
mock_response_class.return_value.get_response.assert_called_once()
from main import ServiceFunnel
###---test html---###
html_string = """
<article class="article " id="id-39334" data-tags="A,B,C"><h2>S1</h2></article>
<article class="article " id="id-39335" data-tags="A,B,D,E"><h2>S2</h2></article>
<article class="article " id="id-39336" data-tags="X,C,D,L"><h2>S3</h2></article>
<article class="article " id="id-39337" data-tags="J,A,B,Q"><h2>S4</h2></article>
<article class="article " id="id-39338" data-tags="M,A,B,C,S"><h2>S5</h2></article>
"""
service_funnel = ServiceFunnel()
service_funnel.scrape_html(html_string)
def test_valid_tags_with_snippet():
result = service_funnel.handle_request(
{"selected_tags": [{"name": "A"}, {"name": "B"}, {"name": "C"}]}
return result
def test_invalid_tags():
result = service_funnel.handle_request(
{"selected_tags": [{"name": "W"}, {"name": "T"}, {"name": "R"}]}
)
return result
def test_valid_tags_without_snippet():
result = service_funnel.handle_request(
{"selected_tags": [{"name": "A"}, {"name": "B"}]}
)
return result