def sequence_episode(episode):
try:
return Sequence.find_one(episode=episode, name='102')
except NoResultFound:
return Sequence.create(project=episode.project,
episode=episode,
name='102')
def test_create_unique_proj_name(sequence):
# Expecting IntegrityError error "Duplicate entry..."
try:
Sequence.create(name=sequence.name, project=sequence.project)
except IntegrityError:
return
raise AssertionError('Expected IntegrityError due to "Duplicate entry"')
def sequence(project):
try:
return Sequence.find_one(project=project, name='101')
except NoResultFound:
return Sequence.create(project=project, name='101')
def test_cls_name():
assert Sequence.cls_name() == 'Sequence'
def test_findby_name_ep(sequence_episode):
assert sequence_episode == Sequence.find_one(
project=sequence_episode.project,
episode=sequence_episode.episode,
name=sequence_episode.name)
def test_find_one(sequence):
assert sequence == Sequence.find_one(id=sequence.id)
def test_findby_ids(sequence):
assert sequence in Sequence.findby_ids([sequence.id])
def test_find(sequence):
assert sequence in Sequence.find()