rambling-trie 2.2.1 → 2.3.1

data/spec/lib/rambling/trie/container_spec.rb

@@ -3,7 +3,8 @@
 require 'spec_helper'
 
 describe Rambling::Trie::Container do
-  let(:container) { Rambling::Trie::Container.new root, compressor }
+  subject(:container) { described_class.new root, compressor }
+
   let(:compressor) { Rambling::Trie::Compressor.new }
   let(:root) { Rambling::Trie::Nodes::Raw.new }
 
@@ -16,7 +17,7 @@ describe Rambling::Trie::Container do
       it 'yields the container' do
         yielded = nil
 
-        container = Rambling::Trie::Container.new root, compressor do |c|
+        container = described_class.new root, compressor do |c|
           yielded = c
         end
 
@@ -26,15 +27,18 @@ describe Rambling::Trie::Container do
   end
 
   describe '#add' do
+    # rubocop:disable RSpec/MultipleExpectations
     it 'adds the word to the root node' do
       add_word container, 'hello'
 
       expect(root.children.size).to eq 1
       expect(root.to_a).to eq %w(hello)
     end
+    # rubocop:enable RSpec/MultipleExpectations
   end
 
   describe '#concat' do
+    # rubocop:disable RSpec/MultipleExpectations
     it 'adds all the words to the root node' do
       container.concat %w(other words)
 
@@ -48,42 +52,7 @@ describe Rambling::Trie::Container do
       expect(nodes.first.letter).to eq :o
       expect(nodes.last.letter).to eq :w
     end
-  end
-
-  describe '#compress!' do
-    let(:node) { Rambling::Trie::Nodes::Compressed.new }
-
-    before do
-      allow(compressor).to receive(:compress).and_return node
-
-      add_word root, 'yes'
-      node[:yes] = Rambling::Trie::Nodes::Compressed.new
-    end
-
-    it 'compresses the trie using the compressor' do
-      container.compress!
-
-      expect(compressor).to have_received(:compress).with root
-    end
-
-    it 'changes to the root returned by the compressor' do
-      container.compress!
-
-      expect(container.root).not_to eq root
-      expect(container.root).to eq node
-    end
-
-    it 'returns itself' do
-      expect(container.compress!).to eq container
-    end
-
-    it 'does not compress multiple times' do
-      container.compress!
-      allow(node).to receive(:compressed?).and_return(true)
-
-      container.compress!
-      expect(compressor).to have_received(:compress).once
-    end
+    # rubocop:enable RSpec/MultipleExpectations
   end
 
   describe '#compress' do
@@ -102,12 +71,14 @@ describe Rambling::Trie::Container do
       expect(compressor).to have_received(:compress).with root
     end
 
+    # rubocop:disable RSpec/MultipleExpectations
     it 'returns a container with the new root' do
       new_container = container.compress
 
       expect(new_container.root).not_to eq root
       expect(new_container.root).to eq node
     end
+    # rubocop:enable RSpec/MultipleExpectations
 
     it 'returns a new container' do
       expect(container.compress).not_to eq container
@@ -128,153 +99,53 @@ describe Rambling::Trie::Container do
     end
   end
 
-  describe '#word?' do
-    let(:root) do
-      double :root,
-        compressed?: compressed,
-        word?: nil
-    end
+  describe '#compress!' do
+    let(:node) { Rambling::Trie::Nodes::Compressed.new }
 
-    context 'for an uncompressed root' do
-      let(:compressed) { true }
+    context 'with a mocked result' do
+      before do
+        allow(compressor).to receive(:compress).and_return node
 
-      it 'calls the root with the word characters' do
-        container.word? 'words'
-        expect(root).to have_received(:word?).with %w(w o r d s)
+        add_word root, 'yes'
+        node[:yes] = Rambling::Trie::Nodes::Compressed.new
       end
-    end
 
-    context 'for a compressed root' do
-      let(:compressed) { false }
+      it 'compresses the trie using the compressor' do
+        container.compress!
 
-      it 'calls the root with the full word' do
-        container.word? 'words'
-        expect(root).to have_received(:word?).with %w(w o r d s)
+        expect(compressor).to have_received(:compress).with root
       end
-    end
-  end
 
