################################################################################
#
#      Author: Zachary Patten <zachary AT jovelabs DOT com>
#   Copyright: Copyright (c) Zachary Patten
#     License: Apache License, Version 2.0
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.
#
################################################################################

require "spec_helper"

describe ZTK::Parallel do

  subject { ZTK::Parallel.new }

  describe "class" do

    it "should be an instance of ZTK::Parallel" do
      subject.should be_an_instance_of ZTK::Parallel
    end

  end

  describe "methods" do


    describe "#process" do

      it "should throw an exception if the process method is called without a block" do
        lambda{ subject.process }.should raise_error ZTK::ParallelError, "You must supply a block to the process method!"
      end

      it "should spawn multiple processes to handle each iteration" do
        3.times do |x|
          subject.process do
            Process.pid
          end
        end

        subject.waitall

        subject.results.all?{ |r| r.should be_kind_of Integer }
        subject.results.all?{ |r| r.should > 0 }
        subject.results.uniq.count.should == 3
        subject.results.include?(Process.pid).should be false
      end

      it "should stop all execution when the ZTK::Parallel::Break exception is raised" do
        3.times do |x|
          subject.process do
            raise ZTK::Parallel::Break
          end
        end

        lambda { subject.waitall }.should raise_error
      end

      it "should stop all execution when any exception is raised" do
        3.times do |x|
          subject.process do
            raise "SomeException"
          end
        end

        lambda { subject.waitall }.should raise_error
      end

    end

    describe "#wait" do

      it "should be able to incrementally wait the forks" do
        3.times do |x|
          subject.process do
            Process.pid
          end
        end

        while subject.count > 0 do
          subject.wait
        end

        subject.results.all?{ |r| r.should be_kind_of Integer }
        subject.results.all?{ |r| r.should > 0 }
        subject.results.uniq.count.should == 3
        subject.results.include?(Process.pid).should be false
      end

    end

    describe "#waitall" do

      it "should be able to wait for all forks to stop" do
        3.times do |x|
          subject.process do
            Process.pid
          end
        end

        subject.waitall
        subject.count.should == 0
      end

    end

    describe "#count" do

      it "should return the number of active forked processes" do
        3.times do |x|
          subject.process do
            sleep(WAIT_SMALL)
          end
        end

        subject.count.should == 3
        subject.waitall
      end

    end

  end

end