require 'spec_helper'
require 'wordlist/operators/union'

describe Wordlist::Operators::Union do
  let(:left)  { %w[foo bar] }
  let(:right) { %w[bar baz] }

  subject { described_class.new(left,right) }

  describe "#each" do
    context "when given a block" do
      it "must yield the words from both wordlists, without duplicates" do
        expect { |b|
          subject.each(&b)
        }.to yield_successive_args(*(left | right))
      end

      context "when the wordlists do not have any common words" do
        let(:left)  { %w[foo bar] }
        let(:right) { %w[baz qux] }

        it "must yield words from both wordlists" do
          expect { |b|
            subject.each(&b)
          }.to yield_successive_args(*(left + right))
        end
      end
    end

    context "when not given a block" do
      it "must return an Enumerator for the #each" do
        expect(subject.each).to be_kind_of(Enumerator)
        expect(subject.each.to_a).to eq(left | right)
      end
    end
  end
end