def build_sweetiebot(config=None):
if config is None: import config
resource = config.nickname + randomstr()
if config.fake_redis:
log.debug('faking out redis')
redis_conn = FakeRedis()
else:
redis_conn = redis.from_url(config.redis_url)
jid = config.username + '/' + resource
nick = config.nickname
room = config.chatroom
password = config.password
if config.hostname is not None:
address = (config.hostname, config.port)
else:
address = ()
bot = MUCJabberBot(jid, password, room, nick, address)
crest = SweetieCrest(config.crest_base_url, config.crest_client_id,
config.crest_client_secret,
config.crest_refresh_token)
lookup = SweetieLookup(bot, crest)
admin = SweetieAdmin(bot, config.chatroom)
mq = SweetieMQ(config)
de = SweetieDe(bot, admin, mq, ResponsesFile('data/deowl_fails.txt'))
actions = ResponsesFile('data/actions.txt')
sass = ResponsesFile('data/sass.txt')
cadmusic = ResponsesFile('data/cadmusic.txt')
tell = SweetieTell(bot, redis_conn)
dictionary = SweetieDictionary(bot)
chat = SweetieChat(bot, actions, sass, config.chatroom, cadmusic, tell,
dictionary)
roulette = SweetieRoulette(bot, admin)
pings = SweetiePings(bot, redis_conn)
moon = SweetieMoon(bot)
if config.twitter_key is not None:
twitter = TwitterClient.get_client(config.twitter_key,
config.twitter_secret)
watchers = list(map(twitter.get_timeline_watcher, ['EVE_Status']))
else:
watchers = []
watchers.append(
AtomWatcher.get_watcher(
'http://eveion.blogspot.com/feeds/posts/default'))
seen = SweetieSeen(bot, redis_conn)
sweet = Sweetiebot(config.nickname, bot, lookup, mq, admin, chat, roulette,
de, pings, watchers, moon)
return sweet
def setUp(self):
self.bot = MagicMock()
self.crest = MagicMock()
self.lookup = SweetieLookup(self.bot, self.crest)
class LookupTests(unittest.TestCase):
def setUp(self):
self.bot = MagicMock()
self.crest = MagicMock()
self.lookup = SweetieLookup(self.bot, self.crest)
def test_can_roll_simple_dice(self):
message = create_message('!roll 1d20')
response = self.lookup.roll(message)
self.assertIsNotNone(int(response))
def test_wants_the_d(self):
response = self.lookup.roll(create_message('!roll butts'))
self.assertEqual(
"Dice need to be specified in the form 2d20 [>x] [<x] [!] [=] [+n]",
response)
def test_fails_nicely_with_unrecognised_dice(self):
response = self.lookup.roll(create_message('!roll 1dbutts'))
self.assertEqual("Sorry, don't know how to roll 'butts'", response)
def test_fails_nicely_with_bad_dice_count(self):
response = self.lookup.roll(create_message('!roll oned20'))
self.assertEqual("Sorry, don't know how to roll 'one' dice", response)
def test_can_parse_number_of_successes(self):
dice_spec = self.lookup.parse_dice('1d6>5')
self.assertIsNone(dice_spec.error)
self.assertEqual(6, dice_spec.sides)
self.assertEqual(5, dice_spec.threshold)
def test_can_count_number_of_successes(self):
response = self.lookup.roll(create_message('!roll 1d6>5'))
self.assertIn(" successes)", response)
def test_can_parse_sum(self):
dice_spec = self.lookup.parse_dice('1d6>5=')
self.assertTrue(dice_spec.show_sum)
def test_can_show_sum(self):
response = self.lookup.roll(create_message('!roll 1d6>5='))
self.assertIn("(sum ", response)
def test_can_parse_5d5_sum_ge5(self):
dice_spec = self.lookup.parse_dice('5d5=>5')
self.assertTrue(dice_spec.show_sum)
self.assertEqual(5, dice_spec.threshold)
def test_can_parse_empty_die_count(self):
dice_spec = self.lookup.parse_dice('d20')
self.assertEqual(1, dice_spec.dice)
self.assertEqual(20, dice_spec.sides)
def test_can_handle_spaces_in_dice_spec(self):
response = self.lookup.roll(create_message('!roll 1d6 > 5'))
self.assertIn(" successes)", response)
def test_can_parse_percent_die(self):
dice_spec = self.lookup.parse_dice('1d%')
self.assertEqual(100, dice_spec.sides)
def test_can_parse_lt_threshold(self):
dice_spec = self.lookup.parse_dice('2d20<5')
self.assertEqual(5, dice_spec.lt_threshold)
def test_can_parse_two_thresholds(self):
dice_spec = self.lookup.parse_dice('2d20<15>10')
self.assertEqual(15, dice_spec.lt_threshold)
self.assertEqual(10, dice_spec.threshold)
def test_complains_about_non_overlapping_thresholds(self):
response = self.lookup.roll(create_message('!roll 2d10>10<5'))
self.assertEqual(
'Requirements unsatisfactory: thresholds conflict. Try again.',
response)
def rolls(self, *rolls):
rolls = list(rolls)
def mock_function(self, dice=1, sides=6):
result = rolls.pop(0)
assert (len(result) == dice)
return result
return mock_function
def test_can_mock_out_get_rolls(self):
with patch.object(SweetieLookup, 'get_rolls',
self.rolls([1, 2, 3], [2, 4, 6])):
response = self.lookup.roll(create_message('!roll 3d3'))
self.assertEqual("1, 2, 3", response)
response = self.lookup.roll(create_message('!roll 3d3'))
self.assertEqual("2, 4, 6", response)
def test_can_add_bonus(self):
with patch.object(SweetieLookup, 'get_rolls', self.rolls([1, 2, 3])):
response = self.lookup.roll(create_message('!roll 3d3+10>12'))
self.assertEqual("11, 12, 13 (2 successes)", response)
def test_uses_both_thresholds_for_success_checks(self):
with patch.object(SweetieLookup, 'get_rolls', self.rolls([1, 2, 3])):
response = self.lookup.roll(create_message('!roll 3d3+10>12<12'))
self.assertEqual("11, 12, 13 (1 successes)", response)
def test_can_parse_exploding_dice(self):
dice_spec = self.lookup.parse_dice('10d6!')
self.assertTrue(dice_spec.explode)
def test_dice_can_explode(self):
"""exploding dice means that if a dice is rolled at the max value,
it is re-rolled and the value of that dice is extended with the
new roll value. This can happen multiple times per dice"""
with patch.object(SweetieLookup, 'get_rolls',
self.rolls([1, 1, 1, 6, 6, 6], [1, 1, 6], [1], [])):
response = self.lookup.roll(create_message('!roll 6d6!'))
self.assertEqual("1, 1, 1, 7, 7, 13", response)
def test_can_roll_some_shadowrun_dice(self):
with patch.object(SweetieLookup, 'get_rolls',
self.rolls([1, 1, 1, 6, 6, 6], [1, 1, 6], [1], [])):
response = self.lookup.roll(create_message('!roll 6d6!>5='))
self.assertEqual("1, 1, 1, 7, 7, 13 (3 successes) (sum 30)",
response)
def setUp(self):
self.bot = MagicMock()
self.crest = MagicMock()
self.lookup = SweetieLookup(self.bot, self.crest)
class LookupTests(unittest.TestCase):
def setUp(self):
self.bot = MagicMock()
self.crest = MagicMock()
self.lookup = SweetieLookup(self.bot, self.crest)
def test_can_roll_simple_dice(self):
message = create_message('!roll 1d20')
response = self.lookup.roll(message)
self.assertIsNotNone(int(response))
def test_wants_the_d(self):
response = self.lookup.roll(create_message('!roll butts'))
self.assertEqual("Dice need to be specified in the form 2d20", response)
def test_fails_nicely_with_unrecognised_dice(self):
response = self.lookup.roll(create_message('!roll 1dbutts'))
self.assertEqual("Sorry, don't know how to roll 'butts'", response)
def test_fails_nicely_with_bad_dice_count(self):
response = self.lookup.roll(create_message('!roll oned20'))
self.assertEqual("Sorry, don't know how to roll 'one' dice", response)
def test_can_parse_number_of_successes(self):
dice_spec = self.lookup.parse_dice('1d6>5')
self.assertIsNone(dice_spec.error)
self.assertEqual(6, dice_spec.sides)
self.assertEqual(5, dice_spec.threshold)
def test_can_count_number_of_successes(self):
response = self.lookup.roll(create_message('!roll 1d6>5'))
self.assertIn(" successes)", response)
def test_can_parse_sum(self):
dice_spec = self.lookup.parse_dice('1d6>5=')
self.assertTrue(dice_spec.show_sum)
def test_can_show_sum(self):
response = self.lookup.roll(create_message('!roll 1d6>5='))
self.assertIn("(sum ", response)
def test_can_parse_5d5_sum_ge5(self):
dice_spec = self.lookup.parse_dice('5d5=>5')
self.assertTrue(dice_spec.show_sum)
self.assertEqual(5, dice_spec.threshold)
def rolls(self, *rolls):
rolls = list(rolls)
def mock_function(self, dice=1, sides=6):
result = rolls.pop(0)
assert(len(result) == dice)
return result
return mock_function
def test_can_mock_out_get_rolls(self):
with patch.object(SweetieLookup, 'get_rolls',
self.rolls([1,2,3], [2, 4, 6])):
response = self.lookup.roll(create_message('!roll 3d3'))
self.assertEqual("1, 2, 3", response)
response = self.lookup.roll(create_message('!roll 3d3'))
self.assertEqual("2, 4, 6", response)
def test_can_parse_exploding_dice(self):
dice_spec = self.lookup.parse_dice('10d6!')
self.assertTrue(dice_spec.explode)
def test_dice_can_explode(self):
"""exploding dice means that if a dice is rolled at the max value,
it is re-rolled and the value of that dice is extended with the
new roll value. This can happen multiple times per dice"""
with patch.object(SweetieLookup, 'get_rolls',
self.rolls([1, 1, 1, 6, 6, 6], [1, 1, 6], [1], [])):
response = self.lookup.roll(create_message('!roll 6d6!'))
self.assertEqual("1, 1, 1, 7, 7, 13", response)
def test_can_roll_some_shadowrun_dice(self):
with patch.object(SweetieLookup, 'get_rolls',
self.rolls([1, 1, 1, 6, 6, 6], [1, 1, 6], [1], [])):
response = self.lookup.roll(create_message('!roll 6d6!>5='))
self.assertEqual("1, 1, 1, 7, 7, 13 (3 successes) (sum 30)", response)
class LookupTests(unittest.TestCase):
def setUp(self):
self.bot = MagicMock()
self.lookup = SweetieLookup(self.bot)
def test_can_roll_simple_dice(self):
message = create_message('!roll 1d20')
response = self.lookup.roll(message)
self.assertIsNotNone(int(response))
def test_wants_the_d(self):
response = self.lookup.roll(create_message('!roll butts'))
self.assertEqual("Dice need to be specified in the form 2d20 [>x] [<x] [!] [=] [+n]", response)
def test_fails_nicely_with_unrecognised_dice(self):
response = self.lookup.roll(create_message('!roll 1dbutts'))
self.assertEqual("Sorry, don't know how to roll 'butts'", response)
def test_fails_nicely_with_bad_dice_count(self):
response = self.lookup.roll(create_message('!roll oned20'))
self.assertEqual("Sorry, don't know how to roll 'one' dice", response)
def test_can_parse_number_of_successes(self):
dice_spec = self.lookup.parse_dice('1d6>5')
self.assertIsNone(dice_spec.error)
self.assertEqual(6, dice_spec.sides)
self.assertEqual(5, dice_spec.threshold)
def test_can_count_number_of_successes(self):
response = self.lookup.roll(create_message('!roll 1d6>5'))
self.assertIn(" successes)", response)
def test_can_parse_sum(self):
dice_spec = self.lookup.parse_dice('1d6>5=')
self.assertTrue(dice_spec.show_sum)
def test_can_show_sum(self):
response = self.lookup.roll(create_message('!roll 1d6>5='))
self.assertIn("(sum ", response)
def test_can_parse_5d5_sum_ge5(self):
dice_spec = self.lookup.parse_dice('5d5=>5')
self.assertTrue(dice_spec.show_sum)
self.assertEqual(5, dice_spec.threshold)
def test_can_parse_empty_die_count(self):
dice_spec = self.lookup.parse_dice('d20')
self.assertEqual(1, dice_spec.dice)
self.assertEqual(20, dice_spec.sides)
def test_can_handle_spaces_in_dice_spec(self):
response = self.lookup.roll(create_message('!roll 1d6 > 5'))
self.assertIn(" successes)", response)
def test_can_parse_percent_die(self):
dice_spec = self.lookup.parse_dice('1d%')
self.assertEqual(100, dice_spec.sides)
def test_can_parse_lt_threshold(self):
dice_spec = self.lookup.parse_dice('2d20<5')
self.assertEqual(5, dice_spec.lt_threshold)
def test_can_parse_two_thresholds(self):
dice_spec = self.lookup.parse_dice('2d20<15>10')
self.assertEqual(15, dice_spec.lt_threshold)
self.assertEqual(10, dice_spec.threshold)
def test_can_parse_bonuses(self):
self.assertEqual(+5, self.lookup.parse_dice("2d20+5").bonus)
self.assertEqual(-5, self.lookup.parse_dice("2d20-5").bonus)
self.assertEqual(+2, self.lookup.parse_dice("2d20-5+7").bonus)
def test_complains_about_non_overlapping_thresholds(self):
response = self.lookup.roll(create_message('!roll 2d10>10<5'))
self.assertEqual('Requirements unsatisfactory: thresholds conflict. Try again.', response)
def rolls(self, *rolls):
rolls = list(rolls)
def mock_function(self, dice=1, sides=6):
result = rolls.pop(0)
assert(len(result) == dice)
return result
return mock_function
def test_can_mock_out_get_rolls(self):
with patch.object(SweetieLookup, 'get_rolls',
self.rolls([1,2,3], [2, 4, 6])):
response = self.lookup.roll(create_message('!roll 3d3'))
self.assertEqual("1, 2, 3", response)
response = self.lookup.roll(create_message('!roll 3d3'))
self.assertEqual("2, 4, 6", response)
def test_can_add_bonus(self):
with patch.object(SweetieLookup, 'get_rolls',
self.rolls([1,2,3])):
response = self.lookup.roll(create_message('!roll 3d3+10>12'))
self.assertEqual("11, 12, 13 (2 successes)", response)
def test_uses_both_thresholds_for_success_checks(self):
with patch.object(SweetieLookup, 'get_rolls',
self.rolls([1,2,3])):
response = self.lookup.roll(create_message('!roll 3d3+10>12<12'))
self.assertEqual("11, 12, 13 (1 successes)", response)
def test_can_parse_exploding_dice(self):
dice_spec = self.lookup.parse_dice('10d6!')
self.assertTrue(dice_spec.explode)
def test_dice_can_explode(self):
"""exploding dice means that if a dice is rolled at the max value,
it is re-rolled and the value of that dice is extended with the
new roll value. This can happen multiple times per dice"""
with patch.object(SweetieLookup, 'get_rolls',
self.rolls([1, 1, 1, 6, 6, 6], [1, 1, 6], [1], [])):
response = self.lookup.roll(create_message('!roll 6d6!'))
self.assertEqual("1, 1, 1, 7, 7, 13", response)
def test_can_roll_some_shadowrun_dice(self):
with patch.object(SweetieLookup, 'get_rolls',
self.rolls([1, 1, 1, 6, 6, 6], [1, 1, 6], [1], [])):
response = self.lookup.roll(create_message('!roll 6d6!>5='))
self.assertEqual("1, 1, 1, 7, 7, 13 (3 successes) (sum 30)", response)
