lib/hackerone/client/activity.rb in hackerone-client-0.5.2 vs lib/hackerone/client/activity.rb in hackerone-client-0.6.0

@@ -51,15 +51,20 @@
 
       class ReferenceIdAdded < Activity
         delegate :reference, :reference_url, to: :attributes
       end
 
+      class CommentAdded < Activity
+        delegate :message, :internal, to: :attributes
+      end
+
       ACTIVITY_TYPE_CLASS_MAPPING = {
         'activity-bounty-awarded' => BountyAwarded,
         'activity-swag-awarded' => SwagAwarded,
         'activity-user-assigned-to-bug' => UserAssignedToBug,
         'activity-bug-triaged' => BugTriaged,
-        'activity-reference-id-added' => ReferenceIdAdded
+        'activity-reference-id-added' => ReferenceIdAdded,
+        'activity-comment' => CommentAdded
       }.freeze
 
       def self.build(activity_data)
         activity_type_class = ACTIVITY_TYPE_CLASS_MAPPING.fetch \
           activity_data[:type], Activity