require_relative '../../spec_helper'

describe Arachni::Module::ElementDB do

    before( :all ) do
        @edb = Class.new
        @edb.extend Arachni::Module::ElementDB

        @page = Arachni::Page.new(
            :url => 'http://blah.com',
            :links => [
                Arachni::Element::Link.new(
                    'http://blah.com',
                    :inputs => {
                        'link_input' => 'link_value'
                    }
                )
            ],
            :forms => [
                Arachni::Element::Form.new(
                    'http://blah.com',
                    :inputs => {
                        'form_input' => 'form_value'
                    }
                )
            ],
            :cookies => [
                Arachni::Element::Cookie.new(
                    'http://blah.com',
                    'cookie_input' => 'cookie_value'
                )
            ],
            :headers => [
                Arachni::Element::Header.new(
                    'http://blah.com',
                    'header_input' => 'header_value'
                )
            ]
        )

        @edb.init_db_from_page( @page )
    end

    describe '#update_links' do
        context 'when there are new links' do
            it 'should add them to the DB and return them' do
                link = Arachni::Element::Link.new(
                    'http://blah.com',
                    :inputs => {
                        'new_link_input' => 'new_link_value'
                    }
                )
                new_links, new_link_cnt = @edb.update_links( [link] | @page.links )
                new_link_cnt.should == 1
                new_links.size.should == 1
                new_links.first.auditable[link.auditable.keys.first].should == link.auditable.values.first
            end
        end

        context 'when there are no new links' do
            it 'should return empty results' do
                new_links, new_link_cnt = @edb.update_links( @page.links )
                new_link_cnt.should == 0
                new_links.should be_empty
            end
        end
    end

    describe '#update_forms' do
        context 'when there are new forms' do
            it 'should add them to the DB and return them' do
                form = Arachni::Element::Form.new(
                    'http://blah.com',
                    :inputs => {
                        'new_form_input' => 'new_form_value'
                    }
                )
                new_forms, new_form_cnt = @edb.update_forms( [form] | @page.forms )
                new_form_cnt.should == 1
                new_forms.size.should == 1
                new_forms.first.auditable[form.auditable.keys.first].should == form.auditable.values.first
            end
        end

        context 'when there are no new links' do
            it 'should return empty results' do
                new_forms, new_form_cnt = @edb.update_forms( @page.forms )
                new_form_cnt.should == 0
                new_forms.should be_empty
            end
        end
    end

    describe '#update_cookies' do
        context 'when there are new cookies' do
            it 'should add them to the DB, return all cookies but only return the count of the new ones' do
                cookie = Arachni::Element::Cookie.new(
                    'http://blah.com',
                    'new_cookie_input' => 'new_cookie_value'
                )
                cookies = [cookie] | @page.cookies
                new_cookies, new_cookie_cnt = @edb.update_cookies( cookies )
                new_cookie_cnt.should == 1
                new_cookies.size.should == cookies.size
            end
        end

        context 'when there are cookies with existing names but have different values' do
            it 'should update the values' do
                cookie = Arachni::Element::Cookie.new(
                    'http://blah.com',
                    'cookie_input' => 'foo!'
                )
                new_cookies, new_cookie_cnt = @edb.update_cookies( [cookie] | @page.cookies )
                new_cookie_cnt.should == 0
                new_cookies.size.should == 2
                new_cookies.find{ |c| c.raw['name'] == 'cookie_input' }.raw['value'].should ==
                    cookie.raw['value']
            end
        end

        context 'when there are no new cookies' do
            it 'should return empty results' do
                new_cookies, new_cookie_cnt = @edb.update_cookies( @page.cookies )
                new_cookie_cnt.should == 0
                new_cookies.size.should == 2
            end
        end
    end

end