adding tests for new pipeline
This commit is contained in:
parent
57a08b67b1
commit
8a99f5ecce
|
@ -17,6 +17,7 @@
|
|||
</ItemGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<Folder Include="Processors\TweetsCleanUp\" />
|
||||
<Folder Include="Tools\" />
|
||||
</ItemGroup>
|
||||
|
||||
|
|
|
@ -13,7 +13,7 @@ using BirdsiteLive.Twitter.Models;
|
|||
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
||||
using Moq;
|
||||
|
||||
namespace BirdsiteLive.Pipeline.Tests.Processors
|
||||
namespace BirdsiteLive.Pipeline.Tests.Processors.Federation
|
||||
{
|
||||
[TestClass]
|
||||
public class RefreshTwitterUserStatusProcessorTests
|
||||
|
@ -60,7 +60,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
var result = await processor.ProcessAsync(users.ToArray(), CancellationToken.None);
|
||||
|
||||
#region Validations
|
||||
Assert.AreEqual(2 , result.Length);
|
||||
Assert.AreEqual(2, result.Length);
|
||||
Assert.IsTrue(result.Any(x => x.User.Id == userId1));
|
||||
Assert.IsTrue(result.Any(x => x.User.Id == userId2));
|
||||
|
||||
|
@ -288,7 +288,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
twitterUserServiceMock
|
||||
.Setup(x => x.GetUser(It.Is<string>(y => y == acct2)))
|
||||
.Throws(new Exception());
|
||||
|
||||
|
||||
var twitterUserDalMock = new Mock<ITwitterUserDal>(MockBehavior.Strict);
|
||||
twitterUserDalMock
|
||||
.Setup(x => x.GetTwitterUserAsync(It.Is<string>(y => y == acct2)))
|
||||
|
@ -360,7 +360,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
twitterUserServiceMock
|
||||
.Setup(x => x.GetUser(It.Is<string>(y => y == acct2)))
|
||||
.Returns((TwitterUser)null);
|
||||
|
||||
|
||||
var twitterUserDalMock = new Mock<ITwitterUserDal>(MockBehavior.Strict);
|
||||
twitterUserDalMock
|
||||
.Setup(x => x.GetTwitterUserAsync(It.Is<string>(y => y == acct2)))
|
||||
|
@ -432,7 +432,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
twitterUserServiceMock
|
||||
.Setup(x => x.GetUser(It.Is<string>(y => y == acct2)))
|
||||
.Returns((TwitterUser)null);
|
||||
|
||||
|
||||
var twitterUserDalMock = new Mock<ITwitterUserDal>(MockBehavior.Strict);
|
||||
twitterUserDalMock
|
||||
.Setup(x => x.GetTwitterUserAsync(It.Is<string>(y => y == acct2)))
|
||||
|
@ -506,7 +506,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
{
|
||||
Protected = true
|
||||
});
|
||||
|
||||
|
||||
var twitterUserDalMock = new Mock<ITwitterUserDal>(MockBehavior.Strict);
|
||||
twitterUserDalMock
|
||||
.Setup(x => x.GetTwitterUserAsync(It.Is<string>(y => y == acct2)))
|
||||
|
@ -581,7 +581,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
{
|
||||
Protected = true
|
||||
});
|
||||
|
||||
|
||||
var twitterUserDalMock = new Mock<ITwitterUserDal>(MockBehavior.Strict);
|
||||
twitterUserDalMock
|
||||
.Setup(x => x.GetTwitterUserAsync(It.Is<string>(y => y == acct2)))
|
||||
|
@ -609,14 +609,14 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
removeTwitterAccountActionMock.VerifyAll();
|
||||
#endregion
|
||||
}
|
||||
|
||||
|
||||
[TestMethod]
|
||||
public async Task ProcessAsync_Error_NotInit_Test()
|
||||
{
|
||||
#region Stubs
|
||||
var userId1 = 1;
|
||||
var acct1 = "user1";
|
||||
|
||||
|
||||
var users = new List<SyncTwitterUser>
|
||||
{
|
||||
new SyncTwitterUser
|
||||
|
@ -640,7 +640,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
twitterUserServiceMock
|
||||
.Setup(x => x.GetUser(It.Is<string>(y => y == acct1)))
|
||||
.Returns((TwitterUser)null);
|
||||
|
||||
|
||||
var twitterUserDalMock = new Mock<ITwitterUserDal>(MockBehavior.Strict);
|
||||
twitterUserDalMock
|
||||
.Setup(x => x.GetTwitterUserAsync(It.Is<string>(y => y == acct1)))
|
||||
|
@ -708,7 +708,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
twitterUserServiceMock
|
||||
.Setup(x => x.GetUser(It.Is<string>(y => y == acct2)))
|
||||
.Throws(new RateLimitExceededException());
|
||||
|
||||
|
||||
var twitterUserDalMock = new Mock<ITwitterUserDal>(MockBehavior.Strict);
|
||||
twitterUserDalMock
|
||||
.Setup(x => x.GetTwitterUserAsync(It.Is<string>(y => y == acct2)))
|
|
@ -9,7 +9,7 @@ using BirdsiteLive.Pipeline.Processors.Federation;
|
|||
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
||||
using Moq;
|
||||
|
||||
namespace BirdsiteLive.Pipeline.Tests.Processors
|
||||
namespace BirdsiteLive.Pipeline.Tests.Processors.Federation
|
||||
{
|
||||
[TestClass]
|
||||
public class RetrieveFollowersProcessorTests
|
|
@ -12,7 +12,7 @@ using Microsoft.Extensions.Logging;
|
|||
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
||||
using Moq;
|
||||
|
||||
namespace BirdsiteLive.Pipeline.Tests.Processors
|
||||
namespace BirdsiteLive.Pipeline.Tests.Processors.Federation
|
||||
{
|
||||
[TestClass]
|
||||
public class RetrieveTwitterUsersProcessorTests
|
||||
|
@ -43,7 +43,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
It.Is<int>(y => y == maxUsers),
|
||||
It.Is<bool>(y => y == false)))
|
||||
.ReturnsAsync(users);
|
||||
|
||||
|
||||
var loggerMock = new Mock<ILogger<RetrieveTwitterUsersProcessor>>();
|
||||
#endregion
|
||||
|
||||
|
@ -100,7 +100,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
var t = processor.GetTwitterUsersAsync(buffer, CancellationToken.None);
|
||||
|
||||
await Task.WhenAny(t, Task.Delay(300));
|
||||
|
||||
|
||||
#region Validations
|
||||
maxUsersNumberProviderMock.VerifyAll();
|
||||
twitterUserDalMock.VerifyAll();
|
||||
|
@ -139,7 +139,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
.ReturnsAsync(new SyncTwitterUser[0])
|
||||
.ReturnsAsync(new SyncTwitterUser[0])
|
||||
.ReturnsAsync(new SyncTwitterUser[0]);
|
||||
|
||||
|
||||
var loggerMock = new Mock<ILogger<RetrieveTwitterUsersProcessor>>();
|
||||
#endregion
|
||||
|
||||
|
@ -190,7 +190,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
|
||||
var processor = new RetrieveTwitterUsersProcessor(twitterUserDalMock.Object, maxUsersNumberProviderMock.Object, loggerMock.Object);
|
||||
processor.WaitFactor = 1;
|
||||
var t =processor.GetTwitterUsersAsync(buffer, CancellationToken.None);
|
||||
var t = processor.GetTwitterUsersAsync(buffer, CancellationToken.None);
|
||||
|
||||
await Task.WhenAny(t, Task.Delay(50));
|
||||
|
||||
|
@ -200,7 +200,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
Assert.AreEqual(0, buffer.Count);
|
||||
#endregion
|
||||
}
|
||||
|
||||
|
||||
[TestMethod]
|
||||
public async Task GetTwitterUsersAsync_Exception_Test()
|
||||
{
|
||||
|
@ -258,7 +258,7 @@ namespace BirdsiteLive.Pipeline.Tests.Processors
|
|||
.ReturnsAsync(maxUsers);
|
||||
|
||||
var twitterUserDalMock = new Mock<ITwitterUserDal>(MockBehavior.Strict);
|
||||
|
||||
|
||||
var loggerMock = new Mock<ILogger<RetrieveTwitterUsersProcessor>>();
|
||||
#endregion
|
||||
|
|
@ -0,0 +1,160 @@
|
|||
using System;
|
||||
using System.Net;
|
||||
using System.Net.Http;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using BirdsiteLive.DAL.Models;
|
||||
using BirdsiteLive.Domain;
|
||||
using BirdsiteLive.Pipeline.Models;
|
||||
using BirdsiteLive.Pipeline.Processors.Federation;
|
||||
using BirdsiteLive.Pipeline.Processors.TweetsCleanUp;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
||||
using Moq;
|
||||
|
||||
namespace BirdsiteLive.Pipeline.Tests.Processors.TweetsCleanUp
|
||||
{
|
||||
[TestClass]
|
||||
public class DeleteTweetsProcessorTests
|
||||
{
|
||||
[TestMethod]
|
||||
public async Task Process_DeleteTweet_Test()
|
||||
{
|
||||
#region Stubs
|
||||
var tweetId = 42;
|
||||
var tweet = new TweetToDelete
|
||||
{
|
||||
Tweet = new SyncTweet
|
||||
{
|
||||
Id = tweetId
|
||||
}
|
||||
};
|
||||
#endregion
|
||||
|
||||
#region Mocks
|
||||
var serviceMock = new Mock<IActivityPubService>(MockBehavior.Strict);
|
||||
serviceMock
|
||||
.Setup(x => x.DeleteNoteAsync(It.Is<SyncTweet>(y => y.Id == tweetId)))
|
||||
.Returns(Task.CompletedTask);
|
||||
|
||||
var loggerMock = new Mock<ILogger<DeleteTweetsProcessor>>();
|
||||
#endregion
|
||||
|
||||
var processor = new DeleteTweetsProcessor(serviceMock.Object, loggerMock.Object);
|
||||
var result = await processor.ProcessAsync(tweet, CancellationToken.None);
|
||||
|
||||
#region Validations
|
||||
serviceMock.VerifyAll();
|
||||
loggerMock.VerifyAll();
|
||||
|
||||
Assert.IsNotNull(result);
|
||||
Assert.IsTrue(result.DeleteSuccessful);
|
||||
#endregion
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public async Task Process_DeleteTweet_NotFound_Test()
|
||||
{
|
||||
#region Stubs
|
||||
var tweetId = 42;
|
||||
var tweet = new TweetToDelete
|
||||
{
|
||||
Tweet = new SyncTweet
|
||||
{
|
||||
Id = tweetId
|
||||
}
|
||||
};
|
||||
#endregion
|
||||
|
||||
#region Mocks
|
||||
var serviceMock = new Mock<IActivityPubService>(MockBehavior.Strict);
|
||||
serviceMock
|
||||
.Setup(x => x.DeleteNoteAsync(It.Is<SyncTweet>(y => y.Id == tweetId)))
|
||||
.Throws(new HttpRequestException("not found", null, HttpStatusCode.NotFound));
|
||||
|
||||
var loggerMock = new Mock<ILogger<DeleteTweetsProcessor>>();
|
||||
#endregion
|
||||
|
||||
var processor = new DeleteTweetsProcessor(serviceMock.Object, loggerMock.Object);
|
||||
var result = await processor.ProcessAsync(tweet, CancellationToken.None);
|
||||
|
||||
#region Validations
|
||||
serviceMock.VerifyAll();
|
||||
loggerMock.VerifyAll();
|
||||
|
||||
Assert.IsNotNull(result);
|
||||
Assert.IsTrue(result.DeleteSuccessful);
|
||||
#endregion
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public async Task Process_DeleteTweet_HttpError_Test()
|
||||
{
|
||||
#region Stubs
|
||||
var tweetId = 42;
|
||||
var tweet = new TweetToDelete
|
||||
{
|
||||
Tweet = new SyncTweet
|
||||
{
|
||||
Id = tweetId
|
||||
}
|
||||
};
|
||||
#endregion
|
||||
|
||||
#region Mocks
|
||||
var serviceMock = new Mock<IActivityPubService>(MockBehavior.Strict);
|
||||
serviceMock
|
||||
.Setup(x => x.DeleteNoteAsync(It.Is<SyncTweet>(y => y.Id == tweetId)))
|
||||
.Throws(new HttpRequestException("bad gateway", null, HttpStatusCode.BadGateway));
|
||||
|
||||
var loggerMock = new Mock<ILogger<DeleteTweetsProcessor>>();
|
||||
#endregion
|
||||
|
||||
var processor = new DeleteTweetsProcessor(serviceMock.Object, loggerMock.Object);
|
||||
var result = await processor.ProcessAsync(tweet, CancellationToken.None);
|
||||
|
||||
#region Validations
|
||||
serviceMock.VerifyAll();
|
||||
loggerMock.VerifyAll();
|
||||
|
||||
Assert.IsNotNull(result);
|
||||
Assert.IsFalse(result.DeleteSuccessful);
|
||||
#endregion
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public async Task Process_DeleteTweet_GenericException_Test()
|
||||
{
|
||||
#region Stubs
|
||||
var tweetId = 42;
|
||||
var tweet = new TweetToDelete
|
||||
{
|
||||
Tweet = new SyncTweet
|
||||
{
|
||||
Id = tweetId
|
||||
}
|
||||
};
|
||||
#endregion
|
||||
|
||||
#region Mocks
|
||||
var serviceMock = new Mock<IActivityPubService>(MockBehavior.Strict);
|
||||
serviceMock
|
||||
.Setup(x => x.DeleteNoteAsync(It.Is<SyncTweet>(y => y.Id == tweetId)))
|
||||
.Throws(new Exception());
|
||||
|
||||
var loggerMock = new Mock<ILogger<DeleteTweetsProcessor>>();
|
||||
#endregion
|
||||
|
||||
var processor = new DeleteTweetsProcessor(serviceMock.Object, loggerMock.Object);
|
||||
var result = await processor.ProcessAsync(tweet, CancellationToken.None);
|
||||
|
||||
#region Validations
|
||||
serviceMock.VerifyAll();
|
||||
loggerMock.VerifyAll();
|
||||
|
||||
Assert.IsNotNull(result);
|
||||
Assert.IsFalse(result.DeleteSuccessful);
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
}
|
|
@ -0,0 +1,76 @@
|
|||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using System.Threading.Tasks.Dataflow;
|
||||
using System.Xml.Linq;
|
||||
using BirdsiteLive.Common.Settings;
|
||||
using BirdsiteLive.DAL.Contracts;
|
||||
using BirdsiteLive.DAL.Models;
|
||||
using BirdsiteLive.Pipeline.Models;
|
||||
using BirdsiteLive.Pipeline.Processors.TweetsCleanUp;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
||||
using Moq;
|
||||
using Org.BouncyCastle.Crypto;
|
||||
|
||||
namespace BirdsiteLive.Pipeline.Tests.Processors.TweetsCleanUp
|
||||
{
|
||||
[TestClass]
|
||||
public class RetrieveTweetsToDeleteProcessorTests
|
||||
{
|
||||
[TestMethod]
|
||||
public async Task Process_Test()
|
||||
{
|
||||
#region Stubs
|
||||
var settings = new InstanceSettings
|
||||
{
|
||||
|
||||
};
|
||||
var bufferBlock = new BufferBlock<TweetToDelete>();
|
||||
|
||||
var tweetId1 = 42;
|
||||
var tweetId2 = 43;
|
||||
|
||||
var tweet1 = new SyncTweet
|
||||
{
|
||||
Id = tweetId1
|
||||
};
|
||||
var tweet2 = new SyncTweet
|
||||
{
|
||||
Id = tweetId2
|
||||
};
|
||||
var batch = new List<SyncTweet>
|
||||
{
|
||||
tweet1,
|
||||
tweet2
|
||||
};
|
||||
#endregion
|
||||
|
||||
#region Mocks
|
||||
var dalMock = new Mock<ISyncTweetsPostgresDal>(MockBehavior.Strict);
|
||||
dalMock
|
||||
.Setup(x => x.GetTweetsOlderThanAsync(It.IsAny<DateTime>(), It.Is<long>(y => y == -1), It.Is<int>(y => y == 100)))
|
||||
.ReturnsAsync(batch);
|
||||
#endregion
|
||||
|
||||
var processor = new RetrieveTweetsToDeleteProcessor(dalMock.Object, settings);
|
||||
processor.GetTweetsAsync(bufferBlock, CancellationToken.None);
|
||||
|
||||
await Task.Delay(TimeSpan.FromSeconds(1));
|
||||
|
||||
#region Validations
|
||||
dalMock.VerifyAll();
|
||||
|
||||
Assert.AreEqual(batch.Count, bufferBlock.Count);
|
||||
|
||||
bufferBlock.TryReceiveAll(out var all);
|
||||
foreach (var tweet in all)
|
||||
{
|
||||
Assert.IsTrue(batch.Any(x => x.Id == tweet.Tweet.Id));
|
||||
}
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
}
|
|
@ -0,0 +1,6 @@
|
|||
namespace BirdsiteLive.Pipeline.Tests.Processors.TweetsCleanUp;
|
||||
|
||||
public class SaveDeletedTweetStatusProcessorTests
|
||||
{
|
||||
|
||||
}
|
|
@ -0,0 +1,6 @@
|
|||
namespace BirdsiteLive.Pipeline.Tests;
|
||||
|
||||
public class TweetCleanUpPipelineTests
|
||||
{
|
||||
|
||||
}
|
Loading…
Reference in New Issue