atig 0.0.1

data/spec/db/sized_uniq_array_spec.rb

@@ -0,0 +1,63 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require File.expand_path( '../../spec_helper', __FILE__ )
+require 'atig/db/sized_uniq_array'
+require 'ostruct'
+
+describe Atig::Db::SizedUniqArray do
+  def item(id)
+    item = stub "Item-#{id}"
+    item.stub!(:id).and_return id
+    item
+  end
+
+  before do
+    @array = Atig::Db::SizedUniqArray.new 3
+    @item1 = item 1
+    @item2 = item 2
+    @item3 = item 3
+    @item4 = item 4
+
+    @array << @item1
+    @array << @item2
+    @array << @item3
+  end
+
+  it "should include items" do
+    @array.to_a.should == [ @item1, @item2, @item3 ]
+  end
+
+  it "should rorate array" do
+    @array << @item4
+    @array.to_a.should == [ @item2, @item3, @item4 ]
+  end
+
+  it "should have reverse_each" do
+    xs = []
+    @array.reverse_each {|x| xs << x }
+    xs.should == [ @item3, @item2, @item1 ]
+  end
+
+  it "should not have duplicate element" do
+    @array << item(1)
+    @array.to_a.should == [ @item1, @item2, @item3 ]
+  end
+
+  it "should be accesible by index" do
+    @array[0].should == @item1
+    @array[1].should == @item2
+    @array[2].should == @item3
+  end
+
+  it "should not change index" do
+    @array << @item4
+    @array[0].should == @item4
+    @array[1].should == @item2
+    @array[2].should == @item3
+  end
+
+  it "should return index when add element" do
+    (@array << @item4).should == 0
+    (@array << @item3).should == nil
+  end
+end

data/spec/db/statuses_spec.rb

@@ -0,0 +1,180 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require File.expand_path( '../../spec_helper', __FILE__ )
+require 'fileutils'
+require 'atig/db/statuses'
+
+describe Atig::Db::Statuses do
+  def status(id, text, time)
+    OpenStruct.new(:id => id, :text => text, :created_at => time.strftime("%a %b %d %H:%M:%S +0000 %Y"))
+  end
+
+  def user(name)
+    OpenStruct.new(:screen_name => name, :id => name)
+  end
+
+  before do
+    @listened = []
+    FileUtils.rm_f 'test.db'
+    @db = Atig::Db::Statuses.new 'test.db'
+
+    @a = status 10, 'a', Time.utc(2010,1,5)
+    @b = status 20, 'b', Time.utc(2010,1,6)
+    @c = status 30, 'c', Time.utc(2010,1,7)
+    @d = status 40, 'd', Time.utc(2010,1,8)
+
+    @alice = user 'alice'
+    @bob = user 'bob'
+
+    @db.add :status => @a , :user => @alice, :source => :srcA
+    @db.add :status => @b , :user => @bob  , :source => :srcB
+    @db.add :status => @c , :user => @alice, :source => :srcC
+  end
+
+  it "should be re-openable" do
+    Atig::Db::Statuses.new 'test.db'
+  end
+
+  it "should call listeners" do
+    entry = nil
+    @db.listen{|x| entry = x }
+
+    @db.add :status => @d, :user => @alice, :source => :timeline, :fuga => :hoge
+
+    entry.source.should == :timeline
+    entry.status.should == @d
+    entry.tid.should match(/\w+/)
+    entry.sid.should match(/\w+/)
+    entry.user.should   == @alice
+    entry.source.should == :timeline
+    entry.fuga.should == :hoge
+  end
+
+  it "should not contain duplicate" do
+    called = false
+    @db.listen{|*_| called = true }
+
+    @db.add :status => @c, :user => @bob, :source => :timeline
+    called.should be_false
+  end
+
+  it "should be found by id" do
+    entry = @db.find_by_id 1
+    entry.id.should == 1
+    entry.status.should == @a
+    entry.user  .should == @alice
+    entry.tid   .should match(/\w+/)
+    entry.sid.should match(/\w+/)
+  end
+
+  it "should have unique tid" do
+    db = Atig::Db::Statuses.new 'test.db'
+    db.add :status => @d , :user => @alice, :source => :srcA
+
+    a = @db.find_by_id(1)
+    d = @db.find_by_id(4)
+    a.tid.should_not == d.tid
+    a.sid.should_not == d.cid
+  end
+
+  it "should be found all" do
+    db = @db.find_all
+    db.size.should == 3
+    a,b,c = db
+
+    a.status.should == @c
+    a.user  .should == @alice
+    a.tid   .should match(/\w+/)
+    a.sid   .should match(/\w+/)
+
+    b.status.should == @b
+    b.user  .should == @bob
+    b.tid   .should match(/\w+/)
+    b.sid   .should match(/\w+/)
+
+    c.status.should == @a
+    c.user.should   == @alice
+    c.tid.should    match(/\w+/)
+    c.sid.should    match(/\w+/)
+  end
+
+  it "should be found by tid" do
+    entry = @db.find_by_id(1)
+    @db.find_by_tid(entry.tid).should == entry
+  end
+
+  it "should be found by sid" do
+    entry = @db.find_by_id(1)
+    @db.find_by_sid(entry.sid).should == entry
+  end
+
+  it "should be found by tid" do
+    @db.find_by_tid('__').should be_nil
+  end
+
+  it "should be found by user" do
+    a,b = *@db.find_by_user(@alice)
+
+    a.status.should == @c
+    a.user  .should == @alice
+    a.tid   .should match(/\w+/)
+    a.sid   .should match(/\w+/)
+
+    b.status.should == @a
+    b.user.should   == @alice
+    b.tid.should    match(/\w+/)
+    b.sid.should    match(/\w+/)
+  end
+
+  it "should be found by screen_name" do
+    db = @db.find_by_screen_name('alice')
+    db.size.should == 2
+    a,b = db
+
+    a.status.should == @c
+    a.user  .should == @alice
+    a.tid   .should match(/\w+/)
+    a.sid   .should match(/\w+/)
+
+    b.status.should == @a
+    b.user.should   == @alice
+    b.tid.should    match(/\w+/)
+    b.sid.should    match(/\w+/)
+  end
+
+  it "should be found by screen_name with limit" do
+    xs = @db.find_by_screen_name('alice', :limit => 1)
+    xs.size.should == 1
+
+    a,_ = xs
+    a.status.should == @c
+    a.user  .should == @alice
+    a.tid   .should match(/\w+/)
+    a.sid   .should match(/\w+/)
+  end
+
+  it "should remove by id" do
+    @db.remove_by_id 1
+    @db.find_by_id(1).should be_nil
+  end
+
+  it "should have uniq tid/sid when removed" do
+    old = @db.find_by_id 3
+    @db.remove_by_id 3
+    @db.add :status => @c , :user => @alice, :source => :src
+    new = @db.find_by_id 3
+
+    old.tid.should_not == new.tid
+    old.sid.should_not == new.sid
+  end
+
+  it "should cleanup" do
+    Atig::Db::Statuses::Size = 10 unless defined? Atig::Db::Statuses::Size # hack 
+    Atig::Db::Statuses::Size.times do|i|
+      s = status i+100, 'a', Time.utc(2010,1,5)+i+1
+      @db.add :status => s , :user => @alice  , :source => :srcB
+    end
+    @db.cleanup
+    @db.find_by_status_id(@a.id).should == nil
+  end
+end