-  describe '#partial_word?' do
-    let(:root) do
-      double :root,
-        compressed?: compressed,
-        partial_word?: nil
-    end
-
-    context 'for an uncompressed root' do
-      let(:compressed) { true }
+      # rubocop:disable RSpec/MultipleExpectations
+      it 'changes to the root returned by the compressor' do
+        container.compress!
 
-      it 'calls the root with the word characters' do
-        container.partial_word? 'words'
-        expect(root).to have_received(:partial_word?).with %w(w o r d s)
+        expect(container.root).not_to eq root
+        expect(container.root).to eq node
       end
-    end
-
-    context 'for a compressed root' do
-      let(:compressed) { false }
+      # rubocop:enable RSpec/MultipleExpectations
 
-      it 'calls the root with the word characters' do
-        container.partial_word? 'words'
-        expect(root).to have_received(:partial_word?).with %w(w o r d s)
+      it 'returns itself' do
+        expect(container.compress!).to eq container
       end
-    end
-  end
-
-  describe 'delegates and aliases' do
-    before do
-      allow(root).to receive_messages(
-        :[] => nil,
-        add: nil,
-        as_word: nil,
-        children: nil,
-        children_tree: nil,
-        compressed?: nil,
-        each: nil,
-        key?: nil,
-        inspect: nil,
-        letter: nil,
-        parent: nil,
-        partial_word?: nil,
-        scan: nil,
-        size: nil,
-        to_s: nil,
-        word?: nil,
-      )
-    end
 
-    it 'aliases `#include?` to `#word?`' do
-      container.include? 'words'
-      expect(root).to have_received(:word?).with %w(w o r d s)
-    end
-
-    it 'aliases `#match?` to `#partial_word?`' do
-      container.match? 'words'
-      expect(root).to have_received(:partial_word?).with %w(w o r d s)
-    end
-
-    it 'aliases `#words` to `#scan`' do
-      container.words 'hig'
-      expect(root).to have_received(:scan).with %w(h i g)
-    end
-
-    it 'aliases `#<<` to `#add`' do
-      container << 'words'
-      expect(root).to have_received(:add).with %i(s d r o w)
-    end
-
-    it 'delegates `#[]` to the root node' do
-      container[:yep]
-      expect(root).to have_received(:[]).with :yep
-    end
-
-    it 'delegates `#children` to the root node' do
-      container.children
-      expect(root).to have_received :children
-    end
-
-    it 'delegates `#children_tree` to the root node' do
-      container.children_tree
-      expect(root).to have_received :children_tree
-    end
-
-    it 'delegates `#compressed?` to the root node' do
-      container.compressed?
-      expect(root).to have_received :compressed?
-    end
-
-    it 'delegates `#key?` to the root node' do
-      container.key? :yup
-      expect(root).to have_received(:key?).with :yup
-    end
-
-    it 'aliases `#has_key?` to `#key?`' do
-      container.has_key? :yup
-      expect(root).to have_received(:key?).with :yup
-    end
-
-    it 'aliases `#has_letter?` to `#has_key?`' do
-      container.has_letter? :yup
-      expect(root).to have_received(:key?).with :yup
-    end
-
-    it 'delegates `#inspect` to the root node' do
-      container.inspect
-      expect(root).to have_received :inspect
-    end
+      it 'does not compress multiple times' do
+        container.compress!
+        allow(node).to receive(:compressed?).and_return(true)
 
-    it 'delegates `#size` to the root node' do
-      container.size
-      expect(root).to have_received :size
-    end
-  end
+        container.compress!
+        expect(compressor).to have_received(:compress).once
+      end
 
-  describe '#compress!' do
-    it 'gets a new root from the compressor' do
-      container.compress!
+      # rubocop:disable RSpec/MultipleExpectations
+      it 'gets a new root from the compressor' do
+        container.compress!
 
-      expect(container.root).not_to be root
-      expect(container.root).to be_compressed
-      expect(root).not_to be_compressed
+        expect(container.root).not_to be root
+        expect(container.root).to be_compressed
+        expect(root).not_to be_compressed
+      end
+      # rubocop:enable RSpec/MultipleExpectations
     end
 
     it 'generates a new root with the words from the passed root' do
