cellect-server 0.0.1

data/spec/fixtures/user_data/new_user.json

@@ -0,0 +1,26 @@
+{
+  "random": [
+
+  ],
+  "priority": [
+
+  ],
+  "grouped_random": [
+
+  ],
+  "grouped_priority": [
+
+  ],
+  "pairwise_random": [
+
+  ],
+  "pairwise_priority": [
+
+  ],
+  "grouped_pairwise_random": [
+
+  ],
+  "grouped_pairwise_priority": [
+
+  ]
+}

data/spec/fixtures/user_data/partial_user.json

@@ -0,0 +1,58 @@
+{
+  "random": [
+    1,
+    2,
+    3,
+    4,
+    5
+  ],
+  "priority": [
+    7,
+    6,
+    1,
+    3,
+    10
+  ],
+  "grouped_random": [
+    1,
+    2,
+    3,
+    4,
+    5
+  ],
+  "grouped_priority": [
+    8,
+    7,
+    10,
+    5,
+    6
+  ],
+  "pairwise_random": [
+    1,
+    2,
+    3,
+    4,
+    5
+  ],
+  "pairwise_priority": [
+    10,
+    6,
+    5,
+    2,
+    4
+  ],
+  "grouped_pairwise_random": [
+    1,
+    2,
+    3,
+    4,
+    5
+  ],
+  "grouped_pairwise_priority": [
+    16,
+    17,
+    3,
+    8,
+    7
+  ]
+}

data/spec/server/api/add_seen_spec.rb

@@ -0,0 +1,26 @@
+require 'spec_helper'
+
+module Cellect::Server
+  describe API do
+    include_context 'API'
+    
+    { 'Ungrouped' => nil, 'Grouped' => 'grouped' }.each_pair do |grouping_type, grouping|
+      SET_TYPES.shuffle.each do |set_type|
+        context "#{ grouping_type } #{ set_type }" do
+          let(:project_type){ [grouping, set_type].compact.join '_' }
+          let(:project){ Project[project_type] }
+          let(:user){ project.user 123 }
+          before(:each){ pass_until project, is: :ready }
+          
+          it 'should add seen subjects' do
+            async_project = double
+            project.should_receive(:async).and_return async_project
+            async_project.should_receive(:add_seen_for).with 123, 123
+            put "/projects/#{ project_type }/users/123/add_seen", subject_id: 123
+            last_response.status.should == 200
+          end
+        end
+      end
+    end
+  end
+end

data/spec/server/api/add_spec.rb

@@ -0,0 +1,40 @@
+require 'spec_helper'
+
+module Cellect::Server
+  describe API do
+    include_context 'API'
+    
+    { 'Ungrouped' => nil, 'Grouped' => 'grouped' }.each_pair do |grouping_type, grouping|
+      SET_TYPES.shuffle.each do |set_type|
+        context "#{ grouping_type } #{ set_type }" do
+          let(:project_type){ [grouping, set_type].compact.join '_' }
+          let(:project){ Project[project_type] }
+          let(:user){ project.user 123 }
+          before(:each){ pass_until project, is: :ready }
+          
+          let(:opts) do
+            { subject_id: 123 }.tap do |h|
+              h[:priority] = 456.0 if project.prioritized?
+              h[:group_id] = 1 if project.grouped?
+            end
+          end
+          
+          it 'should add subjects' do
+            if project.grouped? && project.prioritized?
+              project.should_receive(:add).with subject_id: 123, group_id: 1, priority: 456.0
+            elsif project.grouped?
+              project.should_receive(:add).with subject_id: 123, group_id: 1, priority: nil
+            elsif project.prioritized?
+              project.should_receive(:add).with subject_id: 123, group_id: nil, priority: 456.0
+            else
+              project.should_receive(:add).with subject_id: 123, group_id: nil, priority: nil
+            end
+            
+            put "/projects/#{ project_type }/add", opts
+            last_response.status.should == 200
+          end
+        end
+      end
+    end
+  end
+end

data/spec/server/api/remove_spec.rb

@@ -0,0 +1,35 @@
+require 'spec_helper'
+
+module Cellect::Server
+  describe API do
+    include_context 'API'
+    
+    { 'Ungrouped' => nil, 'Grouped' => 'grouped' }.each_pair do |grouping_type, grouping|
+      SET_TYPES.shuffle.each do |set_type|
+        context "#{ grouping_type } #{ set_type }" do
+          let(:project_type){ [grouping, set_type].compact.join '_' }
+          let(:project){ Project[project_type] }
+          let(:user){ project.user 123 }
+          before(:each){ pass_until project, is: :ready }
+          
+          let(:opts) do
+            { subject_id: 123 }.tap do |h|
+              h[:group_id] = 1 if project.grouped?
+            end
+          end
+          
+          it 'should remove subjects' do
+            if project.grouped?
+              project.should_receive(:remove).with subject_id: 123, group_id: 1, priority: nil
+            else
+              project.should_receive(:remove).with subject_id: 123, group_id: nil, priority: nil
+            end
+            
+            put "/projects/#{ project_type }/remove", opts
+            last_response.status.should == 200
+          end
+        end
+      end
+    end
+  end
+end

