tests: improve import/export tests pep8ness (#24152)

This commit is contained in:
Benjamin Dauvergne 2018-05-30 08:50:00 +02:00
parent 4edb442bfc
commit e6e7f3c328
1 changed files with 14 additions and 7 deletions

View File

@ -13,14 +13,14 @@ from django.utils.encoding import force_bytes
from django.utils.six import StringIO
from django.utils.timezone import make_aware
from chrono.agendas.models import (Agenda, Event, MeetingType, TimePeriod,
Desk, TimePeriodException)
from chrono.manager.utils import export_site, import_site
from chrono.agendas.models import (Agenda, Event, TimePeriod, Desk, TimePeriodException)
from chrono.manager.utils import import_site
from test_api import some_data, meetings_agenda, time_zone, mock_now
pytestmark = pytest.mark.django_db
def get_output_of_command(command, *args, **kwargs):
old_stdout = sys.stdout
output = sys.stdout = StringIO()
@ -28,6 +28,7 @@ def get_output_of_command(command, *args, **kwargs):
sys.stdout = old_stdout
return output.getvalue()
def test_import_export(app, some_data, meetings_agenda):
# add exception to meeting agenda
desk = meetings_agenda.desk_set.first()
@ -67,8 +68,11 @@ def test_import_export(app, some_data, meetings_agenda):
event = Event(agenda=agenda1, start_datetime=make_aware(datetime.datetime.now()), places=10)
event.save()
desk, _ = Desk.objects.get_or_create(agenda=agenda2, label='Desk A', slug='desk-a')
timeperiod = TimePeriod(weekday=2, desk=desk,
start_time=datetime.time(10, 0), end_time=datetime.time(11, 0))
timeperiod = TimePeriod(
desk=desk,
weekday=2,
start_time=datetime.time(10, 0),
end_time=datetime.time(11, 0))
timeperiod.save()
exception = TimePeriodException(
desk=desk,
@ -84,8 +88,11 @@ def test_import_export(app, some_data, meetings_agenda):
event = Event(agenda=agenda1, start_datetime=make_aware(datetime.datetime.now()), places=10)
event.save()
timeperiod = TimePeriod(weekday=2, desk=desk,
start_time=datetime.time(10, 0), end_time=datetime.time(11, 0))
timeperiod = TimePeriod(
weekday=2,
desk=desk,
start_time=datetime.time(10, 0),
end_time=datetime.time(11, 0))
timeperiod.save()
exception = TimePeriodException(
desk=desk,