Added single instance policy

2026-04-25 22:36:59 -04:00 · 2013-11-25 18:46:12 -08:00
parent 266d1a43d9
commit c219be8c8d
14 changed files with 254 additions and 51 deletions
@@ -0,0 +1,91 @@
+using System.Collections.Generic;
+using Moq;
+using NUnit.Framework;
+using NzbDrone.Common.Model;
+using NzbDrone.Common.Processes;
+using NzbDrone.Host;
+using NzbDrone.Test.Common;
+
+namespace NzbDrone.App.Test
+{
+    [TestFixture]
+    public class NzbDroneProcessServiceFixture : TestBase<SingleInstancePolicy>
+    {
+        private const int CURRENT_PROCESS_ID = 5;
+
+        [SetUp]
+        public void Setup()
+        {
+            Mocker.GetMock<IProcessProvider>().Setup(c => c.GetCurrentProcess())
+                .Returns(new ProcessInfo() { Id = CURRENT_PROCESS_ID });
+        }
+
+        [Test]
+        public void should_continue_if_only_instance()
+        {
+            Mocker.GetMock<IProcessProvider>()
+                .Setup(c => c.FindProcessByName(ProcessProvider.NZB_DRONE_CONSOLE_PROCESS_NAME))
+                .Returns(new List<ProcessInfo>());
+
+            Mocker.GetMock<IProcessProvider>().Setup(c => c.FindProcessByName(ProcessProvider.NZB_DRONE_PROCESS_NAME))
+                .Returns(new List<ProcessInfo>
+                {
+                    new ProcessInfo{Id = CURRENT_PROCESS_ID}
+                });
+
+
+            Subject.EnforceSingleInstance();
+
+
+            Mocker.GetMock<IBrowserService>().Verify(c => c.LaunchWebUI(), Times.Never());
+
+        }
+
+        [Test]
+        public void should_enforce_if_another_console_is_running()
+        {
+            Mocker.GetMock<IProcessProvider>()
+                .Setup(c => c.FindProcessByName(ProcessProvider.NZB_DRONE_CONSOLE_PROCESS_NAME))
+                    .Returns(new List<ProcessInfo>
+                {
+                    new ProcessInfo{Id = 10}
+                });
+
+            Mocker.GetMock<IProcessProvider>().Setup(c => c.FindProcessByName(ProcessProvider.NZB_DRONE_PROCESS_NAME))
+                .Returns(new List<ProcessInfo>
+                {
+                    new ProcessInfo{Id = CURRENT_PROCESS_ID}
+                });
+
+
+
+            Assert.Throws<TerminateApplicationException>(() => Subject.EnforceSingleInstance());
+            Mocker.GetMock<IBrowserService>().Verify(c => c.LaunchWebUI(), Times.Once());
+            ExceptionVerification.ExpectedWarns(1);
+        }
+
+        [Test]
+        public void should_return_false_if_another_gui_is_running()
+        {
+            Mocker.GetMock<IProcessProvider>()
+                .Setup(c => c.FindProcessByName(ProcessProvider.NZB_DRONE_CONSOLE_PROCESS_NAME))
+                .Returns(new List<ProcessInfo>
+                {
+                      new ProcessInfo{Id = CURRENT_PROCESS_ID}
+                
+                });
+
+            Mocker.GetMock<IProcessProvider>().Setup(c => c.FindProcessByName(ProcessProvider.NZB_DRONE_PROCESS_NAME))
+                .Returns(new List<ProcessInfo>
+                {
+                      new ProcessInfo{Id = 10}
+                });
+
+
+
+            Assert.Throws<TerminateApplicationException>(() => Subject.EnforceSingleInstance());
+            Mocker.GetMock<IBrowserService>().Verify(c => c.LaunchWebUI(), Times.Once());
+            ExceptionVerification.ExpectedWarns(1);
+        }
+    }
+}