Fix tests

JelteF · JelteF · commit d3bba639c402 · 2017-07-28T10:07:26.000+02:00
diff --git a/stream/tests.py b/stream/tests.py
@@ -565,17 +565,25 @@ def test_do_i_follow(self):
         self.assertEqual(followings['results'][0]['feed_id'], social.id)
         self.assertEqual(followings['results'][0]['target_id'], 'user:apy')
 
-    def test_change_activity_to(self):
-        activity_data = {'actor': 1, 'verb': 'tweet', 'object': 1}
+    def test_update_activity_to_targets(self):
+        time = datetime.datetime.utcnow().isoformat()
+        foreign_id = 'user:1'
+        activity_data = {
+            'actor': 1,
+            'verb': 'tweet',
+            'object': 1,
+            'foreign_id': foreign_id,
+            'time': time,
+        }
         activity_data['to'] = ['user:1', 'user:2']
-        activity_id = self.user1.add_activity(activity_data)['id']
+        self.user1.add_activity(activity_data)
 
-        ret = self.user1.change_activity_to(activity_id, replace=['user:3', 'user:2'])
+        ret = self.user1.update_activity_to_targets(foreign_id, time, replace=['user:3', 'user:2'])
         self.assertEqual(len(ret['activity']['to']), 2)
         self.assertTrue('user:2' in ret['activity']['to'])
         self.assertTrue('user:3' in ret['activity']['to'])
 
-        ret = self.user1.change_activity_to(activity_id, add=['user:4', 'user:5'], delete=['user:3'])
+        ret = self.user1.update_activity_to_targets(foreign_id, time, add=['user:4', 'user:5'], remove=['user:3'])
         self.assertEqual(len(ret['activity']['to']), 3)
         self.assertTrue('user:2' in ret['activity']['to'])
         self.assertTrue('user:4' in ret['activity']['to'])
