def test_can_get_fact_definitions(self):
reviewer = Mock()
facter = ImageFacter(reviewer)
definitions = facter.get_fact_definitions()
expect(definitions).to_length(3)
expect('page.images' in definitions).to_be_true()
expect('total.size.img' in definitions).to_be_true()
expect('total.requests.img' in definitions).to_be_true()
def test_handle_url_loaded(self):
page = PageFactory.create()
reviewer = Reviewer(
api_url='http://localhost:2368',
page_uuid=page.uuid,
page_url=page.url,
page_score=0.0,
config=Config(),
facters=[]
)
content = '<html><img src="test.png" alt="a" title="b" /></html>'
result = {
'url': page.url,
'status': 200,
'content': content,
'html': lxml.html.fromstring(content)
}
reviewer.responses[page.url] = result
reviewer._wait_for_async_requests = Mock()
reviewer.save_review = Mock()
response = Mock(status_code=200, text=content, headers={})
reviewer.content_loaded(page.url, response)
facter = ImageFacter(reviewer)
facter.async_get = Mock()
facter.get_facts()
facter.handle_url_loaded(page.url, response)
expect(facter.review.data).to_include('page.all_images')
expect(facter.review.data['page.all_images']).not_to_be_null()
img_src = facter.review.data['page.all_images'][0].get('src')
expect(img_src).to_equal('test.png')
expect(facter.review.data).to_include('page.images')
data = set([(page.url, response)])
expect(facter.review.data['page.images']).to_equal(data)
expect(facter.review.data).to_include('total.size.img')
expect(facter.review.data['total.size.img']).to_equal(0.0517578125)
def test_ignore_base64_images(self):
page = PageFactory.create()
reviewer = Reviewer(
api_url='http://localhost:2368',
page_uuid=page.uuid,
page_url=page.url,
page_score=0.0,
config=Config(),
facters=[]
)
content = '<html><img src="data:image/png;base64,iVBOR" alt="a" title="b" /></html>'
result = {
'url': page.url,
'status': 200,
'content': content,
'html': lxml.html.fromstring(content)
}
reviewer.responses[page.url] = result
reviewer._wait_for_async_requests = Mock()
reviewer.save_review = Mock()
response = Mock(status_code=200, text=content, headers={})
reviewer.content_loaded(page.url, response)
facter = ImageFacter(reviewer)
facter.add_fact = Mock()
facter.async_get = Mock()
facter.get_facts()
expect(facter.add_fact.call_args_list).to_include(
call(
key='page.images',
value=set([]),
))
expect(facter.add_fact.call_args_list).to_include(
call(
key='total.size.img',
value=0,
)),
expect(facter.add_fact.call_args_list).to_include(
call(
key='total.requests.img',
value=0,
))
def test_can_get_facts(self):
page = PageFactory.create(url='http://my-site.com')
reviewer = Reviewer(
api_url='http://localhost:2368',
page_uuid=page.uuid,
page_url=page.url,
page_score=0.0,
config=Config(),
facters=[]
)
content = '<html><img src="test.png" alt="a" title="b" /><img src="" /></html>'
result = {
'url': page.url,
'status': 200,
'content': content,
'html': lxml.html.fromstring(content)
}
reviewer.responses[page.url] = result
reviewer._wait_for_async_requests = Mock()
reviewer.save_review = Mock()
response = Mock(status_code=200, text=content, headers={})
reviewer.content_loaded(page.url, response)
facter = ImageFacter(reviewer)
facter.add_fact = Mock()
facter.async_get = Mock()
facter.get_facts()
expect(facter.review.data).to_length(3)
expect(facter.review.data).to_include('page.all_images')
img = facter.review.data['page.all_images'][0]
expect(img.tag).to_equal('img')
expect(img.get('src')).to_equal('test.png')
expect(facter.review.data).to_include('page.images')
expect(facter.review.data['page.images']).to_equal(set([]))
expect(facter.review.data).to_include('total.size.img')
expect(facter.review.data['total.size.img']).to_equal(0)
expect(facter.add_fact.call_args_list).to_include(
call(
key='page.images',
value=set([]),
))
expect(facter.add_fact.call_args_list).to_include(
call(
key='total.size.img',
value=0,
)),
expect(facter.add_fact.call_args_list).to_include(
call(
key='total.requests.img',
value=1,
))
facter.async_get.assert_called_once_with(
'http://my-site.com/test.png',
facter.handle_url_loaded
)