Get all required dances for useful output
This commit is contained in:
@@ -310,3 +310,32 @@ def test_mergeHtmlData(mocker):
|
||||
}
|
||||
|
||||
assert finalists == expectedFinalists
|
||||
|
||||
@pytest.fixture(params=range(4))
|
||||
def fixture_getAllDancesInCompetition(request, mocker):
|
||||
def mockCompetition(comp):
|
||||
def mockUser():
|
||||
return mocker.patch('solo_turnier.worker.ResultPerson')
|
||||
def mockDances(dances):
|
||||
def mockDance(name):
|
||||
mock = mocker.patch('solo_turnier.worker.CompetitionResult')
|
||||
mock.dance = name
|
||||
return mock
|
||||
return [mockDance(d) for d in dances]
|
||||
return {mockUser(): mockDances(dances) for dances in comp}
|
||||
|
||||
cases = (
|
||||
([['Samba']], ['Samba']),
|
||||
([['Samba', 'Rumba'], ['Cha Cha']], ['Samba', 'Cha Cha', 'Rumba']),
|
||||
([['Samba', 'Rumba'], ['Cha Cha', 'Tango', 'Langs. Walzer']], ['Samba', 'Cha Cha', 'Rumba', 'Langs. Walzer', 'Tango']),
|
||||
([['Cha Cha', 'Rumba', 'Jive'], ['Quickstep', 'Tango', 'Wiener Walzer', 'Langs. Walzer'], ['Slowfox', 'Langs. Walzer', 'Paso Doble', 'Samba']], ['Samba', 'Cha Cha', 'Rumba', 'Paso Doble', 'Jive', 'Langs. Walzer', 'Tango', 'Wiener Walzer', 'Slowfox', 'Quickstep'])
|
||||
)
|
||||
case = cases[request.param]
|
||||
return (mockCompetition(case[0]), case[1])
|
||||
|
||||
def test_getAllDancesInCompetitions(fixture_getAllDancesInCompetition):
|
||||
print(fixture_getAllDancesInCompetition)
|
||||
data = fixture_getAllDancesInCompetition[0]
|
||||
dataWorker = worker.DataWorker()
|
||||
ret = dataWorker.getAllDancesInCompetitions(data)
|
||||
assert ret == fixture_getAllDancesInCompetition[1]
|
||||
|
||||
Reference in New Issue
Block a user