def test_retrieve_user_inventory(self):
# Should make a new record if one does not exist
economy = Economy(MockReddit(), self.BASE_CONFIG)
user = MockUser()
sql = 'SELECT * FROM economy_users WHERE economy_users.reddit_id = ?'
record = Database().connection.cursor().execute(sql, (user.id, )).fetchone()
self.assertIsNone(record)
blank_record = economy.retrieve_user_inventory(user)
self.assertIsNotNone(blank_record)
def test_cmd_reload_funds(self):
# Should reload to the command setting
command_attributes = {
'reload_amount': 100,
'reload_threshold': 99
}
user = MockUser()
comment = MockComment(author=user)
economy = Economy(MockReddit(), self.BASE_CONFIG)
self.assertEqual(len(comment.replies), 0)
economy.cmd_reload_funds(comment, command_attributes)
inventory = economy.retrieve_user_inventory(user)
self.assertEqual(inventory['funds_available'], 100)
self.assertEqual(len(comment.replies), 1)
self.assertIn('funds were reloaded', comment.last_reply)
# Should not reload if over the threshold
economy.cmd_reload_funds(comment, command_attributes)
inventory = economy.retrieve_user_inventory(user)
self.assertEqual(inventory['funds_available'], 100)
self.assertEqual(len(comment.replies), 2)
self.assertIn("funds weren't reloaded", comment.last_reply)
def test_cmd_list_inventory(self):
# Should reply to original comment with inventory
self_add_command_attributes = {'text': '!add'}
user = MockUser()
comment = MockComment(author=user, body='!add')
economy = Economy(MockReddit(), self.BASE_CONFIG)
self.assertEqual(len(comment.replies), 0)
economy.cmd_list_inventory(comment)
self.assertEqual(len(comment.replies), 1)
self.assertIn('Items Available', comment.last_reply)
self.assertIn('Funds Available', comment.last_reply)
def test_cmd_add(self):
# Should add to comment user
self_add_command_attributes = {
'text': '!add'
}
parent_commenter = MockUser()
parent_comment = MockComment(author=parent_commenter)
comment = parent_comment.reply('!add')
user = comment.author
economy = Economy(MockReddit(), self.BASE_CONFIG)
inventory = economy.retrieve_user_inventory(user)
self.assertEqual(inventory['funds_available'], 0)
self.assertEqual(inventory['items_available'], 0)
self.assertEqual(len(comment.replies), 0)
economy.cmd_add(comment, self_add_command_attributes)
inventory = economy.retrieve_user_inventory(user)
self.assertEqual(inventory['funds_available'], 0)
self.assertEqual(inventory['items_available'], 1)
self.assertEqual(len(comment.replies), 1)
self.assertIn('inventory has increased by 1', comment.last_reply)
# Should add to comment parent user if 'award_to_parent' flag set
parent_add_command_attributes = {
'text': '!add',
'award_to_parent': True
}
inventory = economy.retrieve_user_inventory(parent_commenter)
self.assertEqual(inventory['funds_available'], 0)
self.assertEqual(inventory['items_available'], 0)
self.assertEqual(len(parent_comment.replies), 1)
economy.cmd_add(comment, parent_add_command_attributes)
inventory = economy.retrieve_user_inventory(parent_commenter)
self.assertEqual(inventory['funds_available'], 0)
self.assertEqual(inventory['items_available'], 1)
self.assertEqual(len(parent_comment.replies), 2)
self.assertIn('inventory has increased by 1', parent_comment.last_reply)
def test_cmd_buy(self):
# Should purchase if the user has the funds available
buy_command_attributes = {
'text': '!buy',
'item_price': 10
}
reload_command_attributes = {
'reload_amount': 100,
'reload_threshold': 1000000
}
user = MockUser()
comment = MockComment(author=user, body='!buy 10')
economy = Economy(MockReddit(), self.BASE_CONFIG)
self.assertEqual(len(comment.replies), 0)
economy.cmd_reload_funds(comment, reload_command_attributes)
inventory = economy.retrieve_user_inventory(user)
self.assertEqual(inventory['funds_available'], 100)
self.assertEqual(inventory['items_available'], 0)
self.assertEqual(len(comment.replies), 1)
economy.cmd_buy(comment, buy_command_attributes)
inventory = economy.retrieve_user_inventory(user)
self.assertEqual(inventory['funds_available'], 0)
self.assertEqual(inventory['items_available'], 10)
self.assertEqual(len(comment.replies), 2)
self.assertIn('inventory has increased', comment.last_reply)
# Should not allow purchase if no funds
economy.cmd_buy(comment, buy_command_attributes)
inventory = economy.retrieve_user_inventory(user)
self.assertEqual(inventory['funds_available'], 0)
self.assertEqual(inventory['items_available'], 10)
self.assertEqual(len(comment.replies), 3)
self.assertIn('transaction was not completed', comment.last_reply)
def test_db_creation(self):
# Should set up the DB if it doesn't exist already
self.assertFalse(os.path.exists(self.DB_PATH))
Economy(MockReddit(), self.BASE_CONFIG)
self.assertTrue(os.path.exists(self.DB_PATH))
def test_cmd_sell(self):
# Should sell if the user has the items available
buy_command_attributes = {
'text': '!buy',
'item_price': 0
}
sell_command_attributes = {
'text': '!sell',
'item_price': 10
}
user = MockUser()
buy_comment = MockComment(author=user, body='!buy 10')
sell_comment = MockComment(author=user, body='!sell 10')
economy = Economy(MockReddit(), self.BASE_CONFIG)
economy.cmd_buy(buy_comment, buy_command_attributes)
inventory = economy.retrieve_user_inventory(user)
self.assertEqual(inventory['funds_available'], 0)
self.assertEqual(inventory['items_available'], 10)
self.assertEqual(len(sell_comment.replies), 0)
economy.cmd_sell(sell_comment, sell_command_attributes)
inventory = economy.retrieve_user_inventory(user)
self.assertEqual(inventory['funds_available'], 100)
self.assertEqual(inventory['items_available'], 0)
self.assertEqual(len(sell_comment.replies), 1)
self.assertIn('inventory has decreased', sell_comment.last_reply)
# Should not allow sell if no items
economy.cmd_sell(sell_comment, sell_command_attributes)
inventory = economy.retrieve_user_inventory(user)
self.assertEqual(inventory['funds_available'], 100)
self.assertEqual(inventory['items_available'], 0)
self.assertEqual(len(sell_comment.replies), 2)
self.assertIn('transaction was not completed', sell_comment.last_reply)