@@ -283,9 +154,7 @@ describe Rambling::Trie::Container do
       add_words container, words
       container.compress!
 
-      words.each do |word|
-        expect(container).to include word
-      end
+      words.each { |word| expect(container).to include word }
     end
 
     describe 'and trying to add a word' do
@@ -301,138 +170,91 @@ describe Rambling::Trie::Container do
   end
 
   describe '#word?' do
-    context 'word is contained' do
-      before do
-        add_words container, %w(hello high)
-      end
+    it_behaves_like 'a propagating node' do
+      let(:method) { :word? }
+    end
 
-      it 'matches the whole word' do
-        expect(container.word? 'hello').to be true
-        expect(container.word? 'high').to be true
-      end
+    context 'when word is contained' do
+      before { add_words container, %w(hello high) }
 
-      context 'and the root has been compressed' do
-        before do
-          container.compress!
-        end
+      it_behaves_like 'a matching container#word'
 
-        it 'matches the whole word' do
-          expect(container.word? 'hello').to be true
-          expect(container.word? 'high').to be true
-        end
+      context 'with compressed root' do
+        before { container.compress! }
+
+        it_behaves_like 'a matching container#word'
       end
     end
 
-    context 'word is not contained' do
-      before do
-        add_word container, 'hello'
-      end
+    context 'when word is not contained' do
+      before { add_word container, 'hello' }
 
-      it 'does not match the whole word' do
-        expect(container.word? 'halt').to be false
-        expect(container.word? 'al').to be false
-      end
+      it_behaves_like 'a non-matching container#word'
 
-      context 'and the root has been compressed' do
-        before do
-          container.compress!
-        end
+      context 'with compressed root' do
+        before { container.compress! }
 
-        it 'does not match the whole word' do
-          expect(container.word? 'halt').to be false
-          expect(container.word? 'al').to be false
-        end
+        it_behaves_like 'a non-matching container#word'
       end
     end
   end
 
   describe '#partial_word?' do
-    context 'word is contained' do
-      before do
-        add_words container, %w(hello high)
+    context 'with underlying node' do
+      it_behaves_like 'a propagating node' do
+        let(:method) { :partial_word? }
       end
+    end
 
-      it 'matches part of the word' do
-        expect(container.partial_word? 'hell').to be true
-        expect(container.partial_word? 'hig').to be true
-      end
+    context 'when word is contained' do
+      before { add_words container, %w(hello high) }
 
-      context 'and the root has been compressed' do
-        before do
-          container.compress!
-        end
+      it_behaves_like 'a matching container#partial_word'
 
-        it 'matches part of the word' do
-          expect(container.partial_word? 'h').to be true
-          expect(container.partial_word? 'he').to be true
-          expect(container.partial_word? 'hell').to be true
-          expect(container.partial_word? 'hello').to be true
-          expect(container.partial_word? 'hi').to be true
-          expect(container.partial_word? 'hig').to be true
-          expect(container.partial_word? 'high').to be true
-        end
-      end
-    end
+      context 'with compressed root' do
+        before { container.compress! }
 
-    shared_examples_for 'a non matching tree' do
-      it 'does not match any part of the word' do
-        %w(ha hal al).each do |word|
-          expect(container.partial_word? word).to be false
-        end
+        it_behaves_like 'a matching container#partial_word'
       end
     end
 
-    context 'word is not contained' do
-      before do
-        add_word container, 'hello'
-      end
+    context 'when word is not contained' do
+      before { add_word container, 'hello' }
 
-      context 'and the root is uncompressed' do
-        it_behaves_like 'a non matching tree'
-      end
+      it_behaves_like 'a non-matching container#partial_word'
 
-      context 'and the root has been compressed' do
-        it_behaves_like 'a non matching tree'
+      context 'with compressed root' do
+        before { container.compress! }
+
+        it_behaves_like 'a non-matching container#partial_word'
       end
     end
   end
 
   describe '#scan' do
-    context 'words that match are not contained' do
+    context 'when words that match are contained' do
       before do
         add_words container, %w(hi hello high hell highlight histerical)
       end
 
-      it 'returns an array with the words that match' do
-        expect(container.scan 'hi').to eq %w(hi high highlight histerical)
-        expect(container.scan 'hig').to eq %w(high highlight)
-      end
+      it_behaves_like 'a matching container#scan'
 
