def test_iter(db):
"""Test simple files use-case."""
fl = Sequence(Template.create('FL', 'File {counter}'))
assert list(islice(fl, 10)) == ['File {}'.format(c) for c in range(0, 10)]
y = Sequence(Template.create('Y', '{year}: {counter}'), year=2016)
assert list(islice(y, 10)) == ['2016: {}'.format(c) for c in range(0, 10)]
def test_iter(db):
"""Test simple files use-case."""
fl = Sequence(Template.create('FL', 'File {counter}'))
assert list(islice(fl, 10)) == ['File {}'.format(c) for c in range(0, 10)]
y = Sequence(Template.create('Y', '{year}: {counter}'), year=2016)
assert list(islice(y, 10)) == ['2016: {}'.format(c) for c in range(0, 10)]
def templates(app, db):
"""Register CDS templates for sequence generation."""
Template.create(name='project-v1_0_0',
meta_template='{category}-{type}-{year}-{counter}',
start=1)
Template.create(name='video-v1_0_0',
meta_template='{project-v1_0_0}-{counter}',
start=1)
db.session.commit()
def sequence_generator():
"""Register CDS templates for sequence generation."""
with db.session.begin_nested():
Template.create(name='project-v1_0_0',
meta_template='{category}-{type}-{year}-{counter:03d}',
start=1)
Template.create(name='video-v1_0_0',
meta_template='{project-v1_0_0}-{counter:03d}',
start=1)
db.session.commit()
click.echo('DONE :)')
def test_get(db):
"""Test get sequence."""
Template.create('A', 'A-{counter}')
Template.create('B', 'B-{counter}')
assert Sequence('A')
assert Sequence('A').next() == 'A-0'
assert Sequence('A').next() == 'A-1'
assert Sequence('B')
assert Sequence('B').next() == 'B-0'
assert Sequence('B').next() == 'B-1'
def test_get(db):
"""Test get sequence."""
Template.create('A', 'A-{counter}')
Template.create('B', 'B-{counter}')
assert Sequence('A')
assert Sequence('A').next() == 'A-0'
assert Sequence('A').next() == 'A-1'
assert Sequence('B')
assert Sequence('B').next() == 'B-0'
assert Sequence('B').next() == 'B-1'
def test_exceptions(db):
"""Test exceptions."""
p1 = Sequence(Template.create('P1', 'P-{counter}'))
assert p1.next() == 'P-0'
Template.create('P2', 'P:{counter}')
with pytest.raises(InvalidTemplate) as exc:
Template.create('invalid', '{P1}-{P2}-{counter}')
assert 'More than 1 parents in template' in str(exc)
with pytest.raises(InvalidTemplate) as exc:
Template.create('invalid', 'INVALID')
assert 'No counter placeholder' in str(exc)
assert Sequence('P1')
with pytest.raises(SequenceNotFound):
Sequence('invalid')
seq = Template.create('C1', '{P1}:{counter}')
seq1 = Sequence(seq, P1='P-0')
assert seq1.next() == 'P-0:0'
with pytest.raises(KeyError) as exc:
Sequence(seq).next()
assert 'P1' in str(exc)
with pytest.raises(InvalidResetCall) as exc:
Sequence('P1').counter.reset()
assert 'Cannot reset sequence: children exist' in str(exc)
def test_exceptions(db):
"""Test exceptions."""
p1 = Sequence(Template.create('P1', 'P-{counter}'))
assert p1.next() == 'P-0'
Template.create('P2', 'P:{counter}')
with pytest.raises(InvalidTemplate) as exc:
Template.create('invalid', '{P1}-{P2}-{counter}')
assert 'More than 1 parents in template' in str(exc)
with pytest.raises(InvalidTemplate) as exc:
Template.create('invalid', 'INVALID')
assert 'No counter placeholder' in str(exc)
assert Sequence('P1')
with pytest.raises(SequenceNotFound):
Sequence('invalid')
seq = Template.create('C1', '{P1}:{counter}')
seq1 = Sequence(seq, P1='P-0')
assert seq1.next() == 'P-0:0'
with pytest.raises(KeyError) as exc:
Sequence(seq).next()
assert 'P1' in str(exc)
with pytest.raises(InvalidResetCall) as exc:
Sequence('P1').counter.reset()
assert 'Cannot reset sequence: children exist' in str(exc)
def test_formatting(db):
"""Test counter formatting options."""
seq1 = Sequence(Template.create('P', 'P-{counter:03d}'))
assert seq1.next() == 'P-000'
assert seq1.next() == 'P-001'
assert seq1.next() == 'P-002'
seq2 = Template.create('C', '{P}-{counter:04d}')
p1 = Sequence(seq2, P='P-000')
assert p1.next() == 'P-000-0000'
assert p1.next() == 'P-000-0001'
assert p1.next() == 'P-000-0002'
p2 = Sequence(seq2, P='P-001')
assert p2.next() == 'P-001-0000'
assert p2.next() == 'P-001-0001'
def test_formatting(db):
"""Test counter formatting options."""
seq1 = Sequence(Template.create('P', 'P-{counter:03d}'))
assert seq1.next() == 'P-000'
assert seq1.next() == 'P-001'
assert seq1.next() == 'P-002'
seq2 = Template.create('C', '{P}-{counter:04d}')
p1 = Sequence(seq2, P='P-000')
assert p1.next() == 'P-000-0000'
assert p1.next() == 'P-000-0001'
assert p1.next() == 'P-000-0002'
p2 = Sequence(seq2, P='P-001')
assert p2.next() == 'P-001-0000'
assert p2.next() == 'P-001-0001'
def sequences():
"""Load test data fixture."""
pl = Template.create('PL', '{year}: Playlist {counter}', start=1)
fl = Template.create('FL', '{PL} > Audio File {counter:02d}', start=1)
pl15 = Sequence(pl, year=2015)
pl15.next()
pl15.next()
pl16 = Sequence(pl, year=2016)
pl16.next()
fl15 = Sequence(fl, PL='2015: Playlist 2')
fl15.next()
fl15.next()
fl16 = Sequence(fl, PL='2016: Playlist 1')
fl16.next()
db.session.commit()
def sequences():
"""Load test data fixture."""
pl = Template.create('PL', '{year}: Playlist {counter}', start=1)
fl = Template.create('FL', '{PL} > Audio File {counter:02d}', start=1)
pl15 = Sequence(pl, year=2015)
pl15.next()
pl15.next()
pl16 = Sequence(pl, year=2016)
pl16.next()
fl15 = Sequence(fl, PL='2015: Playlist 2')
fl15.next()
fl15.next()
fl16 = Sequence(fl, PL='2016: Playlist 1')
fl16.next()
db.session.commit()
def test_delete(db):
"""Test delete."""
fl = Template.create('FL', '{year}: File {counter}')
fl15 = Sequence(fl, year=2015)
assert list(islice(fl15, 20))
assert fl15.next() == '2015: File 20'
fl16 = Sequence(fl, year=2016)
assert list(islice(fl16, 30))
assert fl16.next() == '2016: File 30'
fl17 = Sequence(fl, year=2017)
assert fl17.next() == '2017: File 0'
assert db.session.query(Counter).count() == 3
db.session.delete(fl17.counter)
assert db.session.query(Counter).count() == 2
db.session.delete(fl.model)
db.session.commit()
with pytest.raises(SequenceNotFound):
Sequence('FL', year=2016)
assert db.session.query(Counter).count() == 0
def test_admin(app, db):
"""Test admin interface."""
assert isinstance(ca, dict)
assert isinstance(ta, dict)
assert 'model' in ca
assert 'modelview' in ca
assert 'model' in ta
assert 'modelview' in ta
# Create admin
admin = Admin(app, name='Example: Sequence Generator')
# Add views
admin.add_view(ta['modelview'](ta['model'], db.session))
admin.add_view(ca['modelview'](ca['model'], db.session))
with app.app_context():
# Create test data
seq = Sequence(Template.create('ID', 'File {counter}'))
assert seq.next() == 'File 0'
assert seq.next() == 'File 1'
assert seq.next() == 'File 2'
db.session.commit()
with app.test_request_context():
request_url = url_for('counter.reset_view')
with app.test_client() as client:
# Reset counter
client.post(request_url,
data={'start': 0, 'rowid': 'File {counter}'},
follow_redirects=False)
# Assert that reset was successful
assert seq.next() == 'File 0'
def test_delete(db):
"""Test delete."""
fl = Template.create('FL', '{year}: File {counter}')
fl15 = Sequence(fl, year=2015)
assert list(islice(fl15, 20))
assert fl15.next() == '2015: File 20'
fl16 = Sequence(fl, year=2016)
assert list(islice(fl16, 30))
assert fl16.next() == '2016: File 30'
fl17 = Sequence(fl, year=2017)
assert fl17.next() == '2017: File 0'
assert db.session.query(Counter).count() == 3
db.session.delete(fl17.counter)
assert db.session.query(Counter).count() == 2
db.session.delete(fl.model)
db.session.commit()
with pytest.raises(SequenceNotFound):
Sequence('FL', year=2016)
assert db.session.query(Counter).count() == 0
def test_cern_playlists(db):
"""Test CERN playlists use-case."""
pl = Template.create('PL', 'C-{year}-{counter}')
fl = Template.create('FL', '{PL}:{counter}')
pl15 = Sequence(pl, year=2015)
pl16 = Sequence(pl, year=2016)
assert pl15.next() == 'C-2015-0'
assert pl15.next() == 'C-2015-1'
assert pl15.next() == 'C-2015-2'
assert pl16.next() == 'C-2016-0'
fl_15_2 = Sequence(fl, PL='C-2015-1')
fl_16_1 = Sequence(fl, PL='C-2016-0')
assert fl_15_2.next() == 'C-2015-1:0'
assert fl_15_2.next() == 'C-2015-1:1'
assert fl_15_2.next() == 'C-2015-1:2'
assert fl_16_1.next() == 'C-2016-0:0'
def test_cern_playlists(db):
"""Test CERN playlists use-case."""
pl = Template.create('PL', 'C-{year}-{counter}')
fl = Template.create('FL', '{PL}:{counter}')
pl15 = Sequence(pl, year=2015)
pl16 = Sequence(pl, year=2016)
assert pl15.next() == 'C-2015-0'
assert pl15.next() == 'C-2015-1'
assert pl15.next() == 'C-2015-2'
assert pl16.next() == 'C-2016-0'
fl_15_2 = Sequence(fl, PL='C-2015-1')
fl_16_1 = Sequence(fl, PL='C-2016-0')
assert fl_15_2.next() == 'C-2015-1:0'
assert fl_15_2.next() == 'C-2015-1:1'
assert fl_15_2.next() == 'C-2015-1:2'
assert fl_16_1.next() == 'C-2016-0:0'
def test_repr(db):
"""Test __repr__."""
template = Template.create('A', '{counter}')
assert repr(
template.model) == ("TemplateDefinition(name='A', "
"meta_template='{counter}', start=0, step=1)")
sequence = Sequence(template)
assert repr(sequence.counter) == (
"Counter(template_instance='{counter}', "
"definition_name='A', counter=0, template_definition="
"TemplateDefinition(name='A', meta_template='{counter}', "
"start=0, step=1))")
def test_step(db):
"""Test larger incremental steps."""
fl = Template.create('FL', 'File {counter}', start=0, step=100)
seq = Sequence(fl)
assert seq.next() == 'File 0'
assert seq.next() == 'File 100'
assert seq.next() == 'File 200'
assert seq.next() == 'File 300'
seq.counter.reset(start=15)
assert seq.next() == 'File 15'
assert seq.next() == 'File 115'
assert seq.next() == 'File 215'
assert seq.next() == 'File 315'
def test_repr(db):
"""Test __repr__."""
template = Template.create('A', '{counter}')
assert repr(template.model) == (
"TemplateDefinition(name='A', "
"meta_template='{counter}', start=0, step=1)"
)
sequence = Sequence(template)
assert repr(sequence.counter) == (
"Counter(template_instance='{counter}', "
"definition_name='A', counter=0, template_definition="
"TemplateDefinition(name='A', meta_template='{counter}', "
"start=0, step=1))"
)
def test_step(db):
"""Test larger incremental steps."""
fl = Template.create('FL', 'File {counter}', start=0, step=100)
seq = Sequence(fl)
assert seq.next() == 'File 0'
assert seq.next() == 'File 100'
assert seq.next() == 'File 200'
assert seq.next() == 'File 300'
seq.counter.reset(start=15)
assert seq.next() == 'File 15'
assert seq.next() == 'File 115'
assert seq.next() == 'File 215'
assert seq.next() == 'File 315'
def test_inspire_authors(db):
"""Test INSPIRE authors use-case."""
auth = Template.create('AUTH', '{author}.{counter}')
john = Sequence(auth, author='John.Tester')
tom = Sequence(auth, author='Tom.Phake')
assert john.next() == 'John.Tester.0'
assert john.next() == 'John.Tester.1'
assert john.next() == 'John.Tester.2'
assert tom.next() == 'Tom.Phake.0'
assert tom.next() == 'Tom.Phake.1'
assert john.next() == 'John.Tester.3'
assert tom.next() == 'Tom.Phake.2'
assert tom.next() == 'Tom.Phake.3'
def test_inspire_authors(db):
"""Test INSPIRE authors use-case."""
auth = Template.create('AUTH', '{author}.{counter}')
john = Sequence(auth, author='John.Tester')
tom = Sequence(auth, author='Tom.Phake')
assert john.next() == 'John.Tester.0'
assert john.next() == 'John.Tester.1'
assert john.next() == 'John.Tester.2'
assert tom.next() == 'Tom.Phake.0'
assert tom.next() == 'Tom.Phake.1'
assert john.next() == 'John.Tester.3'
assert tom.next() == 'Tom.Phake.2'
assert tom.next() == 'Tom.Phake.3'
def test_cern_articles(db):
"""Test CERN articles use-case."""
with pytest.raises(SequenceNotFound):
Sequence('ART')
articles = Template.create('ART', 'CERN-ARTICLE-{category}-{counter}')
journals = Sequence(articles, category='JOURNALS')
theses = Sequence(articles, category='THESES')
assert journals.next() == 'CERN-ARTICLE-JOURNALS-0'
assert journals.next() == 'CERN-ARTICLE-JOURNALS-1'
assert journals.next() == 'CERN-ARTICLE-JOURNALS-2'
assert theses.next() == 'CERN-ARTICLE-THESES-0'
assert theses.next() == 'CERN-ARTICLE-THESES-1'
assert journals.next() == 'CERN-ARTICLE-JOURNALS-3'
assert theses.next() == 'CERN-ARTICLE-THESES-2'
def test_cern_articles(db):
"""Test CERN articles use-case."""
with pytest.raises(SequenceNotFound):
Sequence('ART')
articles = Template.create('ART', 'CERN-ARTICLE-{category}-{counter}')
journals = Sequence(articles, category='JOURNALS')
theses = Sequence(articles, category='THESES')
assert journals.next() == 'CERN-ARTICLE-JOURNALS-0'
assert journals.next() == 'CERN-ARTICLE-JOURNALS-1'
assert journals.next() == 'CERN-ARTICLE-JOURNALS-2'
assert theses.next() == 'CERN-ARTICLE-THESES-0'
assert theses.next() == 'CERN-ARTICLE-THESES-1'
assert journals.next() == 'CERN-ARTICLE-JOURNALS-3'
assert theses.next() == 'CERN-ARTICLE-THESES-2'
def test_reset(db):
"""Test reset sequence."""
y = Template.create('FL', '{year}: File {counter}')
y15 = Sequence(y, year=2015)
y16 = Sequence(y, year=2016)
assert y15.next() == '2015: File 0'
assert y15.next() == '2015: File 1'
assert y16.next() == '2016: File 0'
assert y16.next() == '2016: File 1'
y15.counter.reset()
assert y15.next() == '2015: File 0'
assert y15.next() == '2015: File 1'
y16.counter.reset(start=156)
assert y16.next() == '2016: File 156'
assert y16.next() == '2016: File 157'
def test_reset(db):
"""Test reset sequence."""
y = Template.create('FL', '{year}: File {counter}')
y15 = Sequence(y, year=2015)
y16 = Sequence(y, year=2016)
assert y15.next() == '2015: File 0'
assert y15.next() == '2015: File 1'
assert y16.next() == '2016: File 0'
assert y16.next() == '2016: File 1'
y15.counter.reset()
assert y15.next() == '2015: File 0'
assert y15.next() == '2015: File 1'
y16.counter.reset(start=156)
assert y16.next() == '2016: File 156'
assert y16.next() == '2016: File 157'
def test_admin(app, db):
"""Test admin interface."""
assert isinstance(ca, dict)
assert isinstance(ta, dict)
assert 'model' in ca
assert 'modelview' in ca
assert 'model' in ta
assert 'modelview' in ta
# Create admin
admin = Admin(app, name='Example: Sequence Generator')
# Add views
admin.add_view(ta['modelview'](ta['model'], db.session))
admin.add_view(ca['modelview'](ca['model'], db.session))
with app.app_context():
# Create test data
seq = Sequence(Template.create('ID', 'File {counter}'))
assert seq.next() == 'File 0'
assert seq.next() == 'File 1'
assert seq.next() == 'File 2'
db.session.commit()
with app.test_request_context():
request_url = url_for('counter.reset_view')
with app.test_client() as client:
# Reset counter
client.post(request_url,
data={
'start': 0,
'rowid': 'File {counter}'
},
follow_redirects=False)
# Assert that reset was successful
assert seq.next() == 'File 0'
def create(template_name, meta_template, start, step):
"""Create a new template definition."""
Template.create(template_name, meta_template, start, step)
db.session.commit()
click.secho('Template "{0}" created successfully.'.format(template_name),
fg='green')
def test_py3_next(db):
"""Test PY3 __next__."""
assert Sequence(Template.create('PY3', '{counter}')).__next__() == '0'
def test_py3_next(db):
"""Test PY3 __next__."""
assert Sequence(Template.create('PY3', '{counter}')).__next__() == '0'