def setUp(self, stats_module_mock, stranger_module_mock, talk_module_mock):
from randtalkbot.db import RetryingDB
self.stats_module_mock = stats_module_mock
self.stranger_module_mock = stranger_module_mock
self.talk_module_mock = talk_module_mock
self.database = Mock()
self.RetryingDB = RetryingDB
self.RetryingDB.return_value = self.database
self.configuration = Mock()
self.configuration.database_host = 'foo_host'
self.configuration.database_name = 'foo_name'
self.configuration.database_user = '******'
self.configuration.database_password = '******'
self.RetryingDB.reset_mock()
self.db = DB(self.configuration)
def test_init__database_troubles(self):
self.RetryingDB.return_value.connect.side_effect = DatabaseError()
with self.assertRaises(DBError):
DB(self.configuration)
