1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
import optparse
import sys
import time
import subprocess
import os
import sleekxmpp
from sleekxmpp.xmlstream.matcher import StanzaPath, MatchXPath
from sleekxmpp.xmlstream.handler import Callback
class Responder(sleekxmpp.ClientXMPP):
def __init__(self, jid, password, room, room_password, nick):
sleekxmpp.ClientXMPP.__init__(self, jid, password)
self.room = room
self.room_password = room_password
self.nick = nick
self.finished = False
self.tests = {}
class Client(sleekxmpp.ClientXMPP):
def __init__(self, jid, password, room, nick):
sleekxmpp.ClientXMPP.__init__(self, jid, password)
self.room = room
self.nick = nick
self.add_event_handler("session_start", self.start)
self.add_event_handler("message", self.message)
self.add_event_handler('roster_update', self.roster_update)
self.finished = False
self.tests = {}
self.tests["timeline_poll"] = ["Timeline received automatically", False]
self.tests["help"] = ["#help command", False]
self.tests["status1"] = ["#status command response", False]
self.tests["status2"] = ["#status command pushes status", False]
self.tests["follow"] = ["#follow command", False]
self.tests["friends"] = ["#friends command", False]
self.tests["unfollow"] = ["#unfollow command", False]
self.tests["friends2"] = ["#friends after unfollow command", False]
self.tests["mode1"] = ["#mode 1 response", False]
self.tests["mode1buddy"] = ["#mode 1 buddy added", False]
self.tests["mode2"] = ["#mode 2 join room", False]
self.status = "timeline"
self.timestamp = int(time.time())
def message(self, msg):
if self.status == "timeline" and msg['body'].find("spectrum2tests") != -1 and msg['body'].find("MsgId") != -1:
self.tests["timeline_poll"][1] = True
self.send_message(mto=msg['from'], mbody="#help")
self.status = "help"
elif self.status == "help" and msg['body'].find("You will receive tweets of people you follow") != -1:
self.tests["help"][1] = True
self.send_message(mto=msg['from'], mbody="#status My testing status " + str(self.timestamp))
self.status = "status"
elif self.status == "status" and msg['body'].find("Status Update successful") != -1:
self.tests["status1"][1] = True
elif self.status == "status" and msg['body'].find("spectrum2tests") != -1 and msg['body'].find("MsgId") != -1:
self.tests["status2"][1] = True
self.status = "follow"
self.send_message(mto=msg['from'], mbody="#follow colinpwheeler")
elif self.status == "follow" and msg['body'] == "You are now following colinpwheeler":
self.status = "friends"
self.tests["follow"][1] = True
self.send_message(mto=msg['from'], mbody="#friends")
elif self.status == "friends" and msg['body'].find("USER LIST") != -1 and msg['body'].find("colinpwheeler") != -1:
self.status = "mode1"
self.tests["friends"][1] = True
self.send_message(mto=msg['from'], mbody="#mode 1")
elif self.status == "mode1" and msg['body'] == "Changed mode to 1":
self.tests["mode1"][1] = True
elif self.status == "unfollow" and msg['body'] == "You are not following colinpwheeler anymore":
self.status = "friends2"
self.tests["unfollow"][1] = True
self.send_message(mto=msg['from'], mbody="#friends")
elif self.status == "friends2" and msg['body'].find("USER LIST") != -1 and msg['body'].find("colinpwheeler") == -1:
self.status = "mode1"
self.tests["friends2"][1] = True
self.finished = True
def roster_update(self, iq):
roster = iq['roster']['items']
if self.status == "mode1" and roster.has_key("colinpwheeler@localhost"):
self.plugin['xep_0045'].joinMUC("#twitter@localhost", "client", wait=True)
self.tests["mode2"][1] = True
self.plugin['xep_0045'].leaveMUC("#twitter@localhost", "client")
self.status = "unfollow"
self.tests["mode1buddy"][1] = True
self.send_message(mto="twitter.com@localhost", mbody="#unfollow colinpwheeler")
def start(self, event):
self.getRoster()
self.sendPresence()
|
