def test_invite_2_user(self):
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
serv1.type = Server.TYPE_IRC
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
chan1.in_channel = True
chan2 = serv1.get_channel_by_name("test_chan2")
chan2.in_channel = True
user1 = serv1.get_user_by_name("test_user1")
user2 = serv1.get_user_by_name("test_user2")
user_hallo = serv1.get_user_by_name(serv1.get_nick())
chan1.add_user(user1)
chan1_user1 = chan1.get_membership_by_user(user1)
chan1_user1.is_op = False
chan1.add_user(user_hallo)
chan1_hallo = chan1.get_membership_by_user(user_hallo)
chan1_hallo.is_op = True
chan2.add_user(user_hallo)
chan2_hallo = chan2.get_membership_by_user(user_hallo)
chan2_hallo.is_op = True
try:
self.function_dispatcher.dispatch("invite test_user2 test_chan2", user1, chan1)
data = serv1.get_send_data(2)
assert "error" not in data[0][0].lower()
assert data[0][1] is None
assert data[1][1] == chan1
assert data[0][2] == Server.MSG_RAW
assert data[1][2] == Server.MSG_MSG
assert data[0][0][:6] == "INVITE"
assert user2.name+" "+chan2.name in data[0][0]
assert "invite sent" in data[1][0].lower()
finally:
self.hallo.remove_server(serv1)
def test_run_by_title(self):
# Set up test servers and channels
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
chan1 = serv1.get_channel_by_name("test_chan1")
chan2 = serv1.get_channel_by_name("test_chan2")
serv2 = ServerMock(self.hallo)
serv2.name = "test_serv2"
chan3 = serv2.get_channel_by_name("test_chan1")
try:
self.hallo.add_server(serv1)
self.hallo.add_server(serv2)
# Set up rss feeds
rfl = RssFeedList()
rf1 = RssFeed()
rf1.url = "http://spangle.org.uk/hallo/test_rss.xml?1"
rf1.title = "test_feed1"
rf1.server_name = chan1.server.name
rf1.channel_name = chan1.name
rf1.update_frequency = Commons.load_time_delta("PT3600S")
rfl.add_feed(rf1)
rf2 = RssFeed()
rf2.url = "http://spangle.org.uk/hallo/test_rss.xml?2"
rf2.title = "test_feed2"
rf2.server_name = chan2.server.name
rf2.channel_name = chan2.name
rf2.update_frequency = Commons.load_time_delta("PT3600S")
rfl.add_feed(rf2)
rf3 = RssFeed()
rf3.url = "http://spangle.org.uk/hallo/test_rss.xml?3"
rf3.title = "test_feed1"
rf3.server_name = chan3.server.name
rf3.channel_name = chan3.name
rf3.update_frequency = Commons.load_time_delta("PT3600S")
rfl.add_feed(rf3)
# Splice this rss feed list into the function dispatcher's rss check object
rss_check_class = self.function_dispatcher.get_function_by_name("rss check")
rss_check_obj = self.function_dispatcher.get_function_object(rss_check_class) # type: FeedCheck
rss_check_obj.rss_feed_list = rfl
# Invalid title
self.function_dispatcher.dispatch("rss check Not a valid feed", self.test_user, self.test_chan)
data = self.server.get_send_data(1, self.test_chan, Server.MSG_MSG)
assert "error" in data[0][0].lower()
# Correct title but wrong channel
self.function_dispatcher.dispatch("rss check test_feed2", self.test_user, chan1)
data = serv1.get_send_data(1, chan1, Server.MSG_MSG)
assert "error" in data[0][0].lower()
# Correct title check update
self.function_dispatcher.dispatch("rss check test_feed2", self.test_user, chan2)
data = serv1.get_send_data(1, chan2, Server.MSG_MSG)
assert "feed updates were found" in data[0][0].lower()
assert len(data[0][0].lower().split("\n")) == 4
# No updates
rf2.title = "test_feed2"
self.function_dispatcher.dispatch("rss check test_feed2", self.test_user, chan2)
data = serv1.get_send_data(1, chan2, Server.MSG_MSG)
assert "no updates" in data[0][0], "No further updates should be found."
finally:
self.hallo.remove_server(serv2)
self.hallo.remove_server(serv1)
def test_invite_1_chan_user_already_there(self):
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
serv1.type = Server.TYPE_IRC
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
chan1.in_channel = True
chan2 = serv1.get_channel_by_name("test_chan2")
chan2.in_channel = True
user1 = serv1.get_user_by_name("test_user1")
user2 = serv1.get_user_by_name("test_user2")
user_hallo = serv1.get_user_by_name(serv1.get_nick())
chan1.add_user(user1)
chan1.add_user(user_hallo)
chan2.add_user(user1)
chan2.add_user(user_hallo)
chan1_user1 = chan1.get_membership_by_user(user1)
chan1_user1.is_op = False
chan1_hallo = chan1.get_membership_by_user(user_hallo)
chan1_hallo.is_op = True
chan2_user1 = chan2.get_membership_by_user(user1)
chan2_user1.is_op = False
chan2_hallo = chan2.get_membership_by_user(user_hallo)
chan2_hallo.is_op = True
try:
self.function_dispatcher.dispatch("invite test_chan2", user1, chan1)
data = serv1.get_send_data(1, chan1, Server.MSG_MSG)
assert "error" in data[0][0].lower()
assert "test_user1 is already in test_chan2" in data[0][0].lower()
finally:
self.hallo.remove_server(serv1)
def test_deop_2_user_not_op(self):
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
serv1.type = Server.TYPE_IRC
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
chan1.in_channel = True
chan2 = serv1.get_channel_by_name("test_chan2")
chan2.in_channel = True
user1 = serv1.get_user_by_name("test_user1")
user2 = serv1.get_user_by_name("test_user2")
user_hallo = serv1.get_user_by_name(serv1.get_nick())
chan1.add_user(user1)
chan1_user1 = chan1.get_membership_by_user(user1)
chan1_user1.is_op = False
chan1.add_user(user_hallo)
chan1_hallo = chan1.get_membership_by_user(user_hallo)
chan1_hallo.is_op = True
chan2.add_user(user2)
chan2_user2 = chan2.get_membership_by_user(user2)
chan2_user2.is_op = False
chan2.add_user(user_hallo)
chan2_hallo = chan2.get_membership_by_user(user_hallo)
chan2_hallo.is_op = True
try:
self.function_dispatcher.dispatch("deop test_user2 test_chan2", user1, chan1)
data = serv1.get_send_data(1, chan1, Server.MSG_MSG)
assert "error" in data[0][0].lower()
assert "doesn't have op" in data[0][0].lower()
finally:
self.hallo.remove_server(serv1)
def test_run_by_search(self):
# Set up test servers and channels
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
chan1 = serv1.get_channel_by_name("test_chan1")
chan2 = serv1.get_channel_by_name("test_chan2")
serv2 = ServerMock(self.hallo)
serv2.name = "test_serv2"
chan3 = serv2.get_channel_by_name("test_chan1")
try:
self.hallo.add_server(serv1)
self.hallo.add_server(serv2)
# Set up rss feeds
rfl = E621SubList()
rf1 = E621Sub()
rf1.search = "butt"
rf1.server_name = chan1.server.name
rf1.channel_name = chan1.name
rf1.update_frequency = Commons.load_time_delta("PT3600S")
rfl.add_sub(rf1)
rf2 = E621Sub()
rf2.search = "deer"
rf2.server_name = chan2.server.name
rf2.channel_name = chan2.name
rf2.update_frequency = Commons.load_time_delta("PT3600S")
rfl.add_sub(rf2)
rf3 = E621Sub()
rf3.search = "dragon"
rf3.server_name = chan3.server.name
rf3.channel_name = chan3.name
rf3.update_frequency = Commons.load_time_delta("PT3600S")
rfl.add_sub(rf3)
# Splice this rss feed list into the function dispatcher's rss check object
rss_check_class = self.function_dispatcher.get_function_by_name("e621 sub check")
rss_check_obj = self.function_dispatcher.get_function_object(rss_check_class) # type: SubE621Check
rss_check_obj.e621_sub_list = rfl
# Invalid title
self.function_dispatcher.dispatch("e621 sub check Not a valid search", self.test_user, self.test_chan)
data = self.server.get_send_data(1, self.test_chan, Server.MSG_MSG)
assert "error" in data[0][0].lower()
# Correct title but wrong channel
self.function_dispatcher.dispatch("e621 sub check deer", self.test_user, chan1)
data = serv1.get_send_data(1, chan1, Server.MSG_MSG)
assert "error" in data[0][0].lower()
# Correct title check update
self.function_dispatcher.dispatch("e621 sub check deer", self.test_user, chan2)
data = serv1.get_send_data(1, chan2, Server.MSG_MSG)
assert "search updates were found" in data[0][0].lower()
assert len(data[0][0].lower().split("\n")) == 51
# No updates
self.function_dispatcher.dispatch("e621 sub check deer", self.test_user, chan2)
data = serv1.get_send_data(1, chan2, Server.MSG_MSG)
assert "no updates" in data[0][0], "No further updates should be found."
finally:
self.hallo.remove_server(serv2)
self.hallo.remove_server(serv1)
def test_deop_0(self):
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
serv1.type = Server.TYPE_IRC
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
chan1.in_channel = True
user1 = serv1.get_user_by_name("test_user1")
user_hallo = serv1.get_user_by_name(serv1.get_nick())
chan1.add_user(user1)
chan1.add_user(user_hallo)
chan1_user1 = chan1.get_membership_by_user(user1)
chan1_user1.is_op = True
chan1_hallo = chan1.get_membership_by_user(user_hallo)
chan1_hallo.is_op = True
try:
self.function_dispatcher.dispatch("deop", user1, chan1)
data = serv1.get_send_data(2)
assert "error" not in data[0][0].lower()
assert data[0][1] is None
assert data[1][1] == chan1
assert data[0][2] == Server.MSG_RAW
assert data[1][2] == Server.MSG_MSG
assert data[0][0][:4] == "MODE"
assert chan1.name+" -o "+user1.name in data[0][0]
assert "status taken" in data[1][0].lower()
finally:
self.hallo.remove_server(serv1)
def test_run_fail_not_bool(self):
# Set up a test hallo and server and channel and user
hallo1 = Hallo()
perm0 = PermissionMask()
hallo1.permission_mask = perm0
serv1 = ServerMock(hallo1)
serv1.name = "test_serv1"
perm1 = PermissionMask()
serv1.permission_mask = perm1
hallo1.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
perm2 = PermissionMask()
chan1.permission_mask = perm2
user1 = serv1.get_user_by_name("test_user1")
perm3 = PermissionMask()
user1.permission_mask = perm3
# Get permission mask of given channel
test_right = "test_right"
perm1.set_right(test_right, True)
self.function_dispatcher.dispatch("permissions server=test_serv1 "+test_right+" yellow", user1, chan1)
data = serv1.get_send_data(1, chan1, Server.MSG_MSG)
assert "error" in data[0][0].lower()
assert "don't understand your boolean value" in data[0][0].lower()
assert test_right in perm1.rights_map
assert perm1.rights_map[test_right]
def test_check_response(self):
# Setup common testing objects
serv1 = ServerMock(None)
serv1.name = "test_serv1"
chan1 = serv1.get_channel_by_name("test_chan1")
user1 = serv1.get_user_by_name("test_user1")
# Test basic response works
rm1 = ReplyMessage("test")
rm1.add_response("response")
assert rm1.check_response("just a test", user1, chan1) == "response"
# Test regex pattern match
rm2 = ReplyMessage("\\btest[0-9]+\\b")
rm2.add_response("response")
assert rm2.check_response("random testing", user1, chan1) is None
assert rm2.check_response("random test here", user1, chan1) is None
assert rm2.check_response("this is test3 I think", user1, chan1) == "response"
assert rm2.check_response("this is test4", user1, chan1) == "response"
# Test multi-response works
rm3 = ReplyMessage("test")
rm3.add_response("response1")
rm3.add_response("response2")
rm3.add_response("response3")
found_responses = set()
for _ in range(50):
response = rm3.check_response("another test", user1, chan1)
found_responses.add(response)
assert response in ["response1", "response2", "response3"]
assert len(found_responses) > 1
# Test replacements
rm4 = ReplyMessage("test")
rm4.add_response("response {USER} {CHANNEL} {SERVER}")
assert rm4.check_response("test", user1, chan1) == "response test_user1 test_chan1 test_serv1"
def test_1_user_just_name_privmsg(self):
# Set up a test server and channel and user
hallo1 = Hallo()
perm3 = PermissionMask()
hallo1.permission_mask = perm3
serv1 = ServerMock(hallo1)
serv1.name = "test_serv1"
perm0 = PermissionMask()
serv1.permission_mask = perm0
hallo1.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
perm1 = PermissionMask()
chan1.permission_mask = perm1
user1 = serv1.get_user_by_name("test_user1")
perm2 = PermissionMask()
user1.permission_mask = perm2
chan1.add_user(user1)
user2 = serv1.get_user_by_name("test_user2")
perm4 = PermissionMask()
user2.permission_mask = perm4
# Get permissions of specified user group
try:
data = self.perm_cont.find_permission_mask(["test_user2"], user1, None)
assert False, "Find permission mask should have failed."
except modules.PermissionControl.PermissionControlException as e:
assert "error" in str(e).lower()
assert "i can't find that permission mask" in str(e).lower()
def test_run_set_off(self):
# Set up a test hallo and server and channel and user
hallo1 = Hallo()
perm0 = PermissionMask()
hallo1.permission_mask = perm0
serv1 = ServerMock(hallo1)
serv1.name = "test_serv1"
perm1 = PermissionMask()
serv1.permission_mask = perm1
hallo1.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
perm2 = PermissionMask()
chan1.permission_mask = perm2
user1 = serv1.get_user_by_name("test_user1")
perm3 = PermissionMask()
user1.permission_mask = perm3
# Get permission mask of given channel
test_right = "test_right"
perm2.set_right(test_right, True)
self.function_dispatcher.dispatch("permissions server=test_serv1 channel=test_chan1 "+test_right+" off",
user1, chan1)
data = serv1.get_send_data(1, chan1, Server.MSG_MSG)
assert "error" not in data[0][0].lower()
assert "set "+test_right+" to false" in data[0][0].lower()
assert test_right in perm2.rights_map
assert not perm2.rights_map[test_right]
def test_1_user_group_no_name(self):
# Set up a test server and channel and user
hallo1 = Hallo()
perm3 = PermissionMask()
hallo1.permission_mask = perm3
serv1 = ServerMock(hallo1)
serv1.name = "test_serv1"
perm0 = PermissionMask()
serv1.permission_mask = perm0
hallo1.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
perm1 = PermissionMask()
chan1.permission_mask = perm1
user1 = serv1.get_user_by_name("test_user1")
perm2 = PermissionMask()
user1.permission_mask = perm2
group1 = UserGroup("test_group1", hallo1)
perm4 = PermissionMask()
group1.permission_mask = perm4
hallo1.add_user_group(group1)
# Try to get permissions of non-existent user group
try:
data = self.perm_cont.find_permission_mask(["user_group=test_group2"], user1, chan1)
assert False, "Find permission mask should have failed."
except modules.PermissionControl.PermissionControlException as e:
assert "error" in str(e).lower()
assert "no user group exists by that name" in str(e).lower()
def test_get_feeds_by_title(self):
serv1 = ServerMock(None)
serv1.name = "test_serv1"
serv2 = ServerMock(None)
serv2.name = "test_serv2"
chan1 = serv1.get_channel_by_name("test_chan1")
user2 = serv1.get_user_by_name("test_user2")
chan3 = serv2.get_channel_by_name("test_chan3")
# Setup a feed list
rfl = RssFeedList()
rf1 = RssFeed()
rf1.url = "http://spangle.org.uk/hallo/test_feed.xml?1"
rf1.title = "test_feed1"
rf1.server_name = chan1.server.name
rf1.channel_name = chan1.name
rfl.add_feed(rf1)
rf2 = RssFeed()
rf2.url = "http://spangle.org.uk/hallo/test_feed.xml?2"
rf2.title = "test_feed2"
rf2.server_name = user2.server.name
rf2.user_name = user2.name
rfl.add_feed(rf2)
rf3 = RssFeed()
rf3.url = "http://spangle.org.uk/hallo/test_feed.xml?3"
rf3.title = "test_feed3"
rf3.server_name = chan3.server.name
rf3.channel_name = chan3.name
rfl.add_feed(rf3)
rf4 = RssFeed()
rf4.url = "http://spangle.org.uk/hallo/test_feed.xml?4"
rf4.title = "test_feed4"
rf4.server_name = chan3.server.name
rf4.channel_name = chan3.name
rfl.add_feed(rf4)
rf5 = RssFeed()
rf5.url = "http://spangle.org.uk/hallo/test_feed.xml?5"
rf5.title = "test_feed3"
rf5.server_name = chan3.server.name
rf5.channel_name = chan3.name
rfl.add_feed(rf5)
# Check function
feed_list = rfl.get_feeds_by_title("test_feed3", chan3)
assert len(feed_list) == 2
assert rf3 in feed_list
assert rf5 in feed_list
def test_check_destination(self):
serv_name1 = "test_serv1"
serv_name2 = "test_serv2"
serv_name3 = "test_serv3"
chan_name1 = "test_chan1"
chan_name2 = "test_chan2"
chan_name3 = "test_chan3"
chan_name4 = "test_chan4"
chan_name5 = "test_chan5"
# Set up test destinations
serv1 = ServerMock(None)
serv2 = ServerMock(None)
serv3 = ServerMock(None)
serv1.name = serv_name1
serv2.name = serv_name2
serv3.name = serv_name3
chan1 = serv1.get_channel_by_name(chan_name1)
chan2 = serv1.get_channel_by_name(chan_name2)
chan3 = serv2.get_channel_by_name(chan_name3)
chan4 = serv3.get_channel_by_name(chan_name4)
chan5 = serv3.get_channel_by_name(chan_name5)
# Check when no whitelist or blacklist
rm = ReplyMessage("test")
assert rm.check_destination(chan1), "check_destination() not working without list"
assert rm.check_destination(chan2), "check_destination() not working without list"
assert rm.check_destination(chan3), "check_destination() not working without list"
assert rm.check_destination(chan4), "check_destination() not working without list"
assert rm.check_destination(chan5), "check_destination() not working without list"
# Add a blacklist for a specific channel on a specific server
rm.add_blacklist(serv_name1, chan_name1)
assert not rm.check_destination(chan1), "check_destination() not working with blacklist"
assert rm.check_destination(chan2), "check_destination() not working with blacklist"
assert rm.check_destination(chan3), "check_destination() not working with blacklist"
assert rm.check_destination(chan4), "check_destination() not working with blacklist"
assert rm.check_destination(chan5), "check_destination() not working with blacklist"
# Add a whitelist for a specific channel on a specific server
rm.add_whitelist(serv_name3, chan_name5)
assert not rm.check_destination(chan1), "check_destination() not working with blacklist"
assert not rm.check_destination(chan2), "check_destination() not working with blacklist"
assert not rm.check_destination(chan3), "check_destination() not working with blacklist"
assert not rm.check_destination(chan4), "check_destination() not working with blacklist"
assert rm.check_destination(chan5), "check_destination() not working with blacklist"
class TestBase(unittest.TestCase):
def setUp(self):
print("Starting test: "+self.id())
self.start_time = time.time()
# Create a Hallo
self.hallo = Hallo()
# Swap out raw printer function for empty
self.hallo.printer.output_raw = self.empty
# Only the required modules, only 1 (mock) server
# Todo: specify modules by test?
self.function_dispatcher = FunctionDispatcher({"AsciiArt", "Bio", "ChannelControl", "Euler", "Furry",
"HalloControl", "Math", "PermissionControl", "Rss",
"ServerControl", "Silly", "SillyEtd"},
self.hallo)
self.hallo.function_dispatcher = self.function_dispatcher
print("Running test: "+self.id()+". Init took: "+str(time.time()-self.start_time)+" seconds.")
self.server = ServerMock(self.hallo)
self.server.name = "mock-server"
# self.server = unittest.mock.Mock()
self.hallo.add_server(self.server)
# send shit in, check shit out
self.hallo_thread = Thread(target=self.hallo.start,)
self.hallo_thread.start()
self.test_user = self.server.get_user_by_name("test")
self.test_user.online = True
self.test_chan = self.server.get_channel_by_name("#test")
self.test_chan.in_channel = True
# Wait until hallo is open
count = 0
while not self.hallo.open:
time.sleep(0.1)
count += 1
assert count < 100, "Hallo took too long to start."
if count > 100:
break
# Clear any data in the server
self.server.get_send_data()
# Print test startup time
print("Running test: "+self.id()+". Startup took: "+str(time.time()-self.start_time)+" seconds.")
self.start_time = time.time()
def tearDown(self):
print("Finishing test: "+self.id()+". Test took: "+str(time.time()-self.start_time)+" seconds.")
self.hallo.close()
self.hallo_thread.join()
def empty(self, var1=None, var2=None, var3=None, var4=None):
pass
@classmethod
def tearDownClass(cls):
del cls
gc.collect()
def test_2_server_chan(self):
# Set up a test server and channel
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
perm1 = PermissionMask()
chan1.permission_mask = perm1
# Get permission mask of given channel
data = self.perm_cont.find_permission_mask(["server=test_serv1", "channel=test_chan1"],
self.test_user, self.test_chan)
assert perm1 == data, "Did not find the correct permission mask."
def test_1_channel_name(self):
# Set up a test server and channel and user
hallo1 = Hallo()
perm3 = PermissionMask()
hallo1.permission_mask = perm3
serv1 = ServerMock(hallo1)
serv1.name = "test_serv1"
perm0 = PermissionMask()
serv1.permission_mask = perm0
hallo1.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
perm1 = PermissionMask()
chan1.permission_mask = perm1
chan2 = serv1.get_channel_by_name("test_chan2")
perm4 = PermissionMask()
chan2.permission_mask = perm4
user1 = serv1.get_user_by_name("test_user1")
perm2 = PermissionMask()
user1.permission_mask = perm2
# Get permissions of current channel
data = self.perm_cont.find_permission_mask(["channel=test_chan2"], user1, chan1)
assert data == perm4, "Did not find the correct permission mask."
def test_server_specified_no_channel(self):
# Set up test resources
test_serv = ServerMock(self.hallo)
test_serv.name = "TestServer1"
self.hallo.add_server(test_serv)
test_chan = test_serv.get_channel_by_name("#not_in_channel")
test_chan.in_channel = False
# Send command
self.function_dispatcher.dispatch("leave server="+test_serv.name+" "+test_chan.name, self.test_user,
self.test_chan)
# Check response data
data = self.server.get_send_data(1, self.test_chan, Server.MSG_MSG)
self.server.get_left_channels(0)
test_serv.get_left_channels(0)
assert "error" in data[0][0].lower()
def test_reply_pew(self):
reply_func = self.function_dispatcher.get_function_by_name("reply")
reply_obj = self.function_dispatcher.get_function_object(reply_func) # type: Reply
# Check pewpew
response = reply_obj.passive_run(Function.EVENT_MESSAGE, "pew",
None, self.server, self.test_user, self.test_chan)
assert response == "pew pew"
# Check blacklist
serv1 = ServerMock(None)
serv1.name = "canternet"
chan1 = serv1.get_channel_by_name("#ukofequestria")
user1 = serv1.get_user_by_name("test_user")
response = reply_obj.passive_run(Function.EVENT_MESSAGE, "pew",
None, serv1, user1, chan1)
assert response is None
def test_deop_not_irc(self):
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
serv1.type = "NOT_IRC"
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
user1 = serv1.get_user_by_name("test_user1")
chan1.add_user(user1)
chan1.add_user(serv1.get_user_by_name(serv1.get_nick()))
try:
self.function_dispatcher.dispatch("deop", user1, chan1)
data = serv1.get_send_data(1, chan1, Server.MSG_MSG)
assert "error" in data[0][0].lower()
assert "only available for irc" in data[0][0].lower()
finally:
self.hallo.remove_server(serv1)
def test_deop_0_privmsg(self):
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
serv1.type = Server.TYPE_IRC
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
user1 = serv1.get_user_by_name("test_user1")
chan1.add_user(user1)
chan1.add_user(serv1.get_user_by_name(serv1.get_nick()))
try:
self.function_dispatcher.dispatch("deop", user1, user1)
data = serv1.get_send_data(1, user1, Server.MSG_MSG)
assert "error" in data[0][0].lower()
assert "in a private message" in data[0][0].lower()
finally:
self.hallo.remove_server(serv1)
def test_server_specified_second(self):
# Set up test resources
test_serv = ServerMock(self.hallo)
test_serv.name = "TestServer1"
self.hallo.add_server(test_serv)
test_chan = test_serv.get_channel_by_name("#other_serv")
test_chan.in_channel = True
# Send command
self.function_dispatcher.dispatch("leave "+test_chan.name+" server="+test_serv.name, self.test_user,
self.test_chan)
# Check response data
data = self.server.get_send_data(1, self.test_chan, Server.MSG_MSG)
self.server.get_left_channels(0)
chans = test_serv.get_left_channels(1)
assert chans[0] == test_chan
assert "left" in data[0][0].lower()
assert test_chan.name in data[0][0].lower()
def test_deop_0_no_power(self):
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
serv1.type = Server.TYPE_IRC
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
chan1.in_channel = True
user1 = serv1.get_user_by_name("test_user1")
chan1.add_user(user1)
chan1.add_user(serv1.get_user_by_name(serv1.get_nick()))
try:
self.function_dispatcher.dispatch("deop", user1, chan1)
data = serv1.get_send_data(1, chan1, Server.MSG_MSG)
assert "error" in data[0][0].lower()
assert "don't have power" in data[0][0].lower()
finally:
self.hallo.remove_server(serv1)
def test_get_response(self):
# Setup common testing objects
serv1 = ServerMock(None)
serv1.name = "test_serv1"
chan1 = serv1.get_channel_by_name("test_chan1")
user1 = serv1.get_user_by_name("test_user1")
# Basic ReplyMessageList get_response() test
rml1 = ReplyMessageList()
rm1 = ReplyMessage("test1")
rm1.add_response("response1")
rm2 = ReplyMessage("test2")
rm2.add_response("response2")
rml1.add_reply_message(rm1)
rml1.add_reply_message(rm2)
# Check response 1 works
assert rml1.get_response("test1", user1, chan1) == "response1"
assert rml1.get_response("test2", user1, chan1) == "response2"
assert rml1.get_response("test3", user1, chan1) is None
# Check blacklists
rml2 = ReplyMessageList()
rm1 = ReplyMessage("test")
rm1.add_response("response1")
rm1.add_blacklist(serv1.name, chan1.name)
rm2 = ReplyMessage("test")
rm2.add_response("response2")
rml2.add_reply_message(rm1)
rml2.add_reply_message(rm2)
assert rml2.get_response("test", user1, chan1) == "response2"
# Check whitelists
rml3 = ReplyMessageList()
rm1 = ReplyMessage("test")
rm1.add_response("response1")
rm1.add_whitelist(serv1.name, "not_chan_1")
rm2 = ReplyMessage("test")
rm2.add_response("response2")
rml3.add_reply_message(rm1)
rml3.add_reply_message(rm2)
assert rml3.get_response("test", user1, chan1) == "response2"
# Check replacements
rml4 = ReplyMessageList()
rm1 = ReplyMessage("test")
rm1.add_response("response {USER} {CHANNEL} {SERVER}")
rml4.add_reply_message(rm1)
assert rml4.get_response("test", user1, chan1) == "response test_user1 test_chan1 test_serv1"
def test_2_server_no_chan_user(self):
# Set up a test server and channel and user
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
perm1 = PermissionMask()
chan1.permission_mask = perm1
user1 = serv1.get_user_by_name("test_user1")
perm2 = PermissionMask()
user1.permission_mask = perm2
# Get permission mask of given channel
try:
data = self.perm_cont.find_permission_mask(["server=test_serv1", "core"],
user1, chan1)
assert False, "Should have failed to find any permission mask."
except modules.PermissionControl.PermissionControlException as e:
assert "error" in str(e).lower()
assert "server but not channel or user" in str(e).lower()
def test_deop_1priv_user_not_there(self):
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
serv1.type = Server.TYPE_IRC
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
chan1.in_channel = True
user1 = serv1.get_user_by_name("test_user1")
user_hallo = serv1.get_user_by_name(serv1.get_nick())
chan1.add_user(user_hallo)
chan1_hallo = chan1.get_membership_by_user(user_hallo)
chan1_hallo.is_op = True
try:
self.function_dispatcher.dispatch("deop test_chan1", user1, user1)
data = serv1.get_send_data(1, user1, Server.MSG_MSG)
assert "error" in data[0][0].lower()
assert "test_user1 is not in test_chan1" in data[0][0].lower()
finally:
self.hallo.remove_server(serv1)
def test_1_hallo(self):
# Set up a test hallo and server and channel and user
hallo1 = Hallo()
perm3 = PermissionMask()
hallo1.permission_mask = perm3
serv1 = ServerMock(hallo1)
serv1.name = "test_serv1"
perm0 = PermissionMask()
serv1.permission_mask = perm0
hallo1.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
perm1 = PermissionMask()
chan1.permission_mask = perm1
user1 = serv1.get_user_by_name("test_user1")
perm2 = PermissionMask()
user1.permission_mask = perm2
# Get permission of hallo
data = self.perm_cont.find_permission_mask(["hallo"], user1, chan1)
assert data == perm3, "Did not find the correct permission mask."
def test_reply_haskell(self):
reply_func = self.function_dispatcher.get_function_by_name("reply")
reply_obj = self.function_dispatcher.get_function_object(reply_func) # type: Reply
# Check haskell.jpg
response = reply_obj.passive_run(Function.EVENT_MESSAGE, "haskell.jpg",
None, self.server, self.test_user, self.test_chan)
assert response is None
# Check on correct channel
serv1 = ServerMock(None)
serv1.name = "shadowworld"
chan1 = serv1.get_channel_by_name("#ecco-the-dolphin")
user1 = serv1.get_user_by_name("test_user")
response = reply_obj.passive_run(Function.EVENT_MESSAGE, "haskell.jpg",
None, serv1, user1, chan1)
assert "http" in response.lower()
assert "haskell.jpg" in response.lower()
# Check image exists
page_request = urllib.request.Request(response)
page_opener = urllib.request.build_opener()
response_data = page_opener.open(page_request).read()
assert len(response_data) > 0, "haskell.jpg image does not exist."
def test_invite_0_fail(self):
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
serv1.type = Server.TYPE_IRC
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
chan1.in_channel = True
user1 = serv1.get_user_by_name("test_user1")
user_hallo = serv1.get_user_by_name(serv1.get_nick())
chan1.add_user(user1)
chan1.add_user(user_hallo)
chan1_user1 = chan1.get_membership_by_user(user1)
chan1_user1.is_op = False
chan1_hallo = chan1.get_membership_by_user(user_hallo)
chan1_hallo.is_op = True
try:
self.function_dispatcher.dispatch("invite", user1, chan1)
data = serv1.get_send_data(1, chan1, Server.MSG_MSG)
assert "error" in data[0][0].lower()
assert "specify a user to invite and/or a channel" in data[0][0].lower()
finally:
self.hallo.remove_server(serv1)
def test_voice_0_is_op(self):
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
serv1.type = Server.TYPE_IRC
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
chan1.in_channel = True
user1 = serv1.get_user_by_name("test_user1")
user_hallo = serv1.get_user_by_name(serv1.get_nick())
chan1.add_user(user1)
chan1.add_user(user_hallo)
chan1_user1 = chan1.get_membership_by_user(user1)
chan1_user1.is_op = True
chan1_hallo = chan1.get_membership_by_user(user_hallo)
chan1_hallo.is_op = True
try:
self.function_dispatcher.dispatch("voice", user1, chan1)
data = serv1.get_send_data(1, chan1, Server.MSG_MSG)
assert "error" in data[0][0].lower()
assert "already has voice" in data[0][0].lower()
finally:
self.hallo.remove_server(serv1)
def test_mute_1_no_power(self):
serv1 = ServerMock(self.hallo)
serv1.name = "test_serv1"
serv1.type = Server.TYPE_IRC
self.hallo.add_server(serv1)
chan1 = serv1.get_channel_by_name("test_chan1")
chan1.in_channel = True
user1 = serv1.get_user_by_name("test_user1")
user_hallo = serv1.get_user_by_name(serv1.get_nick())
chan1.add_user(user1)
chan1.add_user(user_hallo)
chan1_user1 = chan1.get_membership_by_user(user1)
chan1_user1.is_op = False
chan1_hallo = chan1.get_membership_by_user(user_hallo)
chan1_hallo.is_op = False
try:
self.function_dispatcher.dispatch("mute test_chan1", user1, user1)
data = serv1.get_send_data(1, user1, Server.MSG_MSG)
assert "error" in data[0][0].lower()
assert "don't have power" in data[0][0].lower()
finally:
self.hallo.remove_server(serv1)
