Merge pull request huginn#208 from knu/tweet_id

Include tweet_id in payload for each successful tweet.
th3byrdm4n · Apr 13, 2014 · e732234 · e732234
2 parents 0e7c5c5 + ef81624
commit e732234
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/app/models/agents/twitter_publish_agent.rb b/app/models/agents/twitter_publish_agent.rb
@@ -43,10 +43,11 @@ def receive(incoming_events)
       incoming_events.each do |event|
         tweet_text = Utils.value_at(event.payload, options['message_path'])
         begin
-          publish_tweet tweet_text
+          tweet = publish_tweet tweet_text
           create_event :payload => {
             'success' => true,
             'published_tweet' => tweet_text,
+            'tweet_id' => tweet.id,
             'agent_id' => event.agent_id,
             'event_id' => event.id
           }

diff --git a/spec/models/agents/twitter_publish_agent_spec.rb b/spec/models/agents/twitter_publish_agent_spec.rb
@@ -22,7 +22,10 @@
     @event.save!
 
     @sent_messages = []
-    stub.any_instance_of(Agents::TwitterPublishAgent).publish_tweet { |message| @sent_messages << message}
+    stub.any_instance_of(Agents::TwitterPublishAgent).publish_tweet { |message|
+      @sent_messages << message
+      OpenStruct.new(:id => 454209588376502272)
+    }
   end
 
   describe '#receive' do
@@ -53,4 +56,4 @@
       @checker.reload.should_not be_working # More time has passed than the expected receive period without any new events
     end
   end
-end
+end