require 'spec_helper'

class MockBrowserCluster
    attr_reader :result

    def handle_job_result( result )
        @result = result
    end
end

class MockWorker
    def master
        @master ||= MockBrowserCluster.new
    end
end

class JobTest < Arachni::BrowserCluster::Job
    include RSpec::Matchers

    def ran?
        !!@ran
    end

    def run
        expect(browser.class).to eq MockWorker
        @ran = true
    end
end

class JobConfigureAndRunTest < JobTest
    def run
        expect(browser.class).to eq MockWorker
        super
    end
end

class JobSaveResultTest < JobTest
    class Result < Arachni::BrowserCluster::Job::Result
        attr_accessor :my_data
    end

    def run
        val = 'stuff'
        save_result my_data: val

        result = browser.master.result
        expect(result.job.id).to eq self.id
        expect(result.my_data).to eq val

        super
    end
end

class JobCleanCopyTest < JobTest
    def run
        expect(browser.class).to eq MockWorker

        copy = self.clean_copy
        expect(copy.browser).to eq nil
        expect(copy.id).to eq self.id

        expect(browser.class).to eq MockWorker

        super
    end
end

class JobDupTest < JobTest
    attr_accessor :my_data

    def initialize( options )
        super options
        @my_data = options.delete( :my_data )
    end
end

class JobForwardTest < JobDupTest
end

class JobForwardAsTest < JobForwardTest
end

describe Arachni::BrowserCluster::Job do
    let(:browser_cluster) { MockBrowserCluster.new }
    let(:worker) { MockWorker.new }

    describe '#id' do
        it 'gets incremented with each initialization' do
            id = nil
            10.times do |i|
                id = described_class.new.id
                next if i == 0

                expect(described_class.new.id).to eq(id + 1)
            end
        end
    end

    describe '#never_ending?' do
        subject { JobTest.new }

        context 'when #never_ending is' do
            context 'true' do
                it 'returns true' do
                    subject.never_ending = true
                    expect(subject.never_ending?).to be_truthy
                end
            end

            context 'false' do
                it 'returns false' do
                    subject.never_ending = false
                    expect(subject.never_ending?).to be_falsey
                end
            end

            context 'nil' do
                it 'returns false' do
                    expect(subject.never_ending?).to be_falsey
                end
            end
        end
    end

    describe '#configure_and_run' do
        subject { JobConfigureAndRunTest.new }

        it 'sets #browser' do
            subject.configure_and_run( worker )
        end

        it 'calls #run' do
            expect(subject.ran?).to be_falsey
            subject.configure_and_run( worker )
            expect(subject.ran?).to be_truthy
        end

        it 'removes #browser' do
            expect(subject.ran?).to be_falsey
            subject.configure_and_run( worker )
            expect(subject.browser).to be_nil
            expect(subject.ran?).to be_truthy
        end
    end

    describe '#save_result' do
        subject { JobSaveResultTest.new }

        it 'forwards the result to the BrowserCluster' do
            expect(subject.ran?).to be_falsey
            subject.configure_and_run( worker )
            expect(subject.ran?).to be_truthy
        end
    end

    describe '#clean_copy' do
        subject { JobCleanCopyTest.new }

        it 'copies the Job without the resources set by #configure_and_run' do
            expect(subject.ran?).to be_falsey
            subject.configure_and_run( worker )
            expect(subject.ran?).to be_truthy
        end
    end

    describe '#dup' do
        subject { JobDupTest.new( never_ending: true, my_data: 'stuff' ) }

        it 'copies the Job' do
            expect(subject.my_data).to eq('stuff')

            dup = subject.dup
            expect(dup.my_data).to eq('stuff')
            expect(dup.never_ending?).to eq(true)
        end

        it 'preserves #time' do
            subject.time = 10
            expect(subject.time).to eq 10

            dup = subject.dup
            expect(dup.time).to eq 10
        end

        it 'preserves #timed_out' do
            subject.timed_out! 10
            expect(subject.time).to eq 10
            expect(subject).to be_timed_out

            dup = subject.dup
            expect(dup.time).to eq 10
            expect(subject).to be_timed_out
        end
    end

    describe '#forward' do
        subject { JobForwardTest.new( my_data: 'stuff' ) }

        it 'sets the original Job as the #forwarder' do
            id = subject.id
            expect(subject.forward.forwarder).to eq(subject)
        end

        it 'creates a new Job with the same #id' do
            id = subject.id
            expect(subject.forward.id).to eq(id)
        end

        it 'creates a new Job with the same #never_ending' do
            expect(subject.forward.never_ending?).to be_falsey

            job = JobForwardTest.new( never_ending: true, my_data: 'stuff' )
            expect(job.never_ending?).to be_truthy
            expect(job.forward.never_ending?).to be_truthy

            job = JobForwardTest.new( never_ending: false, my_data: 'stuff' )
            expect(job.never_ending?).to be_falsey
            expect(job.forward.never_ending?).to be_falsey
        end

        it 'does not preserve any existing data' do
            expect(subject.forward.my_data).to be_nil
        end

        context 'when options are given' do
            it 'sets initialization options' do
                expect(subject.forward( my_data: 'stuff2' ).my_data).to eq('stuff2')
            end
        end
    end

    describe '#forward_as' do
        subject { JobForwardTest.new( my_data: 'stuff' ) }

        it 'sets the original Job as the #forwarder' do
            id = subject.id
            expect(subject.forward_as( JobForwardAsTest ).forwarder).to eq(subject)
        end

        it 'creates a new Job type with the same #id' do
            expect(subject).not_to be_kind_of JobForwardAsTest

            id = subject.id

            forwarded = subject.forward_as( JobForwardAsTest )

            expect(forwarded.id).to eq(id)
            expect(forwarded).to be_kind_of JobForwardAsTest
        end

        it 'creates a new Job with the same #never_ending' do
            expect(subject.forward_as( JobForwardAsTest ).never_ending?).to be_falsey

            job = JobForwardTest.new( never_ending: true, my_data: 'stuff' )
            expect(job.never_ending?).to be_truthy
            expect(job.forward_as( JobForwardAsTest ).never_ending?).to be_truthy

            job = JobForwardTest.new( never_ending: false, my_data: 'stuff' )
            expect(job.never_ending?).to be_falsey
            expect(job.forward_as( JobForwardAsTest ).never_ending?).to be_falsey
        end

        it 'does not preserve any existing data' do
            expect(subject).not_to be_kind_of JobForwardAsTest

            forwarded = subject.forward_as( JobForwardAsTest )

            expect(forwarded.my_data).to be_nil
            expect(forwarded).to be_kind_of JobForwardAsTest
        end

        context 'when options are given' do
            it 'sets initialization options' do
                expect(subject).not_to be_kind_of JobForwardAsTest

                forwarded = subject.forward_as( JobForwardAsTest, my_data: 'stuff2' )

                expect(forwarded.my_data).to eq('stuff2')
                expect(forwarded).to be_kind_of JobForwardAsTest
            end
        end
    end

end