data/spec/server/api/sample_spec.rb

@@ -0,0 +1,34 @@
+require 'spec_helper'
+
+module Cellect::Server
+  describe API do
+    include_context 'API'
+    
+    { 'Ungrouped' => nil, 'Grouped' => 'grouped' }.each_pair do |grouping_type, grouping|
+      SET_TYPES.shuffle.each do |set_type|
+        context "#{ grouping_type } #{ set_type }" do
+          let(:project_type){ [grouping, set_type].compact.join '_' }
+          let(:project){ Project[project_type] }
+          let(:user){ project.user 123 }
+          before(:each){ pass_until project, is: :ready }
+          
+          it 'should sample without a user, limit, or group' do
+            project.should_receive(:sample).with(limit: 5, user_id: nil, group_id: nil).and_call_original
+            get "/projects/#{ project_type }"
+            last_response.status.should == 200
+            json.should be_a Array
+          end
+          
+          shoulda = grouping ? 'limit, group, and user' : 'limit and user'
+          it "should sample with a #{ shoulda }" do
+            group_id = grouping ? 1 : nil
+            project.should_receive(:sample).with(limit: 3, user_id: 123, group_id: group_id).and_call_original
+            get "/projects/#{ project_type }?limit=3&user_id=123#{ grouping ? '&group_id=1' : '' }"
+            last_response.status.should == 200
+            json.should be_a Array
+          end
+        end
+      end
+    end
+  end
+end

data/spec/server/api/user_load_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+
+module Cellect::Server
+  describe API do
+    include_context 'API'
+    
+    { 'Ungrouped' => nil, 'Grouped' => 'grouped' }.each_pair do |grouping_type, grouping|
+      SET_TYPES.shuffle.each do |set_type|
+        context "#{ grouping_type } #{ set_type }" do
+          let(:project_type){ [grouping, set_type].compact.join '_' }
+          let(:project){ Project[project_type] }
+          before(:each){ pass_until project, is: :ready }
+          
+          it 'should load users' do
+            async_project = double
+            project.should_receive(:async).and_return async_project
+            async_project.should_receive(:user).with 123
+            post "/projects/#{ project_type }/users/123/load"
+            last_response.status.should == 201
+          end
+        end
+      end
+    end
+  end
+end

data/spec/server/grouped_project_spec.rb

@@ -0,0 +1,76 @@
+require 'spec_helper'
+
+module Cellect::Server
+  describe GroupedProject do
+    SET_TYPES.collect{ |type| "grouped_#{ type }" }.each do |project_type|
+      context project_type do
+        it_behaves_like 'project', :project
+        let(:project){ GroupedProject[project_type] }
+        let(:user){ project.user 123 }
+        let(:set_klass){ project.prioritized? ? DiffSet::PrioritySet : DiffSet::RandomSet }
+        before(:each){ pass_until project, is: :ready }
+        
+        it 'should provide unseen from a random group for users' do
+          project.groups = { }
+          project.groups[1] = set_klass.new
+          project.groups[1].should_receive(:subtract).with user.seen, 3
+          project.unseen_for 123, limit: 3
+        end
+        
+        it 'should provide unseen from a specific group for users' do
+          3.times{ |i| project.groups[i] = set_klass.new }
+          project.group(1).should_receive(:subtract).with user.seen, 3
+          project.unseen_for 123, group_id: 1, limit: 3
+        end
+        
+        it 'should sample subjects from a random group without a user' do
+          project.groups = { }
+          project.groups[1] = set_klass.new
+          project.group(1).should_receive(:sample).with 3
+          project.sample limit: 3
+        end
+        
+        it 'should sample subjects from a specific group without a user' do
+          3.times{ |i| project.groups[i] = set_klass.new }
+          project.group(1).should_receive(:sample).with 3
+          project.sample group_id: 1, limit: 3
+        end
+        
+        it 'should sample subjects from a random group for a user' do
+          project.groups = { }
+          project.groups[1] = set_klass.new
+          project.groups[1].should_receive(:subtract).with user.seen, 3
+          project.sample user_id: 123, limit: 3
+        end
+        
+        it 'should sample subjects from a specific group for a user' do
+          3.times{ |i| project.groups[i] = set_klass.new }
+          project.group(1).should_receive(:subtract).with user.seen, 3
+          project.sample user_id: 123, group_id: 1, limit: 3
+        end
+        
+        it 'should add subjects' do
+          project.groups[1] = set_klass.new
+          
+          if project.prioritized?
+            project.groups[1].should_receive(:add).with 123, 456
+            project.add subject_id: 123, group_id: 1, priority: 456
+          else
+            project.groups[1].should_receive(:add).with 123
+            project.add subject_id: 123, group_id: 1
+          end
+        end
+        
+        it 'should remove subjects' do
+          project.groups[1] = set_klass.new
+          project.groups[1].should_receive(:remove).with 123
+          project.remove subject_id: 123, group_id: 1
+        end
+        
+        it 'should be grouped' do
+          project.should be_grouped
+        end
+      end
+    end
+  end
+end

