diff --git a/smash/web/models/visit.py b/smash/web/models/visit.py index 4e96574d24dfeb8569f8a01b6e3f2f1aeb2e979b..c4c7dae7d4accd34adc47181932827b70d353ad8 100644 --- a/smash/web/models/visit.py +++ b/smash/web/models/visit.py @@ -5,6 +5,7 @@ from dateutil.relativedelta import relativedelta from django.db import models from django.db.models.signals import post_save from django.dispatch import receiver +from django.db import transaction from web.models.constants import BOOL_CHOICES, SUBJECT_TYPE_CHOICES_CONTROL from web.models import Study @@ -89,16 +90,18 @@ class Visit(models.Model): datetime_end=visit_started + time_to_next_visit + relativedelta(months=study.default_visit_duration_in_months) ) - @receiver(post_save, sender=Visit) -def update_visit_number(sender, instance, **kwargs): +def check_visit_number(sender, instance, **kwargs): + # no other solution to ensure the visit_number is in cronological order than to sort the whole list if there are future visits visit = instance - if visit.subject is not None: - count = Visit.objects.filter(subject=visit.subject).filter( - datetime_begin__lte=visit.datetime_begin).count() - if count != visit.visit_number: - visit.visit_number = count - visit.save() - subject_visits = Visit.objects.filter(subject=visit.subject).all() - for subject_visit in subject_visits: - update_visit_number(sender, subject_visit) + if visit.subject is not None: #not sure if select_for_update has an effect, the tests work as well without it + visits_before = Visit.objects.select_for_update().filter(subject=visit.subject).filter(datetime_begin__lt=visit.datetime_begin).count() + # we need to sort the future visits respect to this one, if any + visits = Visit.objects.select_for_update().filter(subject=visit.subject).filter(datetime_begin__gte=visit.datetime_begin).order_by('datetime_begin','datetime_end') + with transaction.atomic(): #not sure if it has an effect, the tests work as well without it + for i, v in enumerate(visits): + if v.visit_number != (visits_before + i + 1): + visit_number = (visits_before + i + 1) + Visit.objects.filter(id=v.id).update(visit_number=visit_number) # does not rise post_save, we avoid recursion + if v.id == visit.id: + visit.visit_number = v.visit_number diff --git a/smash/web/models/worker.py b/smash/web/models/worker.py index ede9eb84f7322c5456033e4e2c7fe903d5bb75a0..78d2772e4eed23ab0703c60bcef75fdb2ed4c31d 100644 --- a/smash/web/models/worker.py +++ b/smash/web/models/worker.py @@ -153,6 +153,15 @@ class Worker(models.Model): return True return False + def current_leave_details(self): + holidays = self.holiday_set.filter(datetime_end__gt=datetime.datetime.now(), + datetime_start__lt=datetime.datetime.now(), + kind=AVAILABILITY_HOLIDAY).order_by('-datetime_end') + if len(holidays) > 0: + return holidays[0] + else: + return None + def disable(self): if self.user is not None: self.user.is_active = False diff --git a/smash/web/static/js/daily_planning.js b/smash/web/static/js/daily_planning.js index 2bb9341335a17d58920a792e01b5a56c0834dcad..816a1ec4256b40b965c969e9108898c3b77e47bf 100644 --- a/smash/web/static/js/daily_planning.js +++ b/smash/web/static/js/daily_planning.js @@ -248,148 +248,146 @@ function remove_event(event) { add_event(event, event.color, event.subject_id, event.location_id, boxBody); } -$(document).ready(function () { - $('#calendar').fullCalendar({ +function addDailyPlanningCalendar(calendar_selector, replace_all, calendar_dict_props){ + + var customButtons = { + datePickerButton: { + text: 'select', + click: function () { + var $btnCustom = $('.fc-datePickerButton-button'); + if ($(".calendar-datepicker").length > 0) { + $(".calendar-datepicker").remove(); + } + else { + $btnCustom.after('<div class="calendar-datepicker"/>'); + $(".calendar-datepicker").datepicker().on('changeDate', function (ev) { + $(calendar_selector).fullCalendar('gotoDate', ev.date); + $(".calendar-datepicker").remove(); + }); + } + } + }, + save: { + text: 'Save', + click: function () { + calendarEvents = $(calendar_selector).fullCalendar('clientEvents'); + eventsToPersist = []; + var saveButton = $(".fc-save-button"); + var currentBorder = saveButton.css('border-color'); + $.each(calendarEvents, function (i, calendar_event) { + if (calendar_event.rendering !== "background") { + eventsToPersist.push({ + 'link_id': calendar_event.link_id, + 'appointment_id': calendar_event.appointment_id, + 'link_who': parseInt(calendar_event.resourceId), + 'start': calendar_event.start.format() + }); + if (calendar_event.link_id !== undefined) { + var index = eventsCleared.indexOf(calendar_event.link_id); + if (index > -1) { + eventsCleared.splice(index, 1); + } + } else { + var index = appointmentsCleared.indexOf(calendar_event.appointment_id); + if (index > -1) { + appointmentsCleared.splice(index, 1); + } + } + } + }); + $.post({ + url: events_url, + data: { + events_to_persist: JSON.stringify(eventsToPersist), + events_to_clear: JSON.stringify(eventsCleared), + appointments_to_clear: JSON.stringify(appointmentsCleared) + }, + dataType: "json" + }).done(function (data) { + + saveButton.css('border-color', 'green'); + setTimeout(function () { + saveButton.css('border-color', currentBorder); + }, 200); + }).error(function (data) { + + console.log(data); + saveButton.css('border-color', 'red'); + showErrorInfo("There was an unexpected problem with saving data. " + "Please contact administrators."); + setTimeout(function () { + saveButton.delay(200).css('border-color', currentBorder); + }, 200); + }); + } + }, + clear: { + text: 'Clear', + click: function () { + calendarEvents = $(calendar_selector).fullCalendar('clientEvents'); + $.each(calendarEvents, function (i, calendar_event) { + remove_event(calendar_event); + }); + } + }, + toPdf: { + text: 'PDF', + click: function () { + var srcEl = document.getElementById("calendar"); + var parent = srcEl.parentNode; + var container = document.createElement("div"); + container.style.backgroundColor = "#FFFFFF"; + container.style.width = "3840px"; + document.body.appendChild(container); + container.appendChild(srcEl); + + var pdf = new jsPDF('l', 'mm', [1485, 270]); + pdf.addHTML(container, function () { + pdf.save('daily-planning.pdf'); + parent.appendChild(srcEl); + document.body.removeChild(container); + }); + } + } + }; + + var default_properties = { schedulerLicenseKey: 'CC-Attribution-NonCommercial-NoDerivatives', defaultView: 'agendaDay', eventDurationEditable: false, eventStartEditable: true, editable: true, selectable: true, - eventOverlap: function (stillEvent, movingEvent) { - if (stillEvent.rendering === "background") { - return true; - } - return false; - }, + droppable: true, weekends: false, + selectHelper: true, + groupByResource: true, + displayEventTime: true, scrollTime: '08:00', slotDuration: '00:30', snapDuration: '00:05', - displayEventTime: true, - resourceOrder: 'role', - resourceGroupField: 'role', - dragRevertDuration: 0, minTime: "08:00:00", maxTime: "19:00:00", - groupByResource: true, - height: "auto", - customButtons: { - datePickerButton: { - text: 'select', - click: function () { - var $btnCustom = $('.fc-datePickerButton-button'); - if ($(".calendar-datepicker").length > 0) { - $(".calendar-datepicker").remove(); - } - else { - $btnCustom.after('<div class="calendar-datepicker"/>'); - $(".calendar-datepicker").datepicker().on('changeDate', function (ev) { - $('#calendar').fullCalendar('gotoDate', ev.date); - $(".calendar-datepicker").remove(); - }); - } - } - }, - save: { - text: 'Save', - click: function () { - calendarEvents = $('#calendar').fullCalendar('clientEvents'); - eventsToPersist = []; - var saveButton = $(".fc-save-button"); - var currentBorder = saveButton.css('border-color'); - $.each(calendarEvents, function (i, calendar_event) { - if (calendar_event.rendering !== "background") { - eventsToPersist.push({ - 'link_id': calendar_event.link_id, - 'appointment_id': calendar_event.appointment_id, - 'link_who': parseInt(calendar_event.resourceId), - 'start': calendar_event.start.format() - }); - if (calendar_event.link_id !== undefined) { - var index = eventsCleared.indexOf(calendar_event.link_id); - if (index > -1) { - eventsCleared.splice(index, 1); - } - } else { - var index = appointmentsCleared.indexOf(calendar_event.appointment_id); - if (index > -1) { - appointmentsCleared.splice(index, 1); - } - } - } - }); - $.post({ - url: events_url, - data: { - events_to_persist: JSON.stringify(eventsToPersist), - events_to_clear: JSON.stringify(eventsCleared), - appointments_to_clear: JSON.stringify(appointmentsCleared) - }, - dataType: "json" - }).done(function (data) { - - saveButton.css('border-color', 'green'); - setTimeout(function () { - saveButton.css('border-color', currentBorder); - }, 200); - }).error(function (data) { - - console.log(data); - saveButton.css('border-color', 'red'); - showErrorInfo("There was an unexpected problem with saving data. " + - "Please contact administrators."); - setTimeout(function () { - saveButton.delay(200).css('border-color', currentBorder); - }, 200); - }); - } - }, - clear: { - text: 'Clear', - click: function () { - calendarEvents = $('#calendar').fullCalendar('clientEvents'); - $.each(calendarEvents, function (i, calendar_event) { - remove_event(calendar_event); - }); - } - }, - toPdf: { - text: 'PDF', - click: function () { - var srcEl = document.getElementById("calendar"); - var parent = srcEl.parentNode; - var container = document.createElement("div"); - container.style.backgroundColor = "#FFFFFF"; - container.style.width = "3840px"; - document.body.appendChild(container); - container.appendChild(srcEl); - - var pdf = new jsPDF('l', 'mm', [1485, 270]); - pdf.addHTML(container, function () { - pdf.save('daily-planning.pdf'); - parent.appendChild(srcEl); - document.body.removeChild(container); - }); - } - } - - }, - viewRender: function (view, element) { - var date = view.start.format('YYYY-MM-DD'); - $('#calendar').fullCalendar('removeEvents'); - get_subjects_events(date); - }, businessHours: { start: '08:00', end: '19:00' }, + dragRevertDuration: 0, + height: "auto", + customButtons: customButtons, header: { left: 'prev,next today', center: 'title, datePickerButton', right: 'save, clear, toPdf' }, - droppable: true, + // VIEW + viewRender: function (view, element) { + var date = view.start.format('YYYY-MM-DD'); + $(calendar_selector).fullCalendar('removeEvents'); + get_subjects_events(date); + }, + // RESOURCES + resourceOrder: 'role', + resourceGroupField: 'role', resourceAreaWidth: '15%', resourceLabelText: 'Workers', refetchResourcesOnNavigate: true, @@ -421,7 +419,7 @@ $(document).ready(function () { }, resources: function(callback){ setTimeout(function(){ - var view = $('#calendar').fullCalendar('getView'); + var view = $(calendar_selector).fullCalendar('getView'); $.ajax({ url: resources_url, type: 'GET', @@ -431,13 +429,14 @@ $(document).ready(function () { } }).then(function(resources){ //Filter out roles - var checked_roles = $('.role_list_item > input:checked').map( (i,e) => e.value).toArray(); - resources = resources.filter(resource => checked_roles.includes(resource.role)); - callback(resources) + if($('.role_list_item > input').length > 0){ + var checked_roles = $('.role_list_item > input:checked').map( (i,e) => e.value).toArray(); + resources = resources.filter(resource => checked_roles.includes(resource.role)); + } + callback(resources); }); }, 0); }, - events: [], eventRender: function (event, element) { if (event.rendering !== 'background') { var content = @@ -457,6 +456,13 @@ $(document).ready(function () { } else { } }, + // EVENTS + eventOverlap: function (stillEvent, movingEvent) { + if (stillEvent.rendering === "background") { + return true; + } + return false; + }, selectAllow: function (selectInfo) { return false; }, @@ -471,7 +477,6 @@ $(document).ready(function () { eventDragStart: function (event, jsEvent, view) { $('.popover').popover('hide'); }, - selectHelper: true, drop: function (date, jsEvent, ui, resourceId) { $(this).remove(); }, @@ -482,9 +487,20 @@ $(document).ready(function () { }; setTimeout(function() {resizeCalendarColumns()}, 100); } - }); -}) -; + }; + + // REPLACE DEFAULT PROPERTIES + if(replace_all){ + default_properties = calendar_dict_props; + }else{ + for (var key in calendar_dict_props) { + default_properties[key] = calendar_dict_props[key]; + } + } + + $(calendar_selector).fullCalendar(default_properties); + +} //RESIZE COLUMNS AND ENABLE HORIZONTAL SCROLL function resizeCalendarColumns(){ diff --git a/smash/web/templates/_base.html b/smash/web/templates/_base.html index e34aa85db3fe86f59fcb8206505520d72c35a65c..b08784ea6188dd5a401844dbe45cda74c254ba30 100644 --- a/smash/web/templates/_base.html +++ b/smash/web/templates/_base.html @@ -270,7 +270,7 @@ desired effect {% block footer %} <!-- To the right --> <div class="pull-right hidden-xs"> - Version: <strong>0.12.0</strong> (6 Nov 2018) + Version: <strong>0.12.1</strong> (13 Nov 2018) </div> <!-- Default to the left --> diff --git a/smash/web/templates/appointments/add.html b/smash/web/templates/appointments/add.html index bf87695b3f1f00caf74b6db18777f2ec762445be..9fc04985e5e152a2daf9db4790479f9bff949f36 100644 --- a/smash/web/templates/appointments/add.html +++ b/smash/web/templates/appointments/add.html @@ -4,12 +4,16 @@ {% block styles %} {{ block.super }} + <!-- DataTables --> <link rel="stylesheet" href="{% static 'AdminLTE/plugins/datatables/dataTables.bootstrap.css' %}"> <!-- fullCalendar 2.2.5--> <link rel="stylesheet" href="{% static 'AdminLTE/plugins/fullcalendar/fullcalendar.min.css' %}"> <link rel="stylesheet" href="{% static 'AdminLTE/plugins/fullcalendar/fullcalendar.print.css' %}" media="print"> + <link rel="stylesheet" href="{% static 'AdminLTE/plugins/fullcalendar/fullcalendar_custom.print.css' %}" media="print" /> + <link rel="stylesheet" href="{% static 'fullcalendar-scheduler/scheduler.min.css' %}"> + <link rel="stylesheet" href="{% static 'css/daily_planning.css' %}"> {% include "includes/datetimepicker.css.html" %} <link rel="stylesheet" href="{% static 'css/appointment.css' %}"> @@ -18,6 +22,17 @@ text-align: center !important; margin: 0; } + .tooltip_image{ + width: 20px !important; + margin: 5px; + } + .subject_title{ + padding: 0 !important; + } + .subject_title + .tooltip > .tooltip-inner, .column_title + .tooltip > .tooltip-inner { + background-color: #fafafa; + border: solid 1px #ccc; + } </style> {% endblock styles %} @@ -47,49 +62,61 @@ New appointment for visit from {{visit_start}} to {{visit_end}} <form method="post" action="" class="form-horizontal"> {% csrf_token %} - <div class="box-body"> - <div class="col-sm-6"> - {% for field in form %} - <div class="form-group {% if field.errors %}has-error{% endif %} {% if field|is_checkbox %}multi-checkboxes{% endif %}"> - <label class="col-sm-4 control-label"> - {{ field.label }} - </label> - - <div class="col-sm-8"> - {{ field|add_class:'form-control' }} + <div class="row"> + <div class="box-body"> + <div class="col-md-6"> + {% for field in form %} + <div class="form-group {% if field.errors %}has-error{% endif %} {% if field|is_checkbox %}multi-checkboxes{% endif %}"> + <label class="col-sm-4 control-label"> + {{ field.label }} + </label> + + <div class="col-sm-8"> + {{ field|add_class:'form-control' }} + </div> + + {% if field.errors %} + <span class="help-block"> + {{ field.errors }} + </span> + {% endif %} + </div> + {% endfor %} + </div> + <div class="col-md-6"> + <div class="box box-primary"> + <div class="box-body no-padding"> + <div id="side_calendar"></div> </div> - - {% if field.errors %} - <span class="help-block"> - {{ field.errors }} - </span> - {% endif %} - </div> - {% endfor %} - </div> - <div class="col-md-6"> - <div class="box box-primary"> - <div class="box-body no-padding"> - <div id="calendar"></div> </div> </div> - </div> - </div><!-- /.box-body --> - - - <div class="box-footer"> - <div class="col-sm-6"> - <button type="submit" class="btn btn-block btn-success">Add</button> - </div> - <div class="col-sm-6"> - <a href="{% url 'web.views.visits' %}" class="btn btn-block btn-default">Cancel</a> - </div> - </div><!-- /.box-footer --> + </div><!-- /.box-body --> + </div> + <div class="row"> + <div class="box-footer"> + <div class="col-sm-6"> + <button type="submit" class="btn btn-block btn-success">Add</button> + </div> + <div class="col-sm-6"> + <a href="{% url 'web.views.visits' %}" class="btn btn-block btn-default">Cancel</a> + </div> + </div><!-- /.box-footer --> + </div> </form> </div> + <div class="row"> + <div class="col-md-12"> + <div class="box box-primary"> + <div class="box-body no-padding"> + <div id="calendar"></div> + </div> + </div> + </div> + </div> + {% endblock %} @@ -101,8 +128,9 @@ New appointment for visit from {{visit_start}} to {{visit_end}} <script src="{% static 'AdminLTE/plugins/datatables/jquery.dataTables.min.js' %}"></script> <script src="{% static 'AdminLTE/plugins/datatables/dataTables.bootstrap.min.js' %}"></script> <script src="{% static 'AdminLTE/plugins/moment.js/moment.min.js' %}"></script> - <script src="{% static 'AdminLTE/plugins/fullcalendar/fullcalendar.min.js' %}"></script> <script src="{% static 'js/appointment.js' %}"></script> + <script src="{% static 'fullcalendar-scheduler/lib/fullcalendar.min.js' %}"></script> + <script src="{% static 'fullcalendar-scheduler/scheduler.min.js' %}"></script> <script> $(function () { $('#table').DataTable({ @@ -113,14 +141,16 @@ New appointment for visit from {{visit_start}} to {{visit_end}} "info": true, "autoWidth": false }); - $('#calendar').fullCalendar({ - defaultDate: moment('{{visit_start}}'), + //SIDE CALENDAR + addDailyPlanningCalendar('#side_calendar', true, { + schedulerLicenseKey: 'CC-Attribution-NonCommercial-NoDerivatives', + defaultDate: moment.max(moment('{{visit_start}}'), moment()), + editable: false, header: { left: 'prev,next today', center: 'title', right: 'month,agendaWeek' }, - editable: false, dayClick: function (date, jsEvent, view) { var dateString = date.format(); if (dateString.indexOf("T") >= 0) { @@ -129,6 +159,7 @@ New appointment for visit from {{visit_start}} to {{visit_end}} dateString = dateString + " 09:00"; } document.getElementById("id_datetime_when").value = dateString; + $('#id_datetime_when').change(); getWorkerAvailability(); }, eventClick: function (calEvent, jsEvent, view) { @@ -214,7 +245,35 @@ New appointment for visit from {{visit_start}} to {{visit_end}} } $('select.search_worker_availability, input[name="datetime_when"], input[name="length"]').on("change", getWorkerAvailability); + + $("#id_datetime_when").on("change paste keyup", function() { + var date = $("#id_datetime_when").val(); + date = moment(date); + $('#calendar').fullCalendar('gotoDate', date); + }); + + var resources_url = '{% url 'web.api.workers.daily_planning' %}'; + var events_url = '{% url 'web.api.events_persist' %}'; + + $(document).ready(function () { + addDailyPlanningCalendar('#calendar', false, { + eventDurationEditable: false, + eventStartEditable: false, + editable: false, + selectable: false, + droppable: false, + customButtons: {}, + header: { + left: '', + center: 'title', + right: '' + } + }); + }); + </script> {% include "includes/datetimepicker.js.html" %} + <script src="{% static 'js/daily_planning.js' %}"></script> + {% endblock scripts %} diff --git a/smash/web/templates/daily_planning.html b/smash/web/templates/daily_planning.html index 63b552701589c09f7ce5b04d23c9e827277bc131..4d6cc2058df16128406951a76d9223d55730fbeb 100644 --- a/smash/web/templates/daily_planning.html +++ b/smash/web/templates/daily_planning.html @@ -85,6 +85,10 @@ $('#calendar').fullCalendar('refetchResources'); } + $(document).ready(function () { + addDailyPlanningCalendar('#calendar', false, {}); + }); + </script> {% include "includes/datepicker.js.html" %} <script src="{% static 'js/daily_planning.js' %}"></script> diff --git a/smash/web/templates/doctors/index.html b/smash/web/templates/doctors/index.html index 1accfd94d238279684ff6d4fe11d6085b05976b8..0884c2a59d786277475519c5174b84311e09d260 100644 --- a/smash/web/templates/doctors/index.html +++ b/smash/web/templates/doctors/index.html @@ -5,6 +5,20 @@ {{ block.super }} <!-- DataTables --> <link rel="stylesheet" href="{% static 'AdminLTE/plugins/datatables/dataTables.bootstrap.css' %}"> + <style type="text/css"> + .no_leave{ + height: 100%; + line-height: 2.5; + text-align: center; + background-color: #00a65b; + } + .on_leave{ + height: 100%; + line-height: 2.5; + text-align: center; + background-color: #dd4b39; + } + </style> {% endblock styles %} {% block ui_active_tab %}'workers'{% endblock ui_active_tab %} @@ -53,7 +67,7 @@ <th>Languages</th> <th>Details</th> {% if worker_type == 'STAFF' %} - <th>On leave</th> + <th>At work</th> <th>Disabled</th> {% endif %} </tr> @@ -82,9 +96,9 @@ {% if worker_type == 'STAFF' %} <td> {% if worker.is_on_leave %} - <button type="button" class="btn btn-block btn-danger">YES</button> + <div title="Worker is on leave until: {{ worker.current_leave_details.datetime_end }}" class="on_leave">✘</div> {% else %} - <button type="button" class="btn btn-block btn-success">NO</button> + <div title="Worker is not on leave" class="no_leave">✔</div> {% endif %} </td> <td> @@ -120,5 +134,7 @@ "autoWidth": false }); }); + + $('.no_leave, .on_leave').tooltip(); </script> {% endblock scripts %} diff --git a/smash/web/tests/models/test_visit.py b/smash/web/tests/models/test_visit.py index c7d577fe5ab759f4d29d7e104db402fa33c63e78..85ea3d4fb4ecbf740669cbfcf0b44c43927a3246 100644 --- a/smash/web/tests/models/test_visit.py +++ b/smash/web/tests/models/test_visit.py @@ -5,9 +5,52 @@ from django.test import TestCase from web.models import Visit from web.models.constants import SUBJECT_TYPE_CHOICES_PATIENT from web.tests.functions import create_study_subject, create_visit +from web.utils import get_today_midnight_date +import logging +logger = logging.getLogger(__name__) class VisitModelTests(TestCase): + def test_so_called_no_concurrency(self): + subject = create_study_subject() + Visit.objects.filter(subject=subject).all().delete() + visit1 = create_visit(subject=subject) + visit1.datetime_end = get_today_midnight_date() + datetime.timedelta(days=1) + visit1.save() + self.assertEquals(1, visit1.visit_number) + visit2 = create_visit(subject=subject) + visit2.datetime_end = get_today_midnight_date() + datetime.timedelta(days=2) + visit2.save() + visit2.refresh_from_db() + self.assertEquals(2, visit2.visit_number) + + def test_socalled_concurrency(self): + subject = create_study_subject() + Visit.objects.filter(subject=subject).all().delete() + visit1 = create_visit(subject=subject) #post save will be raised + visit2 = create_visit(subject=subject) #post save will be raised + + visit1.datetime_end = get_today_midnight_date() + datetime.timedelta(days=1) + visit2.datetime_end = get_today_midnight_date() + datetime.timedelta(days=2) + + visit1.save() #post save will be raised + visit2.save() #post save will be raised + + visit1.refresh_from_db() + visit2.refresh_from_db() + self.assertEquals(1, visit1.visit_number) + self.assertEquals(2, visit2.visit_number) + + datetime_begin = get_today_midnight_date() - datetime.timedelta(days=60) + datetime_end = get_today_midnight_date() - datetime.timedelta(days=55) + visit3 = create_visit(subject=subject, datetime_begin=datetime_begin, datetime_end=datetime_end) #post save will be raised + visit1.refresh_from_db() + visit2.refresh_from_db() + visit3.refresh_from_db() + self.assertEquals(1, visit3.visit_number) + self.assertEquals(2, visit1.visit_number) + self.assertEquals(3, visit2.visit_number) + def test_mark_as_finished(self): subject = create_study_subject() visit = create_visit(subject) diff --git a/smash/web/tests/models/test_worker.py b/smash/web/tests/models/test_worker.py index 338ef668e0ca0b9befe465f28e17884ba7613f98..97c0ac5092bbbf483c14fb919fbe944eae5ac5ef 100644 --- a/smash/web/tests/models/test_worker.py +++ b/smash/web/tests/models/test_worker.py @@ -61,6 +61,16 @@ class WorkerModelTests(TestCase): self.assertTrue(worker.is_on_leave()) + def test_current_leave_details(self): + worker = create_worker() + self.assertEqual(worker.current_leave_details(), None) + h = Holiday(person=worker, + datetime_start=get_today_midnight_date() + datetime.timedelta(days=-2), + datetime_end=get_today_midnight_date() + datetime.timedelta(days=2)) + h.save() + + self.assertEqual(worker.current_leave_details(), h) + def test_get_by_user_for_anonymous(self): self.assertIsNone(Worker.get_by_user(AnonymousUser()))