def handle(self, *args, **options):
self.build_id = 'build-%s' % now
# send the end signal
signals.build_item_commence.send(sender=self, thread=thread.get_ident())
generate_screenshot = True if options['generate'] in ['True',True] else False
all_pages = True if options['all'] in ['True',True] else False
num_pages_ids = len(args)
logger.debug('num num_pages_ids %d' % num_pages_ids)
if all_pages in ['False',False] and num_pages_ids == 0:
raise CommandError('Please specify page_id(s) to test in form: tailor_run_layout_test <id> <id> <id> ...')
self.spindle = Spindle(capture=generate_screenshot, output_path=OUTPUT_PATH)
if all_pages == True:
args = []
for page in Page.objects.all():
self.test_page(url=page.url, elements_list=page.get_test_elements(), page=page)
else:
for page_id in args:
try:
page = Page.objects.get(pk=int(page_id))
except Page.DoesNotExist:
raise CommandError('Page "%s" does not exist' % page_id)
self.test_page(url=page.url, elements_list=page.get_test_elements(), page=page)
self.spindle.driver.close()
self.spindle.driver.quit()
# send the end signal
signals.build_item_complete.send(sender=self, thread=thread.get_ident())
def admin_load_page_url(request, page_id):
page = get_object_or_404(Page, pk=page_id)
try:
needle = Spindle(capture=False, output_path='/tmp/')
response = needle.getPageHTML(page.url)
except:
raise Http404('Url "%s" is not available. Please check internet connection and that selenium server is running (tailor_start), and that you have Java and if running headless then Xfvb as well' % page.url)
url = urlparse.urlparse(page.url)
remote_html = smart_unicode(response)
replace = '="%s://%s/' % (url.scheme,url.hostname,)
# remote_html = remote_html.replace('="/', replace)
return render_to_response(
'admin/page/page/remote_page.html', {
'is_ajax': request.is_ajax,
'remote_html': remote_html,
},
context_instance=RequestContext(request)
)
class Command(BaseCommand):
args = '<page_id page_id ...>'
help = 'Generate screenshots for a list of Page objects'
option_list = BaseCommand.option_list + (
make_option('--generate',
action='store',
dest='generate',
default=False,
help='Genereate Screenshot for specified Page'),
make_option('--all',
action='store',
dest='all',
default=False,
help='Perform action on All Pages'),
)
needle = None
def handle(self, *args, **options):
self.build_id = 'build-%s' % now
# send the end signal
signals.build_item_commence.send(sender=self, thread=thread.get_ident())
generate_screenshot = True if options['generate'] in ['True',True] else False
all_pages = True if options['all'] in ['True',True] else False
num_pages_ids = len(args)
logger.debug('num num_pages_ids %d' % num_pages_ids)
if all_pages in ['False',False] and num_pages_ids == 0:
raise CommandError('Please specify page_id(s) to test in form: tailor_run_layout_test <id> <id> <id> ...')
self.spindle = Spindle(capture=generate_screenshot, output_path=OUTPUT_PATH)
if all_pages == True:
args = []
for page in Page.objects.all():
self.test_page(url=page.url, elements_list=page.get_test_elements(), page=page)
else:
for page_id in args:
try:
page = Page.objects.get(pk=int(page_id))
except Page.DoesNotExist:
raise CommandError('Page "%s" does not exist' % page_id)
self.test_page(url=page.url, elements_list=page.get_test_elements(), page=page)
self.spindle.driver.close()
self.spindle.driver.quit()
# send the end signal
signals.build_item_complete.send(sender=self, thread=thread.get_ident())
def test_page(self, url, elements_list=None, page=None):
""" selenium test the url using needle """
# init page path if not exists
call_command('tailor_cutting_room', str(page.pk), initialize=True)
# get needle driver url
logger.info('Loading Test Url: %s' % url)
self.spindle.driver.get( url )
elements = []
if not elements_list or len(elements_list) == 0:
logger.info('No Elements Found for "%s" (%d): Using default (html)' % (url, page.pk,))
elements.append('html')
else:
for e in elements_list:
e = e.strip()
logger.debug('Require Element: %s' % e)
elements.append( e )
# set needle output path
self.spindle.output_path = '%s%s/' % (OUTPUT_PATH, page.pk,)
logger.info('Output path: %s' % self.spindle.output_path)
# if the dir does not exist make it.. make it good
if not os.path.exists(self.spindle.output_path):
logger.warning('Output path Did not exist creating: %s' % self.spindle.output_path)
os.makedirs(self.spindle.output_path)
# loop over provided elements
for e in elements:
c = 0
e_name = slugify(e)
logger.info('Selenium find "%s"' % e)
blocks = self.spindle.driver.find_elements_by_css_selector( e )
num_blocks = len(blocks)
logger.info('%d Blocks Found for element "%s"\n' % (num_blocks, e,), extra={'page': page, 'element': e})
if num_blocks > 0:
for b in blocks:
c = c + 1
element_name = '%s-%02d' % (e_name, c,)
try:
self.spindle.assertScreenshot(b, element_name, NEEDLE_TOLERANCE)
except AssertionError as error:
# this would trigger a failed test - Element Layout No Match
logger.error('ElementLayoutNoMatch - Element Layout does not match its baseline: %s' % error)
else:
# this would trigger a failed test - Element not found
logger.error('ElementNotFound - No Blocks Found for Element "%s"' % e)
