toxygen/tests/tests.py

163 lines
5.9 KiB
Python
Raw Normal View History

2016-07-06 15:25:04 +02:00
from toxygen.profile import *
from toxygen.tox_dns import tox_dns
2017-02-25 21:53:33 +01:00
from toxygen.history import History
2017-02-27 19:44:35 +01:00
from toxygen.smileys import SmileyLoader
from toxygen.messages import *
2017-02-11 18:23:08 +01:00
import toxygen.toxes as encr
2017-02-25 21:53:33 +01:00
import toxygen.util as util
2016-10-22 21:21:26 +02:00
import time
2016-02-23 12:11:00 +01:00
2016-07-01 22:15:00 +02:00
class TestTox:
2016-02-23 12:11:00 +01:00
2016-03-06 13:44:52 +01:00
def test_creation(self):
2016-07-01 22:15:00 +02:00
name = b'Toxygen User'
status_message = b'Toxing on Toxygen'
2016-03-06 13:44:52 +01:00
tox = tox_factory()
tox.self_set_name(name)
tox.self_set_status_message(status_message)
data = tox.get_savedata()
del tox
2016-03-15 20:12:37 +01:00
tox = tox_factory(data)
2016-07-01 22:15:00 +02:00
assert tox.self_get_name() == str(name, 'utf-8')
assert tox.self_get_status_message() == str(status_message, 'utf-8')
2016-03-06 13:44:52 +01:00
2016-10-27 23:55:34 +02:00
class TestProfileHelper:
def test_creation(self):
file_name, path = 'test.tox', os.path.dirname(os.path.realpath(__file__)) + '/'
data = b'test'
with open(path + file_name, 'wb') as fl:
fl.write(data)
ph = ProfileHelper(path, file_name[:4])
assert ProfileHelper.get_path() == path
assert ph.open_profile() == data
assert os.path.exists(path + 'avatars/')
2016-07-01 22:15:00 +02:00
class TestEncryption:
2016-05-15 12:39:03 +02:00
def test_encr_decr(self):
2016-10-22 21:21:26 +02:00
tox = tox_factory()
data = tox.get_savedata()
2017-02-11 18:23:08 +01:00
lib = encr.ToxES()
for password in ('easypassword', 'njvnFjfn7vaGGV6', 'toxygen'):
2016-10-31 22:24:37 +01:00
lib.set_password(password)
copy_data = data[:]
new_data = lib.pass_encrypt(data)
2017-02-27 19:44:35 +01:00
assert lib.is_data_encrypted(new_data)
2016-10-31 22:24:37 +01:00
new_data = lib.pass_decrypt(new_data)
assert copy_data == new_data
2016-10-22 21:21:26 +02:00
2017-02-27 19:44:35 +01:00
class TestSmileys:
def test_loading(self):
settings = {'smiley_pack': 'default', 'smileys': True}
sm = SmileyLoader(settings)
assert sm.get_smileys_path() is not None
l = sm.get_packs_list()
assert len(l) == 4
2017-02-25 21:53:33 +01:00
def create_singletons():
folder = util.curr_directory() + '/abc'
Settings._instance = Settings.get_default_settings()
if not os.path.exists(folder):
os.makedirs(folder)
ProfileHelper(folder, 'test')
2016-10-22 21:21:26 +02:00
2017-02-25 21:53:33 +01:00
def create_friend(name, status_message, number, tox_id):
friend = Friend(None, number, name, status_message, None, tox_id)
return friend
2017-02-27 19:44:35 +01:00
def create_random_friend():
name, status_message, number = 'Friend', 'I am friend!', 0
tox_id = '76518406F6A9F2217E8DC487CC783C25CC16A15EB36FF32E335A235342C48A39218F515C39A6'
friend = create_friend(name, status_message, number, tox_id)
return friend
2017-02-25 21:53:33 +01:00
class TestFriend:
2016-10-22 21:21:26 +02:00
def test_friend_creation(self):
2017-02-25 21:53:33 +01:00
create_singletons()
2016-10-22 21:21:26 +02:00
name, status_message, number = 'Friend', 'I am friend!', 0
tox_id = '76518406F6A9F2217E8DC487CC783C25CC16A15EB36FF32E335A235342C48A39218F515C39A6'
2017-02-25 21:53:33 +01:00
friend = create_friend(name, status_message, number, tox_id)
2016-10-22 21:21:26 +02:00
assert friend.name == name
assert friend.tox_id == tox_id
assert friend.status_message == status_message
assert friend.number == number
def test_friend_corr(self):
2017-02-25 21:53:33 +01:00
create_singletons()
2017-02-27 19:44:35 +01:00
friend = create_random_friend()
2016-10-22 21:21:26 +02:00
t = time.time()
2017-02-27 19:44:35 +01:00
friend.append_message(InfoMessage('Info message', t))
friend.append_message(TextMessage('Hello! It is test!', MESSAGE_OWNER['ME'], t + 0.001, 0))
friend.append_message(TextMessage('Hello!', MESSAGE_OWNER['FRIEND'], t + 0.002, 0))
2016-10-22 21:21:26 +02:00
assert friend.get_last_message_text() == 'Hello! It is test!'
assert len(friend.get_corr()) == 3
assert len(friend.get_corr_for_saving()) == 2
2017-02-27 19:44:35 +01:00
friend.append_message(TextMessage('Not sent', MESSAGE_OWNER['NOT_SENT'], t + 0.002, 0))
2016-10-22 21:21:26 +02:00
arr = friend.get_unsent_messages_for_saving()
assert len(arr) == 1
assert arr[0][0] == 'Not sent'
2017-02-27 19:44:35 +01:00
tm = TransferMessage(MESSAGE_OWNER['FRIEND'],
time.time(),
TOX_FILE_TRANSFER_STATE['RUNNING'],
100, 'file_name', friend.number, 0)
2016-10-22 21:21:26 +02:00
friend.append_message(tm)
friend.clear_corr()
assert len(friend.get_corr()) == 1
assert len(friend.get_corr_for_saving()) == 0
2017-02-27 19:44:35 +01:00
friend.append_message(TextMessage('Hello! It is test!', MESSAGE_OWNER['ME'], t, 0))
2016-10-22 21:21:26 +02:00
assert len(friend.get_corr()) == 2
assert len(friend.get_corr_for_saving()) == 1
2016-10-27 23:55:34 +02:00
2017-02-27 19:44:35 +01:00
def test_history_search(self):
create_singletons()
friend = create_random_friend()
message = 'Hello! It is test!'
friend.append_message(TextMessage(message, MESSAGE_OWNER['ME'], time.time(), 0))
last_message = friend.get_last_message_text()
assert last_message == message
result = friend.search_string('e[m|s]')
assert result is not None
result = friend.search_string('tox')
assert result is None
2017-02-25 21:53:33 +01:00
class TestHistory:
def test_history(self):
create_singletons()
db_name = 'my_name'
name, status_message, number = 'Friend', 'I am friend!', 0
tox_id = '76518406F6A9F2217E8DC487CC783C25CC16A15EB36FF32E335A235342C48A39218F515C39A6'
friend = create_friend(name, status_message, number, tox_id)
history = History(db_name)
history.add_friend_to_db(friend.tox_id)
assert history.friend_exists_in_db(friend.tox_id)
text_message = 'Test!'
t = time.time()
2017-02-27 19:44:35 +01:00
friend.append_message(TextMessage(text_message, MESSAGE_OWNER['ME'], t, 0))
2017-02-25 21:53:33 +01:00
messages = friend.get_corr_for_saving()
history.save_messages_to_db(friend.tox_id, messages)
getter = history.messages_getter(friend.tox_id)
messages = getter.get_all()
assert len(messages) == 1
assert messages[0][0] == text_message
assert messages[0][1] == MESSAGE_OWNER['ME']
assert messages[0][-1] == 0
history.delete_message(friend.tox_id, t)
getter = history.messages_getter(friend.tox_id)
messages = getter.get_all()
assert len(messages) == 0
2017-02-27 19:44:35 +01:00
history.delete_friend_from_db(friend.tox_id)
assert not history.friend_exists_in_db(friend.tox_id)