Radarr/NzbDrone.Core/Providers/Fakes/FakeNotificationProvider.cs

63 lines
2.2 KiB
C#
Raw Normal View History

2010-10-10 19:00:07 +00:00
using System;
using System.Collections.Generic;
using NzbDrone.Core.Model.Notification;
2010-10-10 19:00:07 +00:00
namespace NzbDrone.Core.Providers.Fakes
{
2011-04-10 02:44:01 +00:00
internal class FakeNotificationProvider
2010-10-10 19:00:07 +00:00
{
2011-04-10 02:44:01 +00:00
private readonly Dictionary<Guid, BasicNotification> _basicNotifications =
new Dictionary<Guid, BasicNotification>();
2010-10-10 19:00:07 +00:00
private readonly Object _lock = new object();
2011-04-10 02:44:01 +00:00
private readonly Dictionary<Guid, ProgressNotification> _progressNotification =
new Dictionary<Guid, ProgressNotification>();
2010-10-10 19:00:07 +00:00
2011-04-10 02:44:01 +00:00
private readonly ProgressNotification fakeNotification = new ProgressNotification("Updating Series");
private readonly ProgressNotification fakeNotification2 = new ProgressNotification("Updating Series2");
2010-10-10 19:00:07 +00:00
public List<BasicNotification> BasicNotifications
{
get { return new List<BasicNotification>(_basicNotifications.Values); }
}
2010-10-12 02:49:27 +00:00
public List<ProgressNotification> GetProgressNotifications
2010-10-10 19:00:07 +00:00
{
get
{
fakeNotification.Status = ProgressNotificationStatus.InProgress;
fakeNotification.Status = ProgressNotificationStatus.InProgress;
2011-04-24 05:48:12 +00:00
fakeNotification2.CurrentMessage = DateTime.UtcNow.ToString();
fakeNotification.CurrentMessage = DateTime.Now.ToString();
2011-04-10 02:44:01 +00:00
return new List<ProgressNotification> {fakeNotification};
2010-10-10 19:00:07 +00:00
}
}
2011-04-10 02:44:01 +00:00
public void Register(ProgressNotification notification)
{
_progressNotification.Add(notification.Id, notification);
}
public void Register(BasicNotification notification)
{
_basicNotifications.Add(notification.Id, notification);
}
2010-10-10 19:00:07 +00:00
public void Dismiss(Guid notificationId)
{
lock (_lock)
{
if (_basicNotifications.ContainsKey(notificationId))
{
_basicNotifications.Remove(notificationId);
}
else if (_progressNotification.ContainsKey(notificationId))
{
_progressNotification.Remove(notificationId);
}
}
}
}
}