tests: improve import/export tests pep8ness (#24152)
This commit is contained in:
parent
4edb442bfc
commit
e6e7f3c328
|
@ -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,
|
||||
|
|
Loading…
Reference in New Issue