def test_dice_pipes(self):
dice = DiceSet('basic')
die0 = ['bike', 'moonandstars', 'bag', 'bird', 'crying', 'angry']
die1 = ['tulip', 'mouth', 'caravan', 'clock', 'whale', 'drink']
die2 = ['happy', 'coffee', 'plate', 'bus', 'letter', 'paws']
die3 = ['cat', 'pencil', 'baloon', 'bananas', 'phone', 'icecream']
#die4 = ['ladder', 'car', 'fire', 'bang', 'hat', 'hamburger']
#die5 = ['rain', 'heart', 'glasses', 'poo', 'ball', 'sun']
result = dice.throw_dice("4")
#print(result)
self.assertEqual(len(result), 4)
self.assertIn(result[0], die0)
self.assertIn(result[1], die1)
self.assertIn(result[2], die2)
self.assertIn(result[3], die3)
#self.assertIn(result[4], die4)
#self.assertIn(result[5], die5)
# wrong argument type
with self.assertRaises(WrongArgumentTypeError):
dice.throw_dice("pippo")
# wrong dice number
with self.assertRaises(WrongDiceNumberError):
dice.throw_dice("10")
def _roll(dicenumber, dicesetid):
form = StoryForm()
try:
dice = DiceSet(dicesetid)
except NonExistingSetError:
abort(404)
try:
roll = dice.throw_dice(dicenumber)
phrase = ""
for elem in roll:
phrase = phrase + elem + " "
except WrongDiceNumberError:
return _stories(
"<div class=\"alert alert-danger alert-dismissible fade show\">" +
"<button type=\"button\" class=\"close\" data-dismiss=\"alert\">×</button>"
+ "<strong>Error!</strong> Wrong dice number!</div>")
except WrongArgumentTypeError:
return _stories(
"<div class=\"alert alert-danger alert-dismissible fade show\">" +
"<button type=\"button\" class=\"close\" data-dismiss=\"alert\">×</button>"
+
"<strong>Error!</strong> Argument Dice number needs to be an integer!</div>"
)
return render_template("create_story.html",
form=form,
set=dicesetid,
roll=roll,
phrase=phrase)
def test_empty_story():
diceset = 'standard'
dicenum = 6
dice = DiceSet(diceset, dicenum)
roll = dice.throw_dice()
story = ''
with pytest.raises(NotValidStoryError):
_check_story(roll, story)
def test_correct_story():
diceset = 'standard'
dicenum = 6
dice = DiceSet(diceset, dicenum)
roll = dice.throw_dice()
story = ''
for word in roll:
story += word + ' '
_check_story(roll, story)
def test_throw_and_serialize_dice_set(self):
rnd.seed(574891)
die1 = Die(path + "die0.txt")
die2 = Die(path + "die1.txt")
die3 = Die(path + "die2.txt")
dice = [die1, die2, die3]
dice_set = DiceSet(dice)
# throw dice
expected_res = ['bag', 'clock', 'bus']
self.assertEqual(dice_set.throw_dice(), expected_res)
# serialize set
serialized_set = dice_set.serialize()
expected_serialized_set = json.dumps(dice_set.pips)
self.assertEqual(serialized_set, expected_serialized_set)
def _rollDice():
'''
Rolls the dice and enables the user to start writing a story.
The story is created as a draft, so that it can be edited.
Raises:
Exception: due to eventual failures during the commit of the
created story into the database.
Returns:
302 -> the user is redirected to the page in which he/she can
start writing the story on the faces which came out.
'''
diceset = request.args.get('diceset', 'standard')
dicenum = request.args.get('dicenum', 6, type=int)
try:
dice = DiceSet(diceset, dicenum)
roll = dice.throw_dice()
story = Story()
story.text = ''
story.theme = diceset
story.likes = 0
story.dislikes = 0
story.dice_set = roll
story.author_id = current_user.id
db.session.add(story)
db.session.commit()
except Exception as e:
print(e)
db.session.rollback()
abort(400)
return redirect(url_for('stories._story_edit', storyid=story.id))
def _roll_dice():
# check dice_number
try:
# get number of dice from the form of previous page
dice_number = int(request.form['dice_number'])
session['dice_number'] = dice_number
# check retrieved data
if dice_number not in range(2, 7):
raise ValueError
except BadRequestKeyError: # i'm here after re-rolling dice
dice_number = session['dice_number']
except (KeyError,
ValueError): # i'm here directly, have to go from settings before
flash('Invalid number of dice!', 'error')
session.pop('dice_number', None)
return redirect(url_for('dice._settings'))
# check dice_set
try:
# get dice set from the form of previous page
dice_img_set = str(request.form['dice_img_set'])
session['dice_img_set'] = dice_img_set
# check retrieved data
if dice_img_set not in {'standard', 'animal', 'halloween', 'emptyset'}:
raise ValueError
except BadRequestKeyError: # i'm here after re-rolling dice
dice_img_set = session['dice_img_set']
except (KeyError,
ValueError): # i'm here directly, have to go from settings before
flash('Invalid dice set!', 'error')
session.pop('dice_img_set', None)
return redirect(url_for('dice._settings'))
# random sampling dice and throw them
dice_indexes = rnd.sample(range(0, 6), dice_number)
dice_list = []
for i in dice_indexes:
try:
dirname = os.path.dirname(os.path.abspath(__file__))
path = dirname + "/../resources/" + dice_img_set + "/"
dice_list.append(Die(path + 'die' + str(i) + '.txt'))
except (FileNotFoundError, IndexError):
session.pop('dice_number', None)
session.pop('dice_img_set', None)
flash("Can't find dice on server", 'error')
return redirect(url_for('dice._settings'))
dice_set = DiceSet(dice_list)
dice_set.throw_dice()
session['figures'] = dice_set.pips
context_vars = {
'dice_number': dice_number,
'dice_img_set': dice_img_set,
'dice_indexes': dice_indexes,
'words': dice_set.pips,
'home_url': HOME_URL
}
return render_template('roll_dice.html', **context_vars)
def test_create_dice_set_with_size(self):
diceset = DiceSet('standard', 6)
thrown = diceset.throw_dice()
self.assertEqual(len(thrown), 6)
def test_correct_dices_thrown(self):
diceset = DiceSet('standard', 6)
thrown = diceset.throw_dice()
for i in range(6):
face_list = get_die_faces_list("standard", i)
self.assertTrue(thrown[i] in face_list)