Create attractive VASL scenarios, with loads of useful information embedded to assist with game play.
https://vasl-templates.org
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
237 lines
10 KiB
237 lines
10 KiB
""" Test generating OB SETUP snippets. """
|
|
|
|
import re
|
|
import types
|
|
|
|
from vasl_templates.webapp.tests.utils import \
|
|
get_nationalities, wait_for_clipboard, get_stored_msg, set_stored_msg_marker, select_tab, \
|
|
find_child, find_children, \
|
|
add_simple_note, edit_simple_note, get_sortable_entry_count, drag_sortable_entry_to_trash, \
|
|
init_webapp, wait_for, set_scenario_date, set_player, set_theater
|
|
|
|
# ---------------------------------------------------------------------
|
|
|
|
# NOTE: Handling of OB setups and OB notes is identical (the only difference
|
|
# is in the template files, where OB setups have a colored header).
|
|
|
|
def test_ob_setup( webapp, webdriver ):
|
|
"""Test generating OB setup snippets."""
|
|
_do_test_ob_entries( webapp, webdriver, "ob_setups" )
|
|
|
|
def test_ob_notes( webapp, webdriver ):
|
|
"""Test generating OB note snippets."""
|
|
_do_test_ob_entries( webapp, webdriver, "ob_notes" )
|
|
|
|
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
|
|
|
|
def _do_test_ob_entries( webapp, webdriver, ob_type ):
|
|
"""Test generating OB setup/notes."""
|
|
|
|
# initialize
|
|
init_webapp( webapp, webdriver )
|
|
sortable1 = find_child( "#{}-sortable_1".format( ob_type ) )
|
|
sortable2 = find_child( "#{}-sortable_2".format( ob_type ) )
|
|
|
|
# generate OB setup/note snippets for both players
|
|
select_tab( "ob1" )
|
|
add_simple_note( sortable1, "{} #1".format(ob_type), None )
|
|
add_simple_note( sortable1, "{} #2".format(ob_type), "2px" )
|
|
select_tab( "ob2" )
|
|
add_simple_note( sortable2, "<i>{}</i> #3".format(ob_type), "3px" )
|
|
|
|
# check that snippets are generated correctly
|
|
def check_snippet( sortable, entry_no, expected ):
|
|
"""Generate the snippet for an OB setup/note."""
|
|
elems = find_children( "li img.snippet", sortable )
|
|
elems[entry_no].click()
|
|
if ob_type == "ob_notes":
|
|
expected = re.sub( r" \(col=.*?\)", "", expected )
|
|
assert wait_for_clipboard( 2, expected )
|
|
select_tab( "ob1" )
|
|
check_snippet( sortable1, 0,
|
|
"[German] [{} #1] (col=[OBCOL:german/OBCOL-BORDER:german])".format( ob_type )
|
|
)
|
|
check_snippet( sortable1, 1,
|
|
"[German] [{} #2] (col=[OBCOL:german/OBCOL-BORDER:german]) (width=[2px])".format( ob_type )
|
|
)
|
|
select_tab( "ob2" )
|
|
check_snippet( sortable2, 0,
|
|
"[Russian] [<i>{}</i> #3] (col=[OBCOL:russian/OBCOL-BORDER:russian]) (width=[3px])".format( ob_type )
|
|
)
|
|
|
|
# make some changes and check the snippets again
|
|
edit_simple_note( sortable2, 0, "updated {} #3".format(ob_type), "" )
|
|
select_tab( "ob1" )
|
|
edit_simple_note( sortable1, 1, "<i>updated {} #2</i>".format(ob_type), "200px" )
|
|
edit_simple_note( sortable1, 0, None, "100px" )
|
|
select_tab( "ob2" )
|
|
check_snippet( sortable2, 0,
|
|
"[Russian] [updated {} #3] (col=[OBCOL:russian/OBCOL-BORDER:russian])".format( ob_type )
|
|
)
|
|
select_tab( "ob1" )
|
|
check_snippet( sortable1, 1,
|
|
"[German] [<i>updated {} #2</i>] (col=[OBCOL:german/OBCOL-BORDER:german]) (width=[200px])".format( ob_type )
|
|
)
|
|
check_snippet( sortable1, 0,
|
|
"[German] [{} #1] (col=[OBCOL:german/OBCOL-BORDER:german]) (width=[100px])".format( ob_type ) )
|
|
|
|
# delete an OB setup/note by dragging it into the trash
|
|
assert get_sortable_entry_count( sortable1 ) == 2
|
|
drag_sortable_entry_to_trash( sortable1, 1 )
|
|
assert get_sortable_entry_count( sortable1 ) == 1
|
|
|
|
# delete an OB setup/note by emptying its caption
|
|
edit_simple_note( sortable1, 0, "", None )
|
|
assert get_sortable_entry_count( sortable1 ) == 0
|
|
|
|
# ---------------------------------------------------------------------
|
|
|
|
def test_nationality_specific( webapp, webdriver ): #pylint: disable=too-many-locals
|
|
"""Check that nationality-specific buttons are shown/hidden correctly."""
|
|
|
|
# initialize
|
|
init_webapp( webapp, webdriver )
|
|
nationalities = get_nationalities( webapp )
|
|
|
|
def do_check_snippets( btn, date, expected, warning ):
|
|
"""Check that snippets are being generated correctly."""
|
|
|
|
# change the scenario date, check that the button is displayed correctly
|
|
set_scenario_date( "{:02}/01/{:04}".format( date[1], date[0] ) )
|
|
select_tab( "ob1" )
|
|
classes = btn.get_attribute( "class" )
|
|
classes = classes.split() if classes else []
|
|
if warning:
|
|
assert "inactive" in classes
|
|
else:
|
|
assert "inactive" not in classes
|
|
|
|
# test snippet generation
|
|
marker = set_stored_msg_marker( "_last-warning_" )
|
|
btn.click()
|
|
wait_for_clipboard( 2, expected )
|
|
|
|
# check if a warning was issued
|
|
last_warning = get_stored_msg( "_last-warning_" )
|
|
if warning:
|
|
assert "are only available" in last_warning
|
|
expected_image_url = "snippet-disabled.png"
|
|
else:
|
|
assert last_warning == marker
|
|
expected_image_url = "snippet.png"
|
|
wait_for( 2,
|
|
lambda: expected_image_url in find_child( "img", btn ).get_attribute( "src" )
|
|
)
|
|
|
|
# initialize
|
|
def check_pf_snippets( nat ): #pylint: disable=unused-argument
|
|
"""Check that the PF snippets are generated correctly."""
|
|
btn = find_child( "button[data-id='pf']" )
|
|
col = "[OBCOL:german]/[OBCOL-BORDER:german]"
|
|
do_check_snippets( btn, (1942,1), "PF: range=[1] ; check=[3] ; col={}".format(col), True )
|
|
do_check_snippets( btn, (1943,9), "PF: range=[1] ; check=[3] ; col={}".format(col), True )
|
|
do_check_snippets( btn, (1943,10), "PF: range=[1] ; check=[3] ; col={}".format(col), False )
|
|
do_check_snippets( btn, (1944,5), "PF: range=[1] ; check=[3] ; col={}".format(col), False )
|
|
do_check_snippets( btn, (1944,6), "PF: range=[2] ; check=[3] ; col={}".format(col), False )
|
|
do_check_snippets( btn, (1944,12), "PF: range=[2] ; check=[3] ; col={}".format(col), False )
|
|
do_check_snippets( btn, (1945,1), "PF: range=[3] ; check=[4] ; col={}".format(col), False )
|
|
do_check_snippets( btn, (1946,1), "PF: range=[3] ; check=[4] ; col={}".format(col), False )
|
|
|
|
# initialize
|
|
def check_psk_snippets( nat ): #pylint: disable=unused-argument
|
|
"""Check that the PSK snippets are generated correctly."""
|
|
btn = find_child( "button[data-id='psk']" )
|
|
expected = "====> whoosh! ; col=[OBCOL:german]/[OBCOL-BORDER:german]"
|
|
do_check_snippets( btn, (1942,1), expected, True )
|
|
do_check_snippets( btn, (1943,8), expected, True )
|
|
do_check_snippets( btn, (1943,9), expected, False )
|
|
do_check_snippets( btn, (1944,1), expected, False )
|
|
|
|
# initialize
|
|
def check_baz_snippets( nat ): #pylint: disable=unused-argument
|
|
"""Check that the BAZ snippets are generated correctly."""
|
|
btn = find_child( "button[data-id='baz']" )
|
|
do_check_snippets( btn, (1941,1), "BAZ: none", True )
|
|
do_check_snippets( btn, (1942,10), "BAZ: none", True )
|
|
col = "[OBCOL:american]/[OBCOL-BORDER:american]"
|
|
do_check_snippets( btn, (1942,11), "BAZ: '43 ; range=[4] ; X#=[10] ; TK#=[13] ; col={}".format(col), False )
|
|
do_check_snippets( btn, (1943,1), "BAZ: '43 ; range=[4] ; X#=[10] ; TK#=[13] ; col={}".format(col), False )
|
|
do_check_snippets( btn, (1944,1), "BAZ: '44 ; range=[4] ; X#=[11] ; TK#=[16] ; col={}".format(col), False )
|
|
do_check_snippets( btn, (1945,1),
|
|
"BAZ: '45 ; range=[5] ; X#=[11] ; TK#=[16] ; WP#=[6] ; col={}".format(col),
|
|
False
|
|
)
|
|
do_check_snippets( btn, (1946,1),
|
|
"BAZ: '45 ; range=[5] ; X#=[11] ; TK#=[16] ; WP#=[6] ; col={}".format(col),
|
|
False
|
|
)
|
|
|
|
# initialize
|
|
def check_piat_snippets( nat ):
|
|
"""Check that the PIAT snippets are generated correctly."""
|
|
wait_for_clipboard( 2,
|
|
"piat template ; col=[OBCOL:{}]/[OBCOL-BORDER:{}]".format( nat, nat )
|
|
)
|
|
|
|
# initialize
|
|
def check_atmm_snippets( nat ): #pylint: disable=unused-argument
|
|
"""Check that the ATMM snippets are generated correctly."""
|
|
btn = find_child( "button[data-id='atmm']" )
|
|
expected = "Kaboom!!! ; col=[OBCOL:german]/[OBCOL-BORDER:german]"
|
|
do_check_snippets( btn, (1943,12), expected, True )
|
|
do_check_snippets( btn, (1944,1), expected, False )
|
|
do_check_snippets( btn, (1944,12), expected, False )
|
|
do_check_snippets( btn, (1945,1), expected, False )
|
|
|
|
# initialize
|
|
nationality_specific_buttons = {
|
|
"mol": [ "russian", "Burn, baby, burn! ; col=[OBCOL:russian]/[OBCOL-BORDER:russian]" ],
|
|
"mol-p": [ "russian", "mol-p template ; col=[OBCOL:russian]/[OBCOL-BORDER:russian]" ],
|
|
"pf": [ "german", check_pf_snippets ],
|
|
"psk": [ "german", check_psk_snippets ],
|
|
"atmm": [ "german", check_atmm_snippets ],
|
|
"baz": [ "american", check_baz_snippets ],
|
|
"baz45": [ ("american","Korea"), "BAZ 45 (from K:FW)" ],
|
|
"baz50": [ ("american","Korea"), "BAZ 50 (from K:FW)" ],
|
|
"baz-cpva16": [ "kfw-cpva", "BAZ 44 (from K:FW)" ],
|
|
"baz-cpva17": [ "kfw-cpva", "BAZ Type 51 (from K:FW)" ],
|
|
"piat": [ ["british","british~canadian"], check_piat_snippets ],
|
|
"thh": [ "japanese", "Banzai!!!" ],
|
|
}
|
|
btn_elems = {
|
|
btn: find_child( "button[data-id='{}']".format( btn ) )
|
|
for btn in nationality_specific_buttons
|
|
}
|
|
|
|
# iterate through each nationality
|
|
for nat in nationalities:
|
|
|
|
# change the nationality for player 1
|
|
set_player( 1, nat )
|
|
|
|
# check the nationality-specific buttons
|
|
select_tab( "ob1" )
|
|
for button_id,expected in nationality_specific_buttons.items():
|
|
elem = btn_elems[ button_id ]
|
|
if isinstance( expected[0], str ):
|
|
expected_nats = [ expected[0] ]
|
|
elif isinstance( expected[0], list ):
|
|
expected_nats = expected[0]
|
|
else:
|
|
assert isinstance( expected[0], tuple )
|
|
expected_nats = [ expected[0][0] ]
|
|
set_theater( expected[0][1] )
|
|
if nat in expected_nats:
|
|
# the button should be shown for this nationality
|
|
assert elem.is_displayed()
|
|
# make sure that the template works
|
|
elem.click()
|
|
if isinstance( expected[1], str ):
|
|
wait_for_clipboard( 2, expected[1] )
|
|
elif isinstance( expected[1], types.FunctionType ):
|
|
expected[1]( nat ) #pylint: disable=not-callable
|
|
else:
|
|
assert False
|
|
else:
|
|
# it should be hidden for all other nationalities
|
|
assert not elem.is_displayed()
|
|
|