spec/lib/aggro_spec.rb in aggro-0.0.1 vs spec/lib/aggro_spec.rb in aggro-0.0.2

@@ -1,31 +1,75 @@
-require 'spec_helper'
+RSpec.describe Aggro do
+  describe '.data_dir' do
+    it 'should default to a tmp folder' do
+      Aggro.data_dir = nil
 
-describe Aggro do
-  describe '.initialize_hash_ring' do
-    context 'when called without args' do
-      it 'should initialize the hash ring from env variable' do
-        hash_ring = Aggro.initialize_hash_ring
-        expect(hash_ring.nodes).to include '10.0.0.1'
-        expect(hash_ring.nodes).to include '10.0.0.2'
+      expect(Aggro.data_dir).to eq './tmp/aggro'
+    end
+  end
+
+  describe '.node_list' do
+    let(:nodes) { { 'fluffy' => '10.0.0.50' } }
+    let(:is_server_node) { true }
+    let(:fake_config) do
+      double(
+        node_name: 'kittens',
+        nodes: nodes,
+        server_node?: is_server_node
+      )
+    end
+
+    before do
+      allow(Aggro).to receive(:cluster_config).and_return(fake_config)
+    end
+
+    it 'should return a node list' do
+      expect(Aggro.node_list).to be_a NodeList
+    end
+
+    it 'should initialize the node list with nodes from cluster config' do
+      expect(Aggro.node_list.nodes.map(&:id)).to include 'fluffy'
+    end
+
+    it 'should return the same node list every time' do
+      expect(Aggro.node_list).to eq Aggro.node_list
+    end
+
+    context 'configured to be a server node' do
+      it 'should initialize the node list with the local node' do
+        expect(Aggro.node_list.nodes.map(&:id)).to include 'kittens'
       end
     end
 
-    context 'when called with an array' do
-      it 'should initialize the hash ring from given array' do
-        hash_ring = Aggro.initialize_hash_ring ['10.0.0.50']
-        expect(hash_ring.nodes).to_not include '10.0.0.1'
-        expect(hash_ring.nodes).to include '10.0.0.50'
+    context 'not configured to be a server node' do
+      let(:is_server_node) { false }
+
+      it 'should not initialize the node list with the local node' do
+        expect(Aggro.node_list.nodes.map(&:id)).to_not include 'kittens'
       end
     end
   end
 
-  describe '.hash_ring' do
-    it 'should return a hash ring' do
-      expect(Aggro.hash_ring).to be_a ConsistentHashing::Ring
+  describe '.port' do
+    it 'should return $PORT as an integer' do
+      stub_const 'ENV', 'PORT' => '7000'
+
+      expect(Aggro.port).to eq 7000
     end
 
-    it 'should return the same hash ring every time' do
-      expect(Aggro.hash_ring).to eq Aggro.hash_ring
+    it 'should default to port 5000' do
+      stub_const 'ENV', {}
+
+      expect(Aggro.port).to eq 5000
+    end
+  end
+
+  describe '.cluster_config' do
+    it 'should return a cluster config' do
+      expect(Aggro.cluster_config).to be_a ClusterConfig
+    end
+
+    it 'should return the same config every time' do
+      expect(Aggro.cluster_config).to eq Aggro.cluster_config
     end
   end
 end