neocities
/
puppeteer-ruby
mirror of https://github.com/neocities/puppeteer-ruby


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108
							# frozen_string_literal: true

require 'spec_helper'
require 'tmpdir'

RSpec.describe 'example' do
  def with_network_retry(max_retry: 2, timeout: 4, &block)
    if max_retry <= 0
      Timeout.timeout(timeout, &block)
    else
      begin
        Timeout.timeout(timeout, &block)
      rescue Timeout::Error
        puts "Retry with { remaining: #{max_retry - 1}, timeout: #{timeout * 1.5} }"
        with_network_retry(max_retry: max_retry - 1, timeout: timeout * 1.5, &block)
      end
    end
  end

  it 'should take a screenshot' do
    skip if Puppeteer.env.ci? && !Puppeteer.env.windows?

    with_network_retry { page.goto('https://github.com/YusukeIwaki') }
    tmpdir = Dir.mktmpdir
    begin
      path = File.join(tmpdir, 'YusukeIwaki.png')
      page.screenshot(path: path)
      expect(File.open(path, 'rb').read.size).to be > 1000
    ensure
      FileUtils.remove_entry(tmpdir, true)
    end
  end

  it 'should print PDF with options' do
    skip if Puppeteer.env.ci? && !Puppeteer.env.windows?

    page.viewport = Puppeteer::Viewport.new(width: 1200, height: 800, device_scale_factor: 2)
    with_network_retry { page.goto("https://github.com/YusukeIwaki") }
    page.wait_for_selector(".js-yearly-contributions")
    overlay = page.query_selector('.js-yearly-contributions')

    js = <<-JAVASCRIPT
    graph => {
      const width = getComputedStyle(graph).width;
      graph = graph.cloneNode(true);
      graph.style.width = width;
      document.body.innerHTML = `
        <div style="display:flex;justify-content:center;align-items:center;height:100vh;">;
          ${graph.outerHTML}
        </div>
      `;
    }
    JAVASCRIPT
    page.evaluate(js, overlay)
    page.pdf(
      path: '5.element-to-pdf.github.pdf',
      print_background: true,
      format: "letter",
      margin: { top: "1cm", left: "2cm", right: "3cm", bottom: "4cm" },
      omit_background: true,
    )
  end

  it 'should input text and grab DOM elements' do
    skip if Puppeteer.env.ci? && !Puppeteer.env.windows?

    page.viewport = Puppeteer::Viewport.new(width: 1280, height: 800)
    with_network_retry { page.goto("https://github.com/", wait_until: 'domcontentloaded') }

    form = page.query_selector("form.js-site-search-form")
    search_input = form.query_selector("input.header-search-input")
    search_input.click
    page.keyboard.type_text("puppeteer")

    page.wait_for_navigation do
      search_input.press("Enter")
    end

    list = page.query_selector("ul.repo-list")
    items = list.query_selector_all("div.f4")
    items.each do |item|
      title = item.eval_on_selector("a", "a => a.innerText")
      puts("==> #{title}")
    end
  end

  it 'should evaluate expression' do
    expect(page.evaluate('2 + 3')).to eq(5)
  end

  it 'should evaluate function returning object' do
    expect(page.evaluate('() => { return { a: 3, b: 4 } }')).to eq({ 'a' => 3, 'b' => 4 })
  end

  it 'should work with waitForFunction with timeout: 0' do
    page.content = '<body>Now Loading...</body>'
    page.wait_for_function('() => document.querySelector("body").innerText.includes("Something")', timeout: 0) do
      page.evaluate(<<~JAVASCRIPT)
      () => {
        setTimeout(() => {
          document.body.innerHTML = "<h1>It works!</h1><p>Something</p>"
        }, 500)
      }
      JAVASCRIPT
    end
  end
end