Added Twitter::Mention#source

sferik · Feb 29, 2012 · 6829994 · 6829994
1 parent f50ff25
commit 6829994
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/lib/twitter/mention.rb b/lib/twitter/mention.rb
@@ -14,6 +14,13 @@ def sources
       end
     end
 
+    # The user who mentioned a user
+    #
+    # @return [Twitter::User]
+    def source
+      @source = sources.first
+    end
+
     # A collection of statuses that mention a user
     #
     # @return [Array<Twitter::Status>]

diff --git a/spec/twitter/mention_spec.rb b/spec/twitter/mention_spec.rb
@@ -14,6 +14,17 @@
     end
   end
 
+  describe "#source" do
+    it "should return the user who mentioned a user" do
+      source = Twitter::Mention.new('sources' => [{}]).source
+      source.should be_a Twitter::User
+    end
+    it "should be nil when not set" do
+      source = Twitter::Mention.new.source
+      source.should be_nil
+    end
+  end
+
   describe "#target_objects" do
     it "should return a collection of statuses that mention a user" do
       targets = Twitter::Mention.new('target_objects' => [{}]).target_objects