From 29b3da2aa1f1b4a8514ffea6dc57dda8d8a98170 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Fri, 10 Jul 2020 16:10:17 +0800
Subject: Add unit tests.

---
 Timeline.Tests/IntegratedTests/TimelineTest.cs |  2 ++
 Timeline.Tests/Services/TimelineServiceTest.cs | 39 ++++++++++++++++++++++++++
 2 files changed, 41 insertions(+)

(limited to 'Timeline.Tests')

diff --git a/Timeline.Tests/IntegratedTests/TimelineTest.cs b/Timeline.Tests/IntegratedTests/TimelineTest.cs
index 4f21d8d1..64619864 100644
--- a/Timeline.Tests/IntegratedTests/TimelineTest.cs
+++ b/Timeline.Tests/IntegratedTests/TimelineTest.cs
@@ -942,6 +942,7 @@ namespace Timeline.Tests.IntegratedTests
                     body.Should().NotBeNull();
                     body.Content.Should().BeEquivalentTo(TimelineHelper.TextPostContent(mockContent));
                     body.Author.Should().BeEquivalentTo(UserInfos[1]);
+                    body.Deleted.Should().BeFalse();
                     createRes = body;
                 }
                 {
@@ -963,6 +964,7 @@ namespace Timeline.Tests.IntegratedTests
                     body.Content.Should().BeEquivalentTo(TimelineHelper.TextPostContent(mockContent2));
                     body.Author.Should().BeEquivalentTo(UserInfos[1]);
                     body.Time.Should().BeCloseTo(mockTime2, 1000);
+                    body.Deleted.Should().BeFalse();
                     createRes2 = body;
                 }
                 {
diff --git a/Timeline.Tests/Services/TimelineServiceTest.cs b/Timeline.Tests/Services/TimelineServiceTest.cs
index 7e7242a2..123c2b7f 100644
--- a/Timeline.Tests/Services/TimelineServiceTest.cs
+++ b/Timeline.Tests/Services/TimelineServiceTest.cs
@@ -118,5 +118,44 @@ namespace Timeline.Tests.Services
             posts.Should().HaveCount(3)
                 .And.Subject.Select(p => (p.Content as TextTimelinePostContent).Text).Should().Equal(postContentList.Skip(1));
         }
+
+        [Theory]
+        [InlineData("@user")]
+        [InlineData("tl")]
+        public async Task GetPosts_IncludeDeleted(string timelineName)
+        {
+            var userId = await _userService.GetUserIdByUsername("user");
+
+            var _ = TimelineHelper.ExtractTimelineName(timelineName, out var isPersonal);
+            if (!isPersonal)
+                await _timelineService.CreateTimeline(timelineName, userId);
+
+            var postContentList = new string[] { "a", "b", "c", "d" };
+
+            foreach (var content in postContentList)
+            {
+                await _timelineService.CreateTextPost(timelineName, userId, content, null);
+            }
+
+            var posts = await _timelineService.GetPosts(timelineName);
+            posts.Should().HaveCount(4);
+            posts.Select(p => p.Deleted).Should().Equal(Enumerable.Repeat(false, posts.Count));
+            posts.Select(p => ((TextTimelinePostContent)p.Content).Text).Should().Equal(postContentList);
+
+            foreach (var id in new long[] { posts[0].Id, posts[2].Id })
+            {
+                await _timelineService.DeletePost(timelineName, id);
+            }
+
+            posts = await _timelineService.GetPosts(timelineName);
+            posts.Should().HaveCount(2);
+            posts.Select(p => p.Deleted).Should().Equal(Enumerable.Repeat(false, posts.Count));
+            posts.Select(p => ((TextTimelinePostContent)p.Content).Text).Should().Equal(new string[] { "b", "d" });
+
+            posts = await _timelineService.GetPosts(timelineName, includeDeleted: true);
+            posts.Should().HaveCount(4);
+            posts.Select(p => p.Deleted).Should().Equal(new bool[] { true, false, true, false });
+            posts.Where(p => !p.Deleted).Select(p => ((TextTimelinePostContent)p.Content).Text).Should().Equal(new string[] { "b", "d" });
+        }
     }
 }
-- 
cgit v1.2.3