def test_fail_tok(self):
# Test invalid (ignored) token
c = FTConn(MockFTSock(True))
c.fts.sock.append_bytes(b'Y') # Not a valid token
assert c.receive_data() == (b'Y', None)
assert c.fts.sock.ensure_erecv and c.fts.sock.ensure_esend()
def test_send_req_l(self):
# Testing sending a list request
c = FTConn(MockFTSock(True))
c.request_file_list()
assert c.fts.sock.check_bytes(FTProto.REQ_LIST)
assert c.fts.sock.ensure_esend() and c.fts.sock.ensure_erecv()
def test_recv_req_l(self):
# Testing receive list request
c = FTConn(MockFTSock(True))
c.fts.sock.append_bytes(FTProto.REQ_LIST)
assert c.receive_data() == (FTProto.REQ_LIST, None)
assert c.fts.sock.ensure_esend() and c.fts.sock.ensure_erecv()
def test_send_req_f(self):
# Testing sending a file request
c = FTConn(MockFTSock(True))
c.request_file(test_file_name.encode())
assert c.fts.sock.check_bytes(FTProto.REQ_FILE)
assert c.fts.sock.check_bytes(pr(test_file_name.encode()))
assert c.fts.sock.ensure_erecv() and c.fts.sock.ensure_esend()
def test_recv_req_f(self):
# Testing receive file request
c = FTConn(MockFTSock(True))
c.fts.sock.append_bytes(FTProto.REQ_FILE)
c.fts.sock.append_bytes(pr(test_file_name.encode()))
assert c.receive_data() == (FTProto.REQ_FILE, test_file_name.encode())
assert c.fts.sock.ensure_esend() and c.fts.sock.ensure_erecv()
def test_send_res_l(self):
# Testing sending a file response
c = FTConn(MockFTSock(True))
c.send_file(test_file_name.encode(), test_file_contents)
assert c.fts.sock.check_bytes(FTProto.RES_FILE)
assert c.fts.sock.check_bytes(pr(test_file_name.encode()))
assert c.fts.sock.check_bytes(pr(test_file_contents))
assert c.fts.sock.ensure_erecv() and c.fts.sock.ensure_esend()
def test_recv_res_f(self):
# Testing receiving a file response
c = FTConn(MockFTSock(True))
c.fts.sock.append_bytes(FTProto.RES_FILE)
c.fts.sock.append_bytes(pr(test_file_name.encode()))
c.fts.sock.append_bytes(pr(test_file_contents))
assert c.receive_data() == (FTProto.RES_FILE, \
(test_file_name.encode(), test_file_contents))
assert c.fts.sock.ensure_erecv() and c.fts.sock.ensure_esend()
def tst_connect_hs_s_v(self):
# Testing version mismatch
c = FTConn(MockFTSock())
c.fts.sock.append_bytes(correct_handshake)
c.fts.sock.append_bytes(pi(wrong_version))
assert c.connect(0, 0) == "Handshake failed"
# Checking *sent* handshake
assert c.fts.sock.check_bytes(correct_hs_resp)
assert c.fts.sock.check_bytes(pi(correct_version))
assert c.fts.sock.ensure_esend() and c.fts.sock.ensure_erecv()
def test_connect_hs_s_s(self):
# Testing string mismatch
c = FTConn(MockFTSock())
c.fts.sock.append_bytes(wrong_handshake)
c.fts.sock.append_bytes(pi(correct_version))
assert c.connect(0, 0) == "Handshake failed"
# Should have sent out correct handshake anyway
assert c.fts.sock.check_bytes(wrong_hs_c_resp)
assert c.fts.sock.check_bytes(pi(correct_version))
assert c.fts.sock.ensure_esend() and c.fts.sock.ensure_erecv()
def test_connect_hs_s_c(self):
# Testing successful handshake
c = FTConn(MockFTSock())
c.fts.sock.append_bytes(correct_handshake)
c.fts.sock.append_bytes(pi(correct_version))
assert c.connect(0, 0) == "Success"
# Checking sent handshake
assert c.fts.sock.check_bytes(correct_hs_resp)
assert c.fts.sock.check_bytes(pi(correct_version))
assert c.fts.sock.ensure_esend() and c.fts.sock.ensure_erecv()
assert c.fts.sock.connected
def test_connect_hs_c_c(self):
# Testing successful handshake
c = FTConn(MockFTSock())
c.fts.crv = (True, "Client", "Success")
# Because of the way we handshake, we need to set the
# response before the socket sends the initiation
c.fts.sock.append_bytes(correct_hs_resp)
c.fts.sock.append_bytes(pi(correct_version))
assert c.connect(0, 0) == "Success"
# Check sent handshake initiation
assert c.fts.sock.check_bytes(correct_handshake)
assert c.fts.sock.check_bytes(pi(correct_version))
assert c.fts.sock.ensure_esend() and c.fts.sock.ensure_erecv()
def test_f_sr(self):
# Test send/recv of file
c1 = FTConn(MockFTSock(True))
c2 = FTConn(MockFTSock(True))
c1.send_file(test_file_name.encode(), test_file_contents)
c2.fts.sock.append_bytes(c1.fts.sock.retrieve_bytes())
assert c2.receive_data() == (FTProto.RES_FILE, (test_file_name.encode(), \
test_file_contents))
assert c1.fts.sock.ensure_esend() and c1.fts.sock.ensure_erecv()
assert c2.fts.sock.ensure_esend() and c2.fts.sock.ensure_erecv()
def test_connect_fail(self):
# Server
c = FTConn(MockFTSock())
c.fts.crv = (False, "Server", "Failure")
assert c.connect(0, 0) == "Failure"
# Client
c = FTConn(MockFTSock())
c.fts.crv = (False, "Client", "Failure")
assert c.connect(0, 0) == "Failure"
def test_fl_sr(self):
# Test send/recv of filelist
c1 = FTConn(MockFTSock(True))
c2 = FTConn(MockFTSock(True))
fl = [FileInfo(path=Path(), file_hash=b'12345678901234567890123456789012', \
is_dir=False, mtime=98)]
c1.send_file_list(fl)
c2.fts.sock.append_bytes(c1.fts.sock.retrieve_bytes())
t, flr = c2.receive_data()
assert t == FTProto.RES_LIST
assert len(fl) == len(flr)
for orig, recv in zip(fl, flr):
assert file_info_equals(orig, recv)
assert c1.fts.sock.ensure_esend() and c1.fts.sock.ensure_erecv()
assert c2.fts.sock.ensure_esend() and c2.fts.sock.ensure_erecv()