source: .tests/btlib.py @ bda5012

import socket
import sys
import time
import select

FUN = [
"Did I ask you something?",
"Oh yeah, that's right.",
"Alright, alright. Now go back to work.",
"Buuuuuuuuuuuuuuuurp... Excuse me!",
"Yes?",
"No?",
]

SEPARATOR = "="*60

ALWAYSSHOWLOG = False

class IrcClient:
    def __init__(self, nick, pwd):
        self.nick = nick
        self.pwd = pwd
        self.log = ''
        self.tmplog = ''
        self.sck = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    def send_raw(self, msg, loud = False, log = True):
        self.receive()
        if loud:
            print('FROM '+ self.nick + '|| ' + msg)
        if log:
            self.log += msg+'\r\n'
            self.tmplog += msg+'\r\n'
        self.sck.send((msg+'\r\n').encode())

    def send_priv_msg(self, recip, msg, loud = False):
        self.send_raw('PRIVMSG '+recip+' :'+msg, loud)

    def connect(self):
        connected = False
        for x in range(5):
            try:
                self.sck.connect(('127.0.0.1', 6667))
                connected = True
                break
            except:
                time.sleep(1)

        if not connected:
            return False

        self.send_raw('USER ' + (self.nick + " ")*3)
        self.send_raw('NICK ' + self.nick)
        self.send_raw('JOIN &bitlbee')
        return True

    def jabber_login(self):
        self.send_priv_msg("&bitlbee", "account add jabber "+self.nick+"@localhost "+self.pwd)
        time.sleep(0.3)
        self.send_priv_msg("&bitlbee", "account on")
        time.sleep(1)
        self.receive()
        return (self.log.find('Logged in') != -1)

    def receive(self, wait = 1):
        text = ''
        while True:
            readable, _, _ = select.select([self.sck], [], [], 5)
            if self.sck in readable:
                text += self.sck.recv(2040).decode()
                for line in text.split('\n'):
                    if line.find('PING') != -1:
                        self.send_raw('PONG ' + line.split()[1])
            else:
                time.sleep(wait)
                if self.sck not in readable:
                    break
        self.log += text
        self.tmplog += text
        return text

    def add_jabber_buddy(self, nick):
        self.send_priv_msg("&bitlbee", "add 0 " + nick+"@localhost")
    
    def block_jabber_buddy(self, nick):
        self.send_priv_msg("&bitlbee", "block " + nick)

    def unblock_jabber_buddy(self, nick):
        self.send_priv_msg("&bitlbee", "allow " + nick)

    def rename_jabber_buddy(self, oldnick, newnick):
        self.send_priv_msg("&bitlbee", "rename " + oldnick + " " + newnick)
        
def msg_comes_thru(sender, receiver, message):
    sender.send_priv_msg(receiver.nick, message)
    received = receiver.receive().find(message) != -1
    return received

def perform_test(test_function):
    clis = []
    clis += [IrcClient('test1', 'asd')]
    clis += [IrcClient('test2', 'asd')]

    fail = not test_function(clis)

    if ALWAYSSHOWLOG or fail:
        for cli in clis:
            cli.receive()
            print(SEPARATOR)
            print("Test Log "+ cli.nick+":")
            print(cli.tmplog)
        print(SEPARATOR)
    
    if fail:
        sys.exit(1)

def connect_test(clis):
    ret = True
    for cli in clis:
        ret = ret & cli.connect()
    return ret

def yes_test(clis):
    ret = False
    clis[0].send_priv_msg("&bitlbee", "yes")
    clis[0].receive()
    for x, fun in enumerate(FUN):
        if (clis[0].log.find(fun) != -1):
            ret = True
            if x:
                print("The RNG gods smile upon us")
            break
    return ret

def jabber_login_test(clis):
    ret = True
    for cli in clis:
        ret = ret & cli.jabber_login()
    return ret

def jabber_delete_account_test(clis):
    ret = True
    clis[1].receive()
    clis[1].send_priv_msg("&bitlbee", "account list")
    time.sleep(0.5)
    ret = ret & (clis[1].receive().find(clis[1].nick+'@localhost') != -1)

    clis[1].send_priv_msg("&bitlbee", "account on")
    clis[1].send_priv_msg("&bitlbee", "account "+clis[1].nick+" del")
    clis[1].receive()
    clis[1].send_priv_msg("&bitlbee", "account list")
    time.sleep(0.5)
    ret = ret & (clis[1].receive().find(clis[1].nick+'@localhost') != -1)

    clis[1].send_priv_msg("&bitlbee", "account off")
    clis[1].send_priv_msg("&bitlbee", "account "+clis[1].nick+" del")
    clis[1].receive()
    clis[1].send_priv_msg("&bitlbee", "account list")
    time.sleep(0.5)
    ret = ret & (clis[1].receive().find(clis[1].nick+'@localhost') == -1)
    return ret

def register_test(clis):
    clis[1].send_priv_msg("&bitlbee", "register "+clis[1].pwd*2)
    ret = (clis[1].receive(wait = 10).find('Account successfully created') != -1)
    print("What is this"+ ret)
    return ret
    
def unregister_test(clis):
    clis[1].send_priv_msg("&bitlbee", "drop "+clis[1].pwd*2)
    time.sleep(0.5)
    ret = (clis[1].receive().find('removed') != -1)
    clis[1].send_priv_msg("&bitlbee", "drop "+clis[1].pwd*2)
    time.sleep(0.5)
    ret = ret & (clis[1].receive().find('That account does not exist') != -1)
    return ret