data/spec/ifilter/expand_url_spec.rb

@@ -0,0 +1,44 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require File.expand_path( '../../spec_helper', __FILE__ )
+require 'atig/ifilter/expand_url'
+require 'atig/twitter_struct'
+
+class Atig::IFilter::ExpandUrl
+  def resolve_http_redirect(uri); "[#{uri}]" end
+end
+
+describe Atig::IFilter::ExpandUrl, "when disable whole url" do
+  def filtered(text)
+    ifilter = Atig::IFilter::ExpandUrl.new OpenStruct.new(:log=>mock('log'),:opts=>OpenStruct.new)
+    ifilter.call status(text)
+  end
+
+  it "should expand bit.ly" do
+    filtered("This is http://bit.ly/hoge").should be_text("This is [http://bit.ly/hoge]")
+  end
+
+  it "should through other url" do
+    filtered("http://example.com").should be_text("http://example.com")
+  end
+end
+
+describe Atig::IFilter::ExpandUrl, "when enable whole url" do
+  def filtered(text)
+    context = OpenStruct.new(
+                             :log  => mock('log'),
+                             :opts => OpenStruct.new(:untiny_whole_urls=>true))
+    ifilter = Atig::IFilter::ExpandUrl.new(context)
+    ifilter.call status(text)
+  end
+
+  it "should expand bit.ly" do
+    filtered("This is http://bit.ly/hoge").should be_text("This is [http://bit.ly/hoge]")
+  end
+
+  it "should expand other url" do
+    filtered("http://example.com").should be_text("[http://example.com]")
+    filtered("https://example.com").should be_text("[https://example.com]")
+  end
+end
+

data/spec/ifilter/retweet_spec.rb

@@ -0,0 +1,28 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require File.expand_path( '../../spec_helper', __FILE__ )
+require 'atig/ifilter/retweet'
+require 'atig/twitter_struct'
+
+describe Atig::IFilter::Retweet do
+  def filtered(text, opt={})
+    Atig::IFilter::Retweet.call status(text, opt)
+  end
+
+  before do
+    @rt = Atig::IFilter::Retweet::Prefix
+  end
+
+  it "should throw normal status" do
+    filtered("hello").should be_text("hello")
+  end
+
+  it "should prefix RT for Retweet" do
+    filtered("RT: hello...",
+             'retweeted_status'=>{ 'text' => 'hello',
+               'user' => {
+                 'screen_name' => 'mzp'
+               } }).
+      should be_text("#{@rt}RT @mzp: hello")
+  end
+end

