def test_create_friendship(self): db = User() f = Friendship() db.register_user("alex", "123", self.engine) db.register_user("ben", "123", self.engine) f1 = db.login_user("ben", "123", self.engine) f2 = db.login_user("alex", "123", self.engine) self.assertTrue(f.create_friendship(f1, 'alex', self.engine))
def test_send_friend_request(self): db = User() r = FriendRequest() db.init_data_base(self.engine) db.register_user("ben", "123", self.engine) db.register_user("alex", "123", self.engine) f1 = db.login_user("ben", "123", self.engine) f2 = db.login_user("alex", "123", self.engine) waiting = r.check_waiting_request(f1, self.engine) self.assertEqual(len(waiting), 0) self.assertTrue(r.send_friend_request(f2, 'ben', self.engine)) waiting = r.check_waiting_request(f1, self.engine) self.assertEqual(len(waiting), 1) self.assertTrue(r.denial_request(f1, 'alex', self.engine)) waiting = r.check_waiting_request(f1, self.engine) self.assertEqual(len(waiting), 0) self.assertTrue(r.send_friend_request(f2, 'ben', self.engine)) self.assertTrue(r.confirm_request(f1, 'alex', self.engine)) self.assertEqual(len(waiting), 0)
def setUp(self): self.engine = create_engine('sqlite:///:memory:') db = User() db.init_data_base(self.engine)
def test_update_ip(self): db = User() db.register_user("ben", "1234", self.engine) t = db.login_user("ben", "1234", self.engine) self.assertTrue(db.set_user_ip(t, "127.0.0.1", self.engine))
def test_login(self): db = User() db.register_user("ben1", "1234", self.engine) self.assertTrue(db.login_user("ben1", "1234", self.engine))
def test_register(self): db = User() self.assertTrue(db.register_user("ben", "123", self.engine))
import telebot, time from random import choice from telebot import types import config from data_base import User bot = telebot.TeleBot(config.TOKEN) db = User("data.db") undone_time = lambda x: time.strftime("%H:%M:%S %d.%m.%Y", time.localtime(x)) bad_advice_list = db.copy_bad_advice() choice_advice_list = db.copy_choice_advice() rest_list = db.copy_rest() helper_list = db.copy_helper() @bot.message_handler(commands=['start']) def menu(message): bot.send_message(message.chat.id, text='Вот что я умею: ', reply_markup=keyboard()) @bot.message_handler(content_types=['text']) def message_handler(message): user_id = message.from_user.id stat = undone_time(message.date) if (not db.get_all_id(user_id)): db.add_id_to_db(user_id)