def identify_test(clis):
    ret = True
    clis[1].send_priv_msg("&bitlbee", "identify "+clis[1].pwd)
    time.sleep(0.5)
    ret = ret & (clis[1].receive().find('Incorrect password') != -1)

    clis[1].send_priv_msg("&bitlbee", "identify "+clis[1].pwd*2)
    time.sleep(0.5)
    ret = ret & (clis[1].receive().find('Password accepted') != -1)
    return ret

def identify_nonexist_test(clis):
    clis[1].send_priv_msg("&bitlbee", "register "+clis[1].pwd)
    time.sleep(0.5)
    return (clis[1].receive().find('The nick is (probably) not registered') != -1)

def add_buddy_test(clis):
    clis[0].add_jabber_buddy(clis[1].nick)
    clis[1].send_priv_msg("&bitlbee", "yes")

    clis[1].add_jabber_buddy(clis[0].nick)
    clis[0].send_priv_msg("&bitlbee", "yes")

    clis[0].send_priv_msg("&bitlbee", "blist")
    junk = clis[0].receive()
    ret = junk.find(clis[1].nick) != -1
    ret = ret & (junk.find("1 available") != -1)
    return ret

def remove_buddy_test(clis):
    ret = add_buddy_test(clis)

    clis[0].send_priv_msg("&bitlbee", "remove " +clis[1].nick)
    clis[0].send_priv_msg("&bitlbee", "blist all")
    time.sleep(0.5)
    ret = ret & (clis[0].receive().find(clis[1].nick) == -1)

    clis[0].add_jabber_buddy(clis[1].nick)
    clis[1].send_priv_msg("&bitlbee", "yes")
    time.sleep(1)
    clis[0].send_priv_msg("&bitlbee", "blist all")
    time.sleep(0.5)
    junk = clis[0].receive()
    #ret = ret & (junk.find("1 available") != -1)
    ret = ret & (junk.find(clis[1].nick) != -1)
    return ret

def message_test(clis):
    ret = msg_comes_thru(clis[0], clis[1], 'ohai <3')
    ret = ret & msg_comes_thru(clis[1], clis[0], 'uwu *pounces*')
    return ret

def block_test(clis):
    clis[0].block_jabber_buddy(clis[1].nick)
    ret = not msg_comes_thru(clis[1], clis[0], 'm-meow?')
    clis[0].unblock_jabber_buddy(clis[1].nick)
    ret = ret & msg_comes_thru(clis[1], clis[0], '*purrs*')
    return ret

def rename_test(clis):
    newname = "xXx_pup_LINKENPARK4EVA"
    message = "rawr meanmz i luv<3 u in dinosaur"

    clis[0].rename_jabber_buddy(clis[1].nick, newname)
    clis[0].send_priv_msg(newname, message)
    ret = clis[1].receive().find(message) != -1

    clis[0].rename_jabber_buddy("-del", newname)
    ret = ret & msg_comes_thru(clis[0], clis[1], "rawr")
    return ret

def status_test(clis):
    status = "get out of my room mom"
    clis[1].send_priv_msg("&bitlbee", "set status '"+status+"'")
    clis[0].send_priv_msg("&bitlbee", "info "+clis[1].nick)
    ret = (clis[0].receive().find("jabber - Status message: "+status) != -1)

    clis[1].send_priv_msg("&bitlbee", "set")
    ret = ret & (clis[1].receive().find(status) != -1)

    clis[1].send_priv_msg("&bitlbee", "set -del status")
    clis[0].send_priv_msg("&bitlbee", "info "+clis[1].nick)
    ret = ret & (clis[0].receive().find("jabber - Status message: (none)") != -1)
    return ret

def offline_test(clis):
    clis[0].send_priv_msg("&bitlbee", "account off")

    junk = clis[0].receive()
    ret = (junk.find(clis[1].nick) != -1)
    ret = ret & (junk.find("QUIT") != -1)

    junk = clis[1].receive()
    ret = ret & (junk.find(clis[0].nick) != -1)
    ret = ret & (junk.find("QUIT") != -1)

    clis[0].send_priv_msg(clis[1].nick, "i'm not ur mom")
    ret = ret & (clis[0].receive().find("No such nick/channel") != -1)

    clis[0].send_priv_msg("&bitlbee", "account on")

    junk = clis[0].receive()
    ret = ret & (junk.find(clis[1].nick) != -1)
    ret = ret & (junk.find("JOIN") != -1)

    junk = clis[1].receive()
    ret = ret & (junk.find(clis[0].nick) != -1)
    ret = ret & (junk.find("JOIN") != -1)

    return ret

def default_target_test(clis):
    clis[0].send_priv_msg("&bitlbee", "set default_target last")
    clis[0].send_priv_msg("&bitlbee", "test2: ur mah default now")
    
    ret = (clis[1].receive().find("ur mah default now") != -1)

    clis[0].send_priv_msg("&bitlbee", "mrow")
    ret = ret & (clis[1].receive().find("mrow") != -1)

    clis[0].send_priv_msg("root", "set default_target root")
    junk = clis[0].receive()
    ret = ret & (junk.find("default_target") != -1)
    ret = ret & (junk.find("root") != -1)

    clis[0].send_priv_msg("&bitlbee", "yes")
    ret = ret & (clis[1].receive().find("yes") == -1)
    return ret 

def help_test(clis):
    clis[0].send_priv_msg("&bitlbee", "help")
    ret = (clis[0].receive().find("identify_methods") != -1)
    clis[0].send_priv_msg("&bitlbee", "help commands")
    ret = ret & (clis[0].receive().find("qlist") != -1)
    return ret