[MainViewModel] Add mod enabled/disabled checkbox.

Aleksbgbg · Apr 12, 2019 · 51f12b5 · 51f12b5
1 parent 19f5be3
commit 51f12b5
Show file tree

Hide file tree

Showing 3 changed files with 49 additions and 3 deletions.
diff --git a/Instant Training UI.Tests/ViewModels/MainViewModelTests.cs b/Instant Training UI.Tests/ViewModels/MainViewModelTests.cs
@@ -1,5 +1,6 @@
 namespace Instant.Training.UI.Tests.ViewModels
 {
+    using Instant.Training.UI.Services.Interfaces;
     using Instant.Training.UI.ViewModels;
     using Instant.Training.UI.ViewModels.Interfaces;
 
@@ -9,21 +10,41 @@
 
     public class MainViewModelTests
     {
+        private readonly Mock<IRconService> _rconServiceMock;
+
         private readonly Mock<IArenaDisplayViewModel> _arenaDisplayViewModelMock;
 
         private readonly MainViewModel _mainViewModel;
 
         public MainViewModelTests()
         {
+            _rconServiceMock = new Mock<IRconService>();
+
             _arenaDisplayViewModelMock = new Mock<IArenaDisplayViewModel>();
 
-            _mainViewModel = new MainViewModel(_arenaDisplayViewModelMock.Object);
+            _mainViewModel = new MainViewModel(_arenaDisplayViewModelMock.Object, _rconServiceMock.Object);
         }
 
         [Fact]
         public void TestArenaDisplayViewModelAssigned()
         {
             Assert.Equal(_arenaDisplayViewModelMock.Object, _mainViewModel.ArenaDisplayViewModel);
         }
+
+        [Fact]
+        public void TestInstantTrainingEnabledByDefault()
+        {
+            Assert.True(_mainViewModel.EnableInstantTraining);
+        }
+
+        [Fact]
+        public void TestInstantTrainingToggleCallsRconService()
+        {
+            const bool modEnabled = false;
+
+            _mainViewModel.EnableInstantTraining = modEnabled;
+
+            _rconServiceMock.Verify(rconService => rconService.ToggleModEnabled(modEnabled));
+        }
     }
 }
diff --git a/Instant Training UI/ViewModels/MainViewModel.cs b/Instant Training UI/ViewModels/MainViewModel.cs
@@ -1,14 +1,34 @@
 namespace Instant.Training.UI.ViewModels
 {
+    using Instant.Training.UI.Services.Interfaces;
     using Instant.Training.UI.ViewModels.Interfaces;
 
     public class MainViewModel : ViewModelBase, IMainViewModel
     {
-        public MainViewModel(IArenaDisplayViewModel arenaDisplayViewModel)
+        private readonly IRconService _rconService;
+
+        public MainViewModel(IArenaDisplayViewModel arenaDisplayViewModel, IRconService rconService)
         {
             ArenaDisplayViewModel = arenaDisplayViewModel;
+            _rconService = rconService;
         }
 
         public IArenaDisplayViewModel ArenaDisplayViewModel { get; }
+
+        private bool _enableInstantTraining = true;
+        public bool EnableInstantTraining
+        {
+            get => _enableInstantTraining;
+
+            set
+            {
+                if (_enableInstantTraining == value) return;
+
+                _enableInstantTraining = value;
+                NotifyOfPropertyChange(nameof(EnableInstantTraining));
+
+                _rconService.ToggleModEnabled(_enableInstantTraining);
+            }
+        }
     }
 }
diff --git a/Instant Training UI/Views/MainView.xaml b/Instant Training UI/Views/MainView.xaml
@@ -7,5 +7,10 @@
              xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
              d:DataContext="{d:DesignInstance ViewModels:MainViewModel}"
              mc:Ignorable="d">
-    <ContentControl Caliburn:View.Model="{Binding ArenaDisplayViewModel, Mode=OneTime}"/>
+    <DockPanel>
+        <CheckBox Content="Enable Instant Training"
+                  DockPanel.Dock="Top"
+                  IsChecked="{Binding EnableInstantTraining}"/>
+        <ContentControl Caliburn:View.Model="{Binding ArenaDisplayViewModel, Mode=OneTime}"/>
+    </DockPanel>
 </UserControl>