Update models.history.builder: fix History.add_activity_task_schedule_failed()

Author: ggreg

swf/models/history/builder.py

@@ -169,17 +169,15 @@ def add_decision_task_timed_out(self,
     def add_activity_task_schedule_failed(self,
                                           activity_id,
                                           decision_id,
+                                          activity_type,
                                           cause):
         self.events.append(EventFactory({
             u'eventId': self.next_id,
             u'eventTimestamp': 1386947268.527,
             u'eventType': u'ScheduleActivityTaskFailed',
             u'scheduleActivityTaskFailedEventAttributes': {
                 u'activityId': activity_id,
-                u'activityType': {
-                    u'name': u'analysis.make_bad_link_counter_file',
-                    u'version': u'2.0'
-                },
+                u'activityType': activity_type.copy(),
                 u'cause': cause,
                 u'decisionTaskCompletedEventId': decision_id,
             }
@@ -302,6 +300,8 @@ def add_activity_task(self,
                           result=None,
                           reason=DEFAULT_REASON,
                           details=DEFAULT_DETAILS,
+                          activity_type=None,
+                          cause=None,
                           timeout_type='START_TO_CLOSE'):
         self.add_activity_task_scheduled(
             activity,
@@ -312,6 +312,14 @@ def add_activity_task(self,
         if last_state == 'scheduled':
             return self
 
+        if last_state == 'schedule_failed':
+            self.add_activity_task_schedule_failed(
+                activity_id,
+                decision_id,
+                activity_type,
+                cause)
+            return self
+
         scheduled_id = self.last_id
         self.add_activity_task_started(scheduled=scheduled_id)
         if last_state == 'started':