TaskViewDataStoreSpy.swift

class TaskViewDataStoreSpy: TaskViewDataStore {

    var fetchTodayTaskCalled = false
    var fetchFailedTaskCalled = false
    var fetchUpcomingTaskCalled = false
    var fetchFinishedTaskCalled = false
    var finishTaskCalled = false
    
    var todayTaskCompletion: (([TaskModel]) -> Void)?
    var failedTaskCompletion: (([TaskModel]) -> Void)?
    var upcomingTaskCompletion: (([TaskModel]) -> Void)?
    var finishedTaskCompletion: (([TaskModel]) -> Void)?
    
    func fetchTodayTask(completion: @escaping ([TaskModel]) -> Void) {
        fetchTodayTaskCalled = true
        todayTaskCompletion = completion
    }
    
    func fetchFailedTask(completion: @escaping ([TaskModel]) -> Void) {
        fetchFailedTaskCalled = true
        failedTaskCompletion = completion
    }
    
    func fetchUpcomingTask(completion: @escaping ([TaskModel]) -> Void) {
        fetchUpcomingTaskCalled = true
        upcomingTaskCompletion = completion
    }
    
    func fetchFinishedTask(completion: @escaping ([TaskModel]) -> Void) {
        fetchFinishedTaskCalled = true
        finishedTaskCompletion = completion
    }
    
    func finishTask(_ task: TaskModel) {
        finishTaskCalled = true
    }
}