public void ActionNotFoundShoudReturnPageNotFoundView()
{
var mockHttpContext = new Mock<HttpContextBase>();
var response = new Mock<HttpResponseBase>();
mockHttpContext.SetupGet(x => x.Response).Returns(response.Object);
var db = new PetCareDbContext();
var repo = new Repository<User>(db);
var users = new UsersService(repo);
var controller = new ErrorController(users)
{
ControllerContext = new ControllerContext()
{
HttpContext = mockHttpContext.Object
}
};
controller.WithCallTo(x => x.NotFound())
.ShouldRenderView("NotFound");
}
public override Task OnConnected()
{
if (!Context.User.Identity.IsAuthenticated)
{
return base.OnConnected();
}
User currentUser = new User();
var notifications = new List<Notification>();
var recordedNotifications = new List<NotificationViewModel>();
using (var db = new PetCareDbContext())
{
currentUser = db.Users.FirstOrDefault(x => x.UserName == Context.User.Identity.Name);
if (currentUser.IsVet || currentUser.Pets.Count == 0)
{
return base.OnConnected();
}
var room = db.Rooms.FirstOrDefault(x => x.UserId == currentUser.Id);
if (room == null)
{
db.Rooms.Add(room = new Room()
{
Name = currentUser.UserName + "Room",
UserId = currentUser.Id
});
db.SaveChanges();
}
Groups.Add(Context.ConnectionId, room.Name);
var pets = currentUser.Pets.Where(x => x.HealthRecord != null).ToList();
if (pets.Count == 0)
{
return base.OnConnected();
}
var visits = pets
.SelectMany(x => x.HealthRecord.VetVisits)
.Where(x => x.DateTime.Day > DateTime.UtcNow.Day && x.DateTime.Day <= DateTime.UtcNow.AddDays(1).Day)
.ToList();
foreach (var visit in visits)
{
if (visit.Notification == null)
{
var viewModel = new NotificationViewModel()
{
DateTime = DateTime.UtcNow,
Message = string.Format(GlobalConstants.VetVisitNotificationInThreeDays, visit.Pet.Pet.Name, visit.Vet.FirstName, visit.Vet.LastName),
User = currentUser,
IsSeen = false,
VetVisitId = visit.Id
};
var notification = AutoMapper.Mapper.Map<NotificationViewModel, Notification>(viewModel);
db.Notifications.Add(notification);
}
else
{
notifications.Add(visit.Notification);
}
}
db.SaveChanges();
recordedNotifications = notifications
.Where(x => x.User.Id == currentUser.Id)
.AsQueryable()
.ProjectTo<NotificationViewModel>()
.ToList();
}
var anonymous = recordedNotifications.Select(x => new
{
DateTime = x.DateTime,
IsSeen = x.IsSeen,
Message = x.Message,
VetVisitId = x.VetVisitId
}).ToList();
var json = JsonConvert.SerializeObject(anonymous);
Clients.Caller.notify(json);
return base.OnConnected();
}
public void SetNotificationsSeen(ICollection<int> ids)
{
using (var db = new PetCareDbContext())
{
var notificationsToUpdate = db.Notifications.Where(x => ids.Contains(x.VetVisitId)).ToList();
foreach (var item in notificationsToUpdate)
{
if (item.IsSeen)
{
continue;
}
item.IsSeen = true;
db.Notifications.Attach(item);
db.Entry(item).State = EntityState.Modified;
db.SaveChanges();
}
}
}
