public void AddToWishList() { int id = db.Games.First().Id; var controller = new FriendshipController(); Member member = controller.DbContext.Members.First(); controller.ControllerContext = GetControllerContext(db, member, "Member"); var initCount = controller.DbContext.WishLists.Where(w => w.MemberId == member.Id && w.GameId == id).ToList().Count; // Act ViewResult result = controller.AddToWishList(id) as ViewResult; var finCount = controller.DbContext.WishLists.Where(w => w.MemberId == member.Id && w.GameId == id).ToList().Count; // Assert Assert.AreEqual(initCount + 1, finCount); }
public void RemoveFromWishlist() { var controller = new FriendshipController(); Member member = controller.DbContext.Members.First(); int id = db.Games.FirstOrDefault().Id; controller.ControllerContext = GetControllerContext(db, member, "Member"); //Adding it first controller.AddToWishList(id); var initCount = controller.DbContext.WishLists.Where(w => w.MemberId == member.Id && w.GameId == id).ToList().Count; // Act ViewResult result = controller.RemoveFromWishlist(id, member.Id) as ViewResult; var finCount = controller.DbContext.WishLists.Where(w => w.MemberId == member.Id && w.GameId == id).ToList().Count; // Assert Assert.AreEqual(initCount - 1, finCount); }