spec/predictionio_spec.rb

require 'predictionio'
require 'spec_helper'

client = PredictionIO::Client.new('foobar', 10, 'http://fakeapi.com:8000')
event_client = PredictionIO::EventClient.new(1, 'http://fakeapi.com:8000', 10)
engine_client = PredictionIO::EngineClient.new('http://fakeapi.com:8000', 10)

describe PredictionIO do
  describe 'Events API' do
    it 'create_event should create an event' do
      response = event_client.create_event('register', 'user', 'foobar')
      expect(response).to_not raise_error
    end
    it 'set_user should set user properties' do
      response = event_client.set_user('foobar')
      expect(response).to_not raise_error
    end
    it 'unset_user should unset user properties' do
      response = event_client.unset_user('foobar',
                                         'properties' => { 'bar' => 'baz' })
      expect(response).to_not raise_error
    end
    it 'set_item should set item properties' do
      response = event_client.set_item('foobar')
      expect(response).to_not raise_error
    end
    it 'unset_item should unset item properties' do
      response = event_client.unset_item('foobar',
                                         'properties' => { 'bar' => 'baz' })
      expect(response).to_not raise_error
    end
    it 'record_user_action_on_item should record a U2I action' do
      response = event_client.record_user_action_on_item(
        'greet', 'foobar', 'barbaz', 'properties' => { 'dead' => 'beef' })
      expect(response).to_not raise_error
    end
    it 'delete_user should delete a user' do
      response = event_client.delete_user('foobar')
      expect(response).to_not raise_error
    end
    it 'delete_item should delete an item' do
      response = event_client.delete_item('foobar')
      expect(response).to_not raise_error
    end
  end

  describe 'Engine Client' do
    it 'send_query should get predictions' do
      predictions = engine_client.send_query('uid' => 'foobar')
      expect(predictions).to eq('iids' => %w(dead beef))
    end
  end

  describe 'Users API' do
    it 'create_user should create a user' do
      response = client.create_user('foo')
      expect(response).to_not raise_error
    end
    it 'get_user should get a user' do
      response = client.get_user('foo')
      expect(response).to eq('pio_uid' => 'foo')
    end
    it 'delete_user should delete a user' do
      response = client.delete_user('foo')
      expect(response).to_not raise_error
    end
  end

  describe 'Items API' do
    it 'create_item should create a item' do
      response = client.create_item('bar', %w(dead beef))
      expect(response).to_not raise_error
    end
    it 'get_item should get a item' do
      response = client.get_item('bar')
      expect(response).to eq('pio_iid' => 'bar',
                             'pio_itypes' => %w(dead beef))
    end
    it 'delete_item should delete a item' do
      response = client.delete_item('bar')
      expect(response).to_not raise_error
    end
  end

  describe 'U2I API' do
    it 'record_action_on_item should record an action' do
      client.identify('foo')
      response = client.record_action_on_item('view', 'bar')
      expect(response).to_not raise_error
    end
  end

  describe 'Item Recommendation API' do
    it 'provides recommendations to a user without attributes' do
      client.identify('foo')
      response = client.get_itemrec_top_n('itemrec-engine', 10)
      expect(response).to eq(%w(x y z))
    end
    it 'provides recommendations to a user with attributes' do
      client.identify('foo')
      response = client.get_itemrec_top_n('itemrec-engine', 10,
                                          'pio_attributes' => 'name')
      expect(response).to eq([
        { 'pio_iid' => 'x', 'name' => 'a' },
        { 'pio_iid' => 'y', 'name' => 'b' },
        { 'pio_iid' => 'z', 'name' => 'c' }])
    end
  end

  describe 'Item Rank API' do
    it 'provides ranking to a user without attributes' do
      client.identify('foo')
      response = client.get_itemrank_ranked('itemrank-engine', %w(y z x))
      expect(response).to eq(%w(x y z))
    end
    it 'provides ranking to a user with attributes' do
      client.identify('foo')
      response = client.get_itemrank_ranked('itemrank-engine', %w(y x z),
                                            'pio_attributes' => 'name')
      expect(response).to eq([
        { 'pio_iid' => 'x', 'name' => 'a' },
        { 'pio_iid' => 'y', 'name' => 'b' },
        { 'pio_iid' => 'z', 'name' => 'c' }])
    end
  end

  describe 'Item Similarity API' do
    it 'provides similarities to an item without attributes' do
      response = client.get_itemsim_top_n('itemsim-engine', 'bar', 10)
      expect(response).to eq(%w(x y z))
    end
    it 'provides similarities to an item with attributes' do
      response = client.get_itemsim_top_n('itemsim-engine', 'bar', 10,
                                          'pio_attributes' => 'name')
      expect(response).to eq([
        { 'pio_iid' => 'x', 'name' => 'a' },
        { 'pio_iid' => 'y', 'name' => 'b' },
        { 'pio_iid' => 'z', 'name' => 'c' }])
    end
  end
end