adminsys/welcomebot
adminsys/welcomebot
9
0
Fork 0
Code Issues Pull Requests Releases Activity

Fixed a test involving writing nicks to csv file.

Browse Source
This commit is contained in:
Tarashish Mishra 2014-05-22 21:55:43 +05:30
parent ab2fe9c87f
commit 02b7e4e967
2 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
test_newbot.py
View File

@ -1,5 +1,6 @@
# Yay tests! # Yay tests!
import csv
import unittest import unittest
import newbot import newbot
import time import time
@ -39,11 +40,16 @@ class TestBotClass(unittest.TestCase):
def test_add_nick_to_csv(self): def test_add_nick_to_csv(self):
bot = newbot.Bot('test_nicks.csv') bot = newbot.Bot('test_nicks.csv')
bot.add_known_nick('Roger__') bot.add_known_nick('Roger__')
self.assertEqual(bot.known_nicks, [['Alice'], ['Bob'], ['Roger']]) with open('test_nicks.csv', 'rb') as csv_file:
known_nicks = []
csv_file_data = csv.reader(csv_file, delimiter=',', quotechar='|')
for row in csv_file_data:
known_nicks.append(row)
self.assertEqual(known_nicks, [['Alice'], ['Bob'], ['Roger']])
def tearDown(self): def tearDown(self):
with open('test_nicks.csv', 'w') as csv_file: with open('test_nicks.csv', 'w') as csv_file:
csv_file.write('Alice\nBob') csv_file.write('Alice\nBob\n')
class TestNewComerClass(unittest.TestCase): class TestNewComerClass(unittest.TestCase):
@ -56,15 +62,15 @@ class TestNewComerClass(unittest.TestCase):
def test_newcomer_init_born(self): def test_newcomer_init_born(self):
newComer = newbot.NewComer('Baby', newbot.Bot()) newComer = newbot.NewComer('Baby', newbot.Bot())
time.sleep(0.01) time.sleep(0.01)
self.assertAlmostEqual(newComer.born, time.time() - .01, places=2) self.assertAlmostEqual(newComer.born, time.time() - .01, places=2)
def test_add_newcomer_to_bot(self): def test_add_newcomer_to_bot(self):
pass pass
def test_newcomer_around_for(self): def test_newcomer_around_for(self):
newComer = newbot.NewComer('Shauna', newbot.Bot()) newComer = newbot.NewComer('Shauna', newbot.Bot())
time.sleep(0.01) time.sleep(0.01)
self.assertAlmostEqual(newComer.around_for(), .01, places=2) self.assertAlmostEqual(newComer.around_for(), .01, places=2)
# #
@ -72,18 +78,18 @@ class TestNewComerClass(unittest.TestCase):
# #
class TestProcessNewcomers(unittest.TestCase): class TestProcessNewcomers(unittest.TestCase):
def setUp(self): def setUp(self):
self.bot = newbot.Bot('test_nicks.csv', wait_time=.1) self.bot = newbot.Bot('test_nicks.csv', wait_time=.1)
newbot.NewComer('Harry', self.bot) newbot.NewComer('Harry', self.bot)
newbot.NewComer('Hermione', self.bot) newbot.NewComer('Hermione', self.bot)
time.sleep(.15) time.sleep(.15)
newbot.NewComer('Ron', self.bot) newbot.NewComer('Ron', self.bot)
def test_check_new_newcomers(self): def test_check_new_newcomers(self):
newbot.process_newcomers(self.bot, [i for i in self.bot.newcomers if i.around_for() > self.bot.wait_time], welcome=0) newbot.process_newcomers(self.bot, [i for i in self.bot.newcomers if i.around_for() > self.bot.wait_time], welcome=0)
self.assertEqual(len(self.bot.newcomers), 1) self.assertEqual(len(self.bot.newcomers), 1)
def test_check_new_known_nicks(self): def test_check_new_known_nicks(self):
newbot.process_newcomers(self.bot, [i for i in self.bot.newcomers if i.around_for() > self.bot.wait_time], welcome=0) newbot.process_newcomers(self.bot, [i for i in self.bot.newcomers if i.around_for() > self.bot.wait_time], welcome=0)
self.assertEqual(self.bot.known_nicks,[['Alice'],['Bob'],['Harry'],['Hermione']]) self.assertEqual(self.bot.known_nicks,[['Alice'],['Bob'],['Harry'],['Hermione']])
@ -92,7 +98,7 @@ class TestProcessNewcomers(unittest.TestCase):
def tearDown(self): def tearDown(self):
with open('test_nicks.csv', 'w') as csv_file: with open('test_nicks.csv', 'w') as csv_file:
csv_file.write('Alice\nBob') csv_file.write('Alice\nBob\n')
# #
# Not sure how to test check_messages. # Not sure how to test check_messages.

2
test_nicks.csv
View File

@ -1,2 +1,2 @@
Alice Alice
Bob Bob

1 Alice
2 Bob