data/spec/server/node_set_spec.rb

@@ -0,0 +1,13 @@
+require 'spec_helper'
+
+module Cellect::Server
+  describe NodeSet do
+    it_behaves_like 'node set'
+    let(:node_set){ Cellect::Server.node_set.actors.first }
+    
+    it 'should register this node' do
+      node_set.id.should == 'node0000000000'
+      node_set.zk.get('/nodes/node0000000000').first.should =~ /\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}/
+    end
+  end
+end

data/spec/server/project_spec.rb

@@ -0,0 +1,62 @@
+require 'spec_helper'
+
+module Cellect::Server
+  describe Project do
+    SET_TYPES.each do |project_type|
+      context project_type do
+        it_behaves_like 'project', :project
+        let(:project){ Project[project_type] }
+        let(:user){ project.user 123 }
+        before(:each){ pass_until project, is: :ready }
+        
+        it 'should provide unseen for users' do
+          project.subjects.should_receive(:subtract).with user.seen, 3
+          project.unseen_for 123, limit: 3
+        end
+        
+        it 'should sample subjects without a user' do
+          project.subjects.should_receive(:sample).with 3
+          project.sample limit: 3
+        end
+        
+        it 'should sample subjects with a user' do
+          project.subjects.should_receive(:subtract).with user.seen, 3
+          project.sample user_id: 123, limit: 3
+        end
+        
+        it 'should add subjects' do
+          if project.prioritized?
+            project.subjects.should_receive(:add).with 123, 456
+            project.add subject_id: 123, priority: 456
+          else
+            project.subjects.should_receive(:add).with 123
+            project.add subject_id: 123
+          end
+        end
+        
+        it 'should remove subjects' do
+          project.subjects.should_receive(:add).with 123
+          project.add subject_id: 123
+        end
+        
+        it 'should be notified of a user ttl expiry' do
+          async_project = double
+          project.should_receive(:async).and_return async_project
+          async_project.should_receive(:remove_user).with user.id
+          user.ttl_expired!
+        end
+        
+        it 'should remove users when their ttl expires' do
+          id = user.id
+          project.remove_user id
+          project.users.should_not have_key id
+          expect{ user.id }.to raise_error
+        end
+        
+        it 'should not be grouped' do
+          project.should_not be_grouped
+        end
+      end
+    end
+  end
+end

data/spec/server/server_spec.rb

@@ -0,0 +1,24 @@
+require 'spec_helper'
+
+module Cellect::Server
+  describe Cellect do
+    context 'default adapter' do
+      let(:default){ Cellect::Server::Adapters::Default.new }
+      
+      it 'should raise a NotImplementedError when using the default adapter' do
+        expect{ default.project_list }.to raise_error NotImplementedError
+        expect{ default.load_data_for(Project.new('test')) }.to raise_error NotImplementedError
+        expect{ default.load_user 'random', 123 }.to raise_error NotImplementedError
+      end
+      
+      it 'should return a project given a set of options' do
+        default.project_for('name' => 'a').should be_an_instance_of Project
+        default.project_for('name' => 'b', 'grouped' => true).should be_an_instance_of GroupedProject
+        default.project_for('name' => 'c', 'pairwise' => true).should be_pairwise
+        default.project_for('name' => 'd', 'prioritized' => true).should be_prioritized
+        default.project_for('name' => 'e', 'pairwise' => true, 'prioritized' => true).should be_pairwise
+        default.project_for('name' => 'e', 'pairwise' => true, 'prioritized' => true).should be_prioritized
+      end
+    end
+  end
+end

data/spec/server/user_spec.rb

@@ -0,0 +1,32 @@
+require 'spec_helper'
+
+module Cellect::Server
+  describe User do
+    let(:user){ User.new 1, project_name: 'random' }
+    
+    it 'should store seen ids' do
+      user.seen.should be_a DiffSet::RandomSet
+    end
+    
+    it 'should have a default ttl of 15 minutes' do
+      user.ttl.should == 900 # seconds
+    end
+    
+    it 'should allow custom ttl' do
+      User.new(2, project_name: 'random', ttl: 123).ttl.should == 123
+    end
+    
+    it 'should reset the ttl timer on activity' do
+      user.bare_object.should_receive(:restart_ttl_timer).at_least :once
+      user.seen
+    end
+    
+    it 'should terminate on ttl expiry' do
+      async_project = double
+      Project[user.project_name].should_receive(:async).and_return async_project
+      async_project.should_receive(:remove_user).with user.id
+      user.ttl_expired!
+      user.ttl_timer.should be_nil
+    end
+  end
+end