-      context 'and the root has been compressed' do
-        before do
-          container.compress!
-        end
+      context 'with compressed root' do
+        before { container.compress! }
 
-        it 'returns an array with the words that match' do
-          expect(container.scan 'hi').to eq %w(hi high highlight histerical)
-          expect(container.scan 'hig').to eq %w(high highlight)
-        end
+        it_behaves_like 'a matching container#scan'
       end
     end
 
-    context 'words that match are not contained' do
-      before do
-        add_word container, 'hello'
-      end
+    context 'when words that match are not contained' do
+      before { add_word container, 'hello' }
 
       it 'returns an empty array' do
         expect(container.scan 'hi').to eq %w()
       end
 
-      context 'and the root has been compressed' do
-        before do
-          container.compress!
-        end
+      context 'with compressed root' do
+        before { container.compress! }
 
         it 'returns an empty array' do
           expect(container.scan 'hi').to eq %w()
@@ -442,19 +264,15 @@ describe Rambling::Trie::Container do
   end
 
   describe '#words_within' do
-    before do
-      add_words container, %w(one word and other words)
-    end
+    before { add_words container, %w(one word and other words) }
 
-    context 'phrase does not contain any words' do
+    context 'when phrase does not contain any words' do
       it 'returns an empty array' do
         expect(container.words_within 'xyz').to match_array []
       end
 
-      context 'and the node is compressed' do
-        before do
-          container.compress!
-        end
+      context 'with compressed node' do
+        before { container.compress! }
 
         it 'returns an empty array' do
           expect(container.words_within 'xyz').to match_array []
@@ -462,33 +280,23 @@ describe Rambling::Trie::Container do
       end
     end
 
-    context 'phrase contains one word at the start of the phrase' do
-      it 'returns an array with the word found in the phrase' do
-        expect(container.words_within 'word').to match_array %w(word)
-        expect(container.words_within 'wordxyz').to match_array %w(word)
-      end
+    context 'when phrase contains one word at the start of the phrase' do
+      it_behaves_like 'a matching container#words_within'
 
-      context 'and the node is compressed' do
-        before do
-          container.compress!
-        end
+      context 'with compressed node' do
+        before { container.compress! }
 
-        it 'returns an array with the word found in the phrase' do
-          expect(container.words_within 'word').to match_array %w(word)
-          expect(container.words_within 'wordxyz').to match_array %w(word)
-        end
+        it_behaves_like 'a matching container#words_within'
       end
     end
 
-    context 'phrase contains one word at the end of the phrase' do
+    context 'when phrase contains one word at the end of the phrase' do
       it 'returns an array with the word found in the phrase' do
         expect(container.words_within 'xyz word').to match_array %w(word)
       end
 
-      context 'and the node is compressed' do
-        before do
-          container.compress!
-        end
+      context 'with compressed node' do
+        before { container.compress! }
 
         it 'returns an array with the word found in the phrase' do
           expect(container.words_within 'xyz word').to match_array %w(word)
@@ -496,48 +304,33 @@ describe Rambling::Trie::Container do
       end
     end
 
-    context 'phrase contains a few words' do
-      it 'returns an array with all words found in the phrase' do
-        expect(container.words_within 'xyzword otherzxyone')
-          .to match_array %w(word other one)
-      end
+    context 'when phrase contains a few words' do
+      it_behaves_like 'a non-matching container#words_within'
 
-      context 'and the node is compressed' do
-        before do
-          container.compress!
-        end
+      context 'with compressed node' do
+        before { container.compress! }
 
-        it 'returns an array with all words found in the phrase' do
-          expect(container.words_within 'xyzword otherzxyone')
-            .to match_array %w(word other one)
-        end
+        it_behaves_like 'a non-matching container#words_within'
       end
     end
   end
 
   describe '#words_within?' do
-    before do
-      add_words container, %w(one word and other words)
-    end
+    before { add_words container, %w(one word and other words) }
 
-    context 'phrase does not contain any words' do
-      it 'returns false' do
-        expect(container.words_within? 'xyz').to be false
-      end
-    end
+    it_behaves_like 'a matching container#words_within?'
 
