public void RemoveSnake()//Remove Snake //NEEDS LOTS OF WORK { view.WorkFlowHeader("Remove Snake"); List <Snake> snakesToDisplay = repo.ReadAll(); foreach (var snake in snakesToDisplay) { view.DisplaySnake(snake); } while (true) { view.GeneralWriteLine("Which snake would you like to remove?"); int idToRemove = view.SearchSnake(); Snake snakeToRemove = repo.ReadByID(idToRemove); if (view.ConfirmRemoveSnake(snakeToRemove) == true) { repo.Delete(snakeToRemove.ID); break; } else { continue; } } }
public void CanDeleteSnakeTest() { SMRepository repo = new SMRepository(); Snake newSnake = new Snake(); newSnake.ID = 3; newSnake.Family = SnakeFamily.Viperidae; newSnake.CommonSpeciesName = "James"; newSnake.LengthInCentimeters = 35M; newSnake.Venomous = true; Snake third = repo.Create(newSnake); List <Snake> snakes = repo.ReadAll(); Assert.AreEqual(1, snakes.Count()); repo.Delete(3); Assert.AreEqual(0, snakes.Count()); }