SocialPub/PrivaPub.Tests/Infrastructure/MigrationTests.cs

using MongoDB.Entities;

using PrivaPub.Infrastructure.Data.Migrations;
using PrivaPub.Models.Post;
using PrivaPub.Tests.Support;

namespace PrivaPub.Tests.Infrastructure
{
	[Xunit.Collection(nameof(Exclusive))]
	[Trait("Category", "Integration")]
	public class MigrationTests
	{
		[Fact]
		public async Task Stored_remote_content_is_sanitized_and_local_content_rendered()
		{
			Assert.SkipUnless(MongoFixture.Enabled, MongoFixture.Skip);
			var token = TestContext.Current.CancellationToken;
			var remote = new Post { IsFederatedCopy = true, Text = "<p>hi<script>alert(1)</script></p>", ObjectURI = $"https://r.example/{Guid.NewGuid():N}" };
			var local = new Post { Text = "**bold** <b>raw</b>", ObjectURI = $"https://privapub.test/{Guid.NewGuid():N}" };
			await DB.Default.SaveAsync(new[] { remote, local }, token);

			await new _002_sanitize_stored_content().UpgradeAsync();

			var migratedRemote = await DB.Default.Find<Post>().OneAsync(remote.ID, token);
			var migratedLocal = await DB.Default.Find<Post>().OneAsync(local.ID, token);
			Assert.Equal("<p>hi</p>", migratedRemote.Text);
			Assert.Equal("<p>hi</p>", migratedRemote.ContentHtml);
			Assert.Equal(ContentFormat.Html, migratedRemote.ContentFormat);
			Assert.Equal("**bold** <b>raw</b>", migratedLocal.Text);
			Assert.Equal("<p><strong>bold</strong> &lt;b&gt;raw&lt;/b&gt;</p>", migratedLocal.ContentHtml);
			Assert.Equal(ContentFormat.Markdown, migratedLocal.ContentFormat);
		}
	
		[Fact]
		public async Task Direct_posts_move_into_posts_with_their_conversation()
		{
			Assert.SkipUnless(MongoFixture.Enabled, MongoFixture.Skip);
			var token = TestContext.Current.CancellationToken;
			var dm = new DmPost { GroupId = "conversation1", Text = "psst", ObjectURI = $"https://privapub.test/{Guid.NewGuid():N}", GroupUserId = "a1" };
			await DB.Default.SaveAsync(dm, token);

			await new _005_direct_posts_join_posts().UpgradeAsync();

			var post = await DB.Default.Find<Post>().OneAsync(dm.ID, token);
			Assert.Equal(PostVisibility.Direct, post.Visibility);
			Assert.Equal("conversation1", post.ConversationId);
			Assert.Equal("a1", post.AuthorAccountId);
		}
	}
}