GH-85: Add support for "include_replies=only" flag in UserCommentsReq…

…uest
henrikfroehling · Aug 5, 2020 · f01cd1f · f01cd1f
1 parent 588dc2c
commit f01cd1f
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 4 deletions.
diff --git a/Source/Lib/Trakt.NET/Modules/TraktUsersModule.cs b/Source/Lib/Trakt.NET/Modules/TraktUsersModule.cs
@@ -373,7 +373,7 @@ public Task<TraktListResponse<ITraktCollectionShow>> GetCollectionShowsAsync(str
         public Task<TraktPagedResponse<ITraktUserComment>> GetCommentsAsync(string usernameOrSlug,
                                                                             TraktCommentType commentType = null,
                                                                             TraktObjectType objectType = null,
-                                                                            bool? includeReplies = null,
+                                                                            TraktIncludeReplies? includeReplies = null,
                                                                             TraktExtendedInfo extendedInfo = null,
                                                                             TraktPagedParameters pagedParameters = null,
                                                                             CancellationToken cancellationToken = default)

diff --git a/Source/Lib/Trakt.NET/Requests/Parameters/TraktIncludeReplies.cs b/Source/Lib/Trakt.NET/Requests/Parameters/TraktIncludeReplies.cs
@@ -0,0 +1,9 @@
+namespace TraktNet.Requests.Parameters
+{
+    public enum TraktIncludeReplies
+    {
+        True,
+        False,
+        Only
+    }
+}
diff --git a/Source/Lib/Trakt.NET/Requests/Users/OAuth/UserCommentsRequest.cs b/Source/Lib/Trakt.NET/Requests/Users/OAuth/UserCommentsRequest.cs
@@ -3,6 +3,7 @@
     using Enums;
     using Extensions;
     using Objects.Get.Users;
+    using Parameters;
     using System;
     using System.Collections.Generic;
 
@@ -14,7 +15,7 @@ internal sealed class UserCommentsRequest : AUsersPagedGetRequest<ITraktUserComm
 
         internal TraktObjectType ObjectType { get; set; }
 
-        internal bool? IncludeReplies { get; set; }
+        internal TraktIncludeReplies? IncludeReplies { get; set; }
 
         public override string UriTemplate => "users/{username}/comments{/comment_type}{/object_type}{?include_replies,extended,page,limit}";
 
@@ -31,7 +32,7 @@ public override IDictionary<string, object> GetUriPathParameters()
                 uriParams.Add("object_type", ObjectType.UriName);
 
             if (IncludeReplies.HasValue)
-                uriParams.Add("include_replies", IncludeReplies.ToString().ToLower());
+                uriParams.Add("include_replies", IncludeReplies.Value.ToString().ToLower());
 
             return uriParams;
         }

diff --git a/Source/Tests/Trakt.NET.Requests.Tests/Users/OAuth/UserCommentsRequest_Tests.cs b/Source/Tests/Trakt.NET.Requests.Tests/Users/OAuth/UserCommentsRequest_Tests.cs
@@ -68,7 +68,7 @@ public class UserCommentsRequest_TestData : IEnumerable<object[]>
             private static readonly TraktExtendedInfo _extendedInfo = new TraktExtendedInfo { Full = true };
             private const int _page = 4;
             private const int _limit = 20;
-            private const bool _includeReplies = true;
+            private const TraktIncludeReplies _includeReplies = TraktIncludeReplies.Only;
 
             private static readonly UserCommentsRequest _request1 = new UserCommentsRequest
             {