EclecticLabs
/
ContactRocket


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174
							# encoding: UTF-8
require 'cassandra'
require 'logger'
require 'polipus-cassandra'
require 'spec_helper'

describe Polipus::Storage::CassandraStore do
  before(:all)do
    @logger = Logger.new(STDOUT).tap { |logger| logger.level = Logger::WARN }
    @cluster = Cassandra.cluster hosts: ['127.0.0.1'], logger: @logger
    @keyspace = 'polipus_cassandra_test'
    @table = 'cassandra_store_test'
    @storage = Polipus::Storage::CassandraStore.new(
      cluster: @cluster,
      keyspace: @keyspace,
      table: @table,
    )

    @storage.keyspace!
    @storage.table!

    @storage_without_code_and_body = Polipus::Storage::CassandraStore.new(
      cluster: @cluster,
      keyspace: @keyspace,
      table: @table,
      except: ['code', 'body']
    )
  end

  after(:all) do
    @storage.clear
  end

  it 'should store a page' do
    p = page_factory 'http://www.google.com'
    uuid = @storage.add p
    expect(uuid).to eq('ed646a3334ca891fd3467db131372140')
    p = @storage.get p
    expect(p.url.to_s).to eq('http://www.google.com')
    expect(p.body).to eq('<html></html>')
  end

  it 'should store all the relevant data from the page' do
    url = "http://www.duckduckgo.com"
    referer = "http://www.actually.nowhere.com"
    redirectto = "#{url}/your_super_awesome_results?page=42"
    now = Time.now.to_i
    p = page_factory(
      url,
      {
        referer: referer,
        redirect_to: redirectto,
        fetched_at: now
      })
    uuid = @storage.add p
    expect(uuid).to eq('3cd657f53c74f22c1a21b420ce3863fd')
    p = @storage.get p

    expect(p.url.to_s).to eq(url)
    expect(p.referer.to_s).to eq(referer)
    expect(p.redirect_to.to_s).to eq(redirectto)
    expect(p.fetched_at).to eq(now)
    expect(p.body).to eq('<html></html>')

    # for the sake of the other tests...
    expect(@storage.remove(p)).to be_truthy
  end

  it 'should update a page' do
    p = page_factory 'http://www.google.com', code: 301
    @storage.add p
    p = @storage.get p
    expect(p.code).to eq(301)
  end

  it 'should iterate over stored pages' do
    @storage.each do |k, page|
      expect(k).to eq('ed646a3334ca891fd3467db131372140')
      expect(page.url.to_s).to eq('http://www.google.com')
    end
  end

  it 'should delete a page' do
    p = page_factory 'http://www.google.com', code: 301
    @storage.remove p
    expect(@storage.get(p)).to be_nil
  end

  it 'should store a page removing a query string from the uuid generation' do
    p = page_factory 'http://www.asd.com/?asd=lol'
    p_no_query = page_factory 'http://www.asd.com/?asdas=dasda&adsda=1'
    @storage.include_query_string_in_uuid = false
    @storage.add p
    expect(@storage.exists?(p_no_query)).to be_truthy
    @storage.remove p
  end

  it 'should store a page removing a query string from the uuid generation no ending slash' do
    p = page_factory 'http://www.asd.com?asd=lol'
    p_no_query = page_factory 'http://www.asd.com'
    @storage.include_query_string_in_uuid = false
    @storage.add p
    expect(@storage.exists?(p_no_query)).to be_truthy
    @storage.remove p
  end

  it 'should store a page with user data associated' do
    p = page_factory 'http://www.user.com'
    p.user_data.name = 'Test User Data'
    @storage.add p
    expect(@storage.exists?(p)).to be_truthy
    p = @storage.get(p)
    expect(p.user_data.name).to eq('Test User Data')
    @storage.remove p
  end

  it 'should honor the except parameters' do
    pag = page_factory 'http://www.user-doo.com'
    expect(pag.code).to eq(200)
    expect(pag.body).to eq('<html></html>')

    @storage_without_code_and_body.add(pag)
    pag = @storage_without_code_and_body.get(pag)

    expect(pag.body).to be_nil
    expect(pag.code).to eq(0)
    @storage_without_code_and_body.remove(pag)
  end

  it 'should return false if a doc not exists' do
    @storage.include_query_string_in_uuid = false
    p_other  = page_factory 'http://www.asdrrrr.com'
    expect(@storage.exists?(p_other)).to be_falsey
    @storage.add p_other
    expect(@storage.exists?(p_other)).to be_truthy
    p_other  = page_factory 'http://www.asdrrrr.com?trk=asd-lol'
    expect(@storage.exists?(p_other)).to be_truthy
    @storage.include_query_string_in_uuid = true
    expect(@storage.exists?(p_other)).to be_falsey
    @storage.include_query_string_in_uuid = false
    @storage.remove p_other
  end

  it 'should set page.fetched_at based on the id creation' do
    p = page_factory 'http://www.user-doojo.com'
    @storage.add p
    expect(p.fetched_at).to be_nil
    p = @storage.get p
    expect(p.fetched_at).not_to be_nil
    @storage.remove p
  end

  it 'should NOT set page.fetched_at if already present' do
    p = page_factory 'http://www.user-doojooo.com'
    p.fetched_at = 10
    @storage.add p
    p = @storage.get p
    expect(p.fetched_at).to be 10
    @storage.remove p
  end

  it 'should store two pages and the count will be two' do
    pages = ['http://www.google.com', 'http://www.duckduckgo.com'].map do |url|
      page_factory(url).tap do |page|
        @storage.add(page)
      end
    end
    expect(@storage.count).to be 2
    pages.each do |page|
      @storage.remove(page)
    end
    expect(@storage.count).to be 0
  end
end