Need to work out how to mock the probe method in BasicDatabase and the best practices around it. End result is to make BasicDatabase.probe() return the status I have defined.