data/spec/ifilter/retweet_time_spec.rb

@@ -0,0 +1,25 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require File.expand_path( '../../spec_helper', __FILE__ )
+require 'atig/ifilter/retweet_time'
+require 'atig/twitter_struct'
+
+describe Atig::IFilter::RetweetTime do
+  def filtered(text, opt={})
+    Atig::IFilter::RetweetTime.call status(text, opt)
+  end
+
+  it "should throw normal status" do
+    filtered("hello").should be_text("hello")
+  end
+
+  it "should prefix RT for Retweet" do
+    filtered("RT @mzp: hello",
+             'retweeted_status'=>{ 'text' => 'hello',
+               'created_at' => 'Sat Sep 25 14:33:19 +0000 2010',
+               'user' => {
+                 'screen_name' => 'mzp'
+               } }).
+      should be_text("#{@rt}RT @mzp: hello \x0310[2010-09-25 23:33]\x0F")
+  end
+end

data/spec/ifilter/sanitize_spec.rb

@@ -0,0 +1,25 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require File.expand_path( '../../spec_helper', __FILE__ )
+require 'atig/ifilter/sanitize'
+require 'atig/twitter_struct'
+
+describe Atig::IFilter::Sanitize do
+  def filtered(text)
+    Atig::IFilter::Sanitize.call status(text)
+  end
+
+  it "should convert escape html" do
+    filtered("&lt; &gt; &quot;").should be_text("< > \"")
+  end
+
+  it "should convert whitespace" do
+    filtered("\r\n").should be_text(" ")
+    filtered("\r").should be_text(" ")
+    filtered("\n").should be_text(" ")
+  end
+
+  it "should delete \\000\\001 sequence" do
+    filtered("\000\001").should be_text("")
+  end
+end

data/spec/ifilter/sid_spec.rb

@@ -0,0 +1,29 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require File.expand_path( '../../spec_helper', __FILE__ )
+require 'atig/ifilter/xid'
+require 'atig/twitter_struct'
+
+describe Atig::IFilter::Sid, "when disable tid" do
+  def filtered(text)
+    ifilter = Atig::IFilter::Sid.new(OpenStruct.new(:log=>mock('log'),
+                                                    :opts=>OpenStruct.new))
+    ifilter.call status(text,'sid'=>1)
+  end
+
+  it "should through text" do
+    filtered("hello").should be_text("hello")
+  end
+end
+
+describe Atig::IFilter::Sid, "when enable tid" do
+  def filtered(text)
+    ifilter = Atig::IFilter::Sid.new(OpenStruct.new(:log=>mock('log'),
+                                                    :opts=>OpenStruct.new(:sid=>true)))
+    ifilter.call status(text,'sid'=>1)
+  end
+
+  it "should append sid" do
+    filtered("hello").should be_text("hello \x0310[1]\x0F")
+  end
+end

data/spec/ifilter/strip_spec.rb

@@ -0,0 +1,23 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require File.expand_path( '../../spec_helper', __FILE__ )
+require 'atig/ifilter/strip'
+require 'atig/twitter_struct'
+
+describe Atig::IFilter::Strip do
+  before do
+    @ifilter = Atig::IFilter::Strip.new %w(*tw* _)
+  end
+
+  it "should strip *tw*" do
+    @ifilter.call(status("hoge *tw*")).should be_text("hoge")
+  end
+
+  it "should strip _" do
+    @ifilter.call(status("hoge _")).should be_text("hoge")
+  end
+
+  it "should strip white-space" do
+    @ifilter.call(status("  hoge  ")).should be_text("hoge")
+  end
+end

data/spec/ifilter/tid_spec.rb

@@ -0,0 +1,29 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require File.expand_path( '../../spec_helper', __FILE__ )
+require 'atig/ifilter/xid'
+require 'atig/twitter_struct'
+
+describe Atig::IFilter::Tid, "when disable tid" do
+  def filtered(text)
+    ifilter = Atig::IFilter::Tid.new(OpenStruct.new(:log=>mock('log'),
+                                                    :opts=>OpenStruct.new))
+    ifilter.call status(text,'tid'=>1)
+  end
+
+  it "should through text" do
+    filtered("hello").should be_text("hello")
+  end
+end
+
+describe Atig::IFilter::Tid, "when enable tid" do
+  def filtered(text)
+    ifilter = Atig::IFilter::Tid.new(OpenStruct.new(:log=>mock('log'),
+                                                    :opts=>OpenStruct.new(:tid=>true)))
+    ifilter.call status(text,'tid'=>1)
+  end
+
+  it "should append tid" do
+    filtered("hello").should be_text("hello \x0310[1]\x0F")
+  end
+end