def test_log_okay_rolls(self):
# function should add one to random value
# so remove 1 from secrets.randbelow arg
with mock.patch('secrets.randbelow') as m:
m.side_effect = [7, 6, 8, 7]
egg_level = CryptRollLogger.EGG_LEVEL
roller = CryptRoll(egg_level)
with self.assertLogs(roller.log, level=egg_level) as cm:
output = roller.get_roll_result([10, 10, 10, 10])
self.assertTrue(any("not bad" in ll.lower() for ll in cm.output))
def test_log_nat_17(self):
# function should add one to random value
# so remove 1 from secrets.randbelow arg
with mock.patch('secrets.randbelow') as m:
m.return_value = 16
egg_level = CryptRollLogger.EGG_LEVEL
roller = CryptRoll(egg_level)
with self.assertLogs(roller.log, level=egg_level) as cm:
output = roller.get_roll_result([20])
self.assertFalse(any("Hey, natural" in ll for ll in cm.output))
def test_get_roll_result_simple_d20_with_disadvantage_lower_second(self):
dice_type = 20
expected_roll = 8
with mock.patch('secrets.randbelow') as m:
# function should add one to random value
m.side_effect = [17, 7]
output = CryptRoll(logging.DEBUG).get_roll_result(
[dice_type], roll_with="disadvantage")
self.assertEqual(output, expected_roll)
def test_get_roll_result_simple_d6_plus_modifier(self):
dice_type = 6
modifier = 4
expected_roll = 2
with mock.patch('secrets.randbelow') as m:
# function should add one to random value
m.return_value = expected_roll - 1
output = CryptRoll(logging.DEBUG).get_roll_result([dice_type],
modifier)
self.assertEqual(output, 6)
def test_roll_d8(self):
expected_value = 7
with mock.patch('secrets.randbelow') as m:
# function should add one to random value
m.return_value = expected_value - 1
output = CryptRoll(logging.DEBUG).roll_dice([8])
self.assertTrue(m.called)
self.assertEqual(m.call_count, 1)
self.assertEqual(m.call_args, mock.call(8))
self.assertEqual(output, [expected_value])
def test_get_roll_result_multiple(self):
dice_type = 10
expected_values = [2, 4, 6, 8]
# function should add one to random value
# so remove 1 from secrets.randbelow arg
rand_results = [val - 1 for val in expected_values]
with mock.patch('secrets.randbelow') as m:
m.side_effect = rand_results
output = CryptRoll(logging.DEBUG).get_roll_result(4 * [dice_type])
self.assertEqual(output, 20)
def test_get_roll_result_simple_d6(self):
dice_type = 6
expected_value = 2
with mock.patch('secrets.randbelow') as m:
# function should add one to random value
m.return_value = expected_value - 1
output = CryptRoll(logging.DEBUG).get_roll_result([dice_type])
self.assertTrue(m.called)
self.assertEqual(m.call_count, 1)
self.assertEqual(m.call_args, mock.call(dice_type))
self.assertEqual(output, expected_value)
def test_roll_multiple_dice_4d10_args(self):
expected_values = [2, 4, 6, 8]
# function should add one to random value
# so remove 1 from secrets.randbelow arg
rand_results = [val - 1 for val in expected_values]
with mock.patch('secrets.randbelow') as m:
m.side_effect = rand_results
output = CryptRoll(logging.DEBUG).roll_dice([10, 10, 10, 10])
self.assertTrue(m.called)
self.assertEqual(m.call_count, 4)
expected_calls = [mock.call(10)] * 4
m.assert_has_calls(expected_calls, any_order=True)
self.assertEqual(output, expected_values)
def test_simple_d6(self):
result = CryptRoll(logging.DEBUG).parse_roll('d6')
self.assertEqual(result, ([6], None))
def test_multiple_dict_3d6_modifier_plus_2(self):
result = CryptRoll(logging.DEBUG).parse_roll('3d6+2')
self.assertEqual(result, ([6, 6, 6], 2))
def test_simple_d6_modifier_minus_7(self):
result = CryptRoll(logging.DEBUG).parse_roll('d6-7')
self.assertEqual(result, ([6], -7))
def test_simple_d6_modifier_plus_4(self):
result = CryptRoll(logging.DEBUG).parse_roll('d6+4')
self.assertEqual(result, ([6], 4))
def test_multiple_dice_4d10(self):
result = CryptRoll(logging.DEBUG).parse_roll('4d10')
self.assertEqual(result, ([10, 10, 10, 10], None))
def test_multiple_dice_d6(self):
result = CryptRoll(logging.DEBUG).parse_roll('2d6')
self.assertEqual(result, ([6, 6], None))