def setup_server(server):
server.users = [User('u1'), User('u2'), User('u3')]
conn1 = Connection('uuid1', None)
conn2 = Connection('uuid2', None)
conn3 = Connection('uuid3', None)
server.users[0].login(conn1)
server.users[1].login(conn2)
server.users[2].login(conn3)
server.connections = [conn1, conn2, conn3]
def test_list_wildcard():
server = Server(host=None)
setup_server(server)
conn = Connection('newconn', None)
server._process_list(conn, '*1')
assert conn.send_buffer[-2:] == b'u1'
def test_list():
server = Server(host=None)
setup_server(server)
conn = Connection('newconn', None)
server._process_list(conn, None)
assert conn.send_buffer[-12:] == b'u1|||u2|||u3'
def test_conn2user_notpresent():
server = Server(host=None)
setup_server(server)
conn = Connection('asdfasdf', None)
user = server.conn2user(conn)
assert user is None
def test_create():
server = Server(host=None)
setup_server(server)
conn = Connection('newconn', None)
server._process_create(conn, 'newuser')
assert len(server.users) == 4
assert server.users[-1].username == 'newuser'
def test_create_failure():
server = Server(host=None)
setup_server(server)
conn = Connection('newconn', None)
server._process_create(conn, 'u2')
assert len(server.users) == 3
assert conn.send_buffer[-len('username already exists'
):] == b'username already exists'
def test_login_alreadylogged():
server = Server(host=None)
setup_server(server)
conn = Connection('newconn', None)
server._process_login(conn, server.users[2].username)
wp = WireProtocol()
wp.parse_incoming_bytes(conn.send_buffer)
assert wp.command == CMD.RESPONSE
assert wp.data_len == len('user is already logged in')
assert wp.data_buffer == b'user is already logged in'
class TestHappyConnection(TestCase):
def setUp(self):
self.client_sock = MagicMock()
self.client_addr = MagicMock()
self.client_server = MagicMock()
self.connection = Connection(self.client_sock, self.client_sock,
self.client_server)
def test_init(self):
self.assertIsInstance(self.connection, Connection)
self.assertFalse(self.connection.is_alive())
self.assertFalse(self.connection.nickname)
def test_send_notlogged():
server = Server(host=None)
setup_server(server)
conn = Connection('newconn', None)
server._process_send(
conn,
[server.users[1].username, server.users[2].username, 'test message'])
wp = WireProtocol()
wp.parse_incoming_bytes(conn.send_buffer)
assert wp.command == CMD.RESPONSE # indicates error
def test_login_noaccount():
server = Server(host=None)
setup_server(server)
server.users[2].logout()
conn = Connection('newconn', None)
server._process_login(conn, server.users[2].username + 'asdfasdf')
wp = WireProtocol()
wp.parse_incoming_bytes(conn.send_buffer)
assert wp.command == CMD.RESPONSE
assert wp.data_len == len('username does not exist')
assert wp.data_buffer == b'username does not exist'
def setUp(self):
self.client_sock = MagicMock()
self.client_addr = MagicMock()
self.client_server = MagicMock()
self.connection = Connection(self.client_sock, self.client_sock,
self.client_server)
def test_conn2user_empty():
server = Server(host=None)
conn = Connection('uuid1', None)
user = server.conn2user(conn)
assert user is None
import requests
from server import Connection , Action
app = Flask(__name__)
app.debug = True
app.secret_key = 'secretKey'
app.config["CACHE_TYPE"] = "null"
class Contacts:
name = None
address = None
email = None
mobile = None
connection , res = Connection.open(host = 'localhost' , user = '******', pwd = '', db = 'db_test' , port =3306)
@app.route('/', methods=['GET'])
def home():
if connection:
return render_template('index.html' ,result = connection )
else:
return render_template('index.html' , result = None)
@app.route('/details/contacts' , methods= ['GET'])
def details():
entries = Action(connection).Show_Details()
return render_template('details.html' , entries = entries )
@app.route('/update/contact/<int:id>' , methods= ['GET'])