mmgen-wallet/test/misc/oneshot_warning.py

56 lines
1 KiB
Python
Executable file

#!/usr/bin/env python3
from mmgen.common import *
cfg = Config()
class foo(oneshot_warning):
color = 'purple'
message = 'foo variant {} selected'
class bar(oneshot_warning):
color = 'yellow'
message = 'bar is experimental software'
class wg(oneshot_warning_group):
class foo:
color = 'yellow'
message = 'foo is experimental software. Proceed at your own risk'
class bar:
color = 'red'
message = 'The bar command is {} and can break your {}!!!'
class baz:
color = 'orange'
message = 'baz variant {} selected'
for i in (1,2,3):
from mmgen.crypto import Crypto
msg('\npw')
for k in ('A','B'):
ret = Crypto.pwfile_reuse_warning(k).warning_shown
assert ret == (i != 1), 'warning_shown incorrect'
msg('wg1')
wg('foo')
msg('wg2')
wg('bar',fmt_args=['dangerous','computer'])
msg('wg3')
wg('baz',div='alpha',fmt_args=['alpha'])
msg('wg4')
wg('baz',div='beta',fmt_args=['beta'])
msg('w1')
foo(div='alpha',fmt_args=['alpha'])
msg('w2')
foo(div='beta',fmt_args=['beta'])
msg('w3')
bar()
msg('bottom')