-    context 'phrase contains any word' do
-      it 'returns true' do
-        expect(container.words_within? 'xyz words').to be true
-        expect(container.words_within? 'xyzone word').to be true
-      end
+    context 'with compressed node' do
+      before { container.compress! }
+
+      it_behaves_like 'a non-matching container#words_within?'
     end
   end
 
   describe '#==' do
     context 'when the root nodes are the same' do
       let(:other_container) do
-        Rambling::Trie::Container.new container.root, compressor
+        described_class.new container.root, compressor
       end
 
       it 'returns true' do
@@ -548,12 +341,10 @@ describe Rambling::Trie::Container do
     context 'when the root nodes are not the same' do
       let(:other_root) { Rambling::Trie::Nodes::Raw.new }
       let(:other_container) do
-        Rambling::Trie::Container.new other_root, compressor
+        described_class.new other_root, compressor
       end
 
-      before do
-        add_word other_container, 'hola'
-      end
+      before { add_word other_container, 'hola' }
 
       it 'returns false' do
         expect(container).not_to eq other_container
@@ -562,9 +353,7 @@ describe Rambling::Trie::Container do
   end
 
   describe '#each' do
-    before do
-      add_words container, %w(yes no why)
-    end
+    before { add_words container, %w(yes no why) }
 
     it 'returns an enumerator when no block is given' do
       expect(container.each).to be_instance_of Enumerator
@@ -576,9 +365,7 @@ describe Rambling::Trie::Container do
   end
 
   describe '#inspect' do
-    before do
-      add_words container, %w(a few words hello hell)
-    end
+    before { add_words container, %w(a few words hello hell) }
 
     it 'returns the container class name plus the root inspection' do
       expect(container.inspect).to eq one_line <<~CONTAINER
@@ -588,4 +375,92 @@ describe Rambling::Trie::Container do
       CONTAINER
     end
   end
+
+  describe 'delegates and aliases' do
+    before do
+      allow(root).to receive_messages(
+        :[] => nil,
+        add: nil,
+        as_word: nil,
+        children: nil,
+        children_tree: nil,
+        compressed?: nil,
+        each: nil,
+        key?: nil,
+        inspect: nil,
+        letter: nil,
+        parent: nil,
+        partial_word?: nil,
+        scan: nil,
+        size: nil,
+        to_s: nil,
+        word?: nil,
+      )
+    end
+
+    it 'aliases `#include?` to `#word?`' do
+      container.include? 'words'
+      expect(root).to have_received(:word?).with %w(w o r d s)
+    end
+
+    it 'aliases `#match?` to `#partial_word?`' do
+      container.match? 'words'
+      expect(root).to have_received(:partial_word?).with %w(w o r d s)
+    end
+
+    it 'aliases `#words` to `#scan`' do
+      container.words 'hig'
+      expect(root).to have_received(:scan).with %w(h i g)
+    end
+
+    it 'aliases `#<<` to `#add`' do
+      container << 'words'
+      expect(root).to have_received(:add).with %i(s d r o w)
+    end
+
+    it 'delegates `#[]` to the root node' do
+      container[:yep]
+      expect(root).to have_received(:[]).with :yep
+    end
+
+    it 'delegates `#children` to the root node' do
+      container.children
+      expect(root).to have_received :children
+    end
+
+    it 'delegates `#children_tree` to the root node' do
+      container.children_tree
+      expect(root).to have_received :children_tree
+    end
+
+    it 'delegates `#compressed?` to the root node' do
+      container.compressed?
+      expect(root).to have_received :compressed?
+    end
+
+    it 'delegates `#key?` to the root node' do
+      container.key? :yup
+      expect(root).to have_received(:key?).with :yup
+    end
+
+    it 'aliases `#has_key?` to `#key?`' do
+      container.has_key? :yup
+      expect(root).to have_received(:key?).with :yup
+    end
+
+    it 'aliases `#has_letter?` to `#has_key?`' do
+      container.has_letter? :yup
+      expect(root).to have_received(:key?).with :yup
+    end
+
+    it 'delegates `#inspect` to the root node' do
+      container.inspect
+      expect(root).to have_received :inspect
+    end
+
+    it 'delegates `#size` to the root node' do
+      container.size
+      expect(root).to have_received :size
+    end
+  end
 end