Re-write tests in serverspec