class OperatorExtensionTestCase(TestCase):
def setUp(self):
self.oper = OperatorExtension(mock_bot())
def test_handle_message_exits_for_non_operator(self):
msg = Mock(response=None)
self.oper.is_operator = Mock(return_value=False)
self.oper.handle_message(None, msg)
assert msg.response in self.oper.nopes
def test_handle_message_does_nothing(self):
msg = Mock(response=None)
self.oper.is_operator = Mock(return_value=True)
self.oper.handle_message(
{
'autojoin': False,
'join': False,
'leave': False
}, msg)
assert msg.response is None
@patch('helga.extensions.operator.db')
def test_add_autojoin_exists(self, db):
db.autojoin.find.return_value = db
db.count.return_value = 1
ret = self.oper.add_autojoin('foo')
assert ret not in self.oper.acks
@patch('helga.extensions.operator.db')
def test_add_autojoin_adds(self, db):
db.autojoin.find.return_value = db
db.count.return_value = 0
self.oper.add_autojoin('foo')
db.autojoin.insert.assertCalledWith({'channel': 'foo'})
@patch('helga.extensions.operator.db')
def test_remove_autojoin(self, db):
self.oper.remove_autojoin('foo')
db.autojoin.remove.assertCalledWith({'channel': 'foo'})
class OperatorExtensionTestCase(TestCase):
def setUp(self):
self.oper = OperatorExtension(mock_bot())
def test_handle_message_exits_for_non_operator(self):
msg = Mock(response=None)
self.oper.is_operator = Mock(return_value=False)
self.oper.handle_message(None, msg)
assert msg.response in self.oper.nopes
def test_handle_message_does_nothing(self):
msg = Mock(response=None)
self.oper.is_operator = Mock(return_value=True)
self.oper.handle_message({"autojoin": False, "join": False, "leave": False}, msg)
assert msg.response is None
@patch("helga.extensions.operator.db")
def test_add_autojoin_exists(self, db):
db.autojoin.find.return_value = db
db.count.return_value = 1
ret = self.oper.add_autojoin("foo")
assert ret not in self.oper.acks
@patch("helga.extensions.operator.db")
def test_add_autojoin_adds(self, db):
db.autojoin.find.return_value = db
db.count.return_value = 0
self.oper.add_autojoin("foo")
db.autojoin.insert.assertCalledWith({"channel": "foo"})
@patch("helga.extensions.operator.db")
def test_remove_autojoin(self, db):
self.oper.remove_autojoin("foo")
db.autojoin.remove.assertCalledWith({"channel": "foo"})
def setUp(self):
self.oper = OperatorExtension(mock_bot())
def setUp(self):
self.oper = OperatorExtension(mock_bot())
