class Test(unittest.TestCase):
def setUp(self):
self.db = DataBase()
self.db._change_file_name('TestDatabase.db')
self.friends_data = \
[{'name': 'Vasya', 'bdate': '01.03.1998', 'skype': 'blabla'},
{'name': 'Bob', 'bdate': '03.06.1876',
'country': 'USA', 'city': 'Boston'}]
self.db.create(data=(self.friends_data, None))
def test_insert_user(self):
user_data = {'name': 'Artyom', 'bdate':
'03.06.1997', 'country': 'Russia'}
resp = self.db.insert_user(user_data)
assert True, resp
def test_get_user_inf(self):
name = 'Vasya'
name1 = 'Bob'
data = self.db.get_user_inf(name)
data1 = self.db.get_user_inf(name1)
flag = data is not None and data1 is not None
assert True, flag
def test_update_user_inf(self):
up_data = {'instagram': 'qwer'}
self.db.update_user_inf('Bob', up_data)
inf_user = self.db.get_user_inf('Bob')
assert True, inf_user.get('instagram') is not None
def test_delete_user(self):
user = {'name': 'Artyom Soldatenko',
'email': '*****@*****.**', 'bdate': '3.6.1997'}
self.db.insert_user(user)
self.db.delete_user(user['name'])
res = self.db.contains_user(user['name'])
assert True, not res
def test_is_same_user_bdate(self):
old_user = {'name': 'Artyom Soldatenko',
'email': '*****@*****.**', 'bdate': '03.06.1997'}
user = {'name': 'Artyom Soldatenko',
'email': '*****@*****.**', 'bdate': '3.6.1997'}
res = self.db.is_same_users(old_user, user)
assert True, res
def test_is_same_user_not_inf(self):
old_user = {'name': 'Artyom Soldatenko', 'email': '', 'bdate': ''}
user = {'name': 'Artyom Soldatenko',
'email': '*****@*****.**', 'bdate': '3.6.1997'}
res = self.db.is_same_users(old_user, user)
assert True, res
def test_is_same_user_error(self):
old_user = {'name': 'Artyom Soldatenko',
'email': '', 'bdate': '12.12.1998'}
user = {'name': 'Artyom Soldatenko',
'email': '*****@*****.**', 'bdate': '3.6.1997'}
res = self.db.is_same_users(old_user, user)
assert True, not res
def test_contains_user(self):
user = {'name': 'Artyom Soldatenko'}
self.db.insert_user(user)
assert True, self.db.contains_user(user['name'])
def test_merge_users(self):
old_user = {'name': 'Artyom Soldatenko', 'email': '', 'bdate': ''}
self.db.insert_user(old_user)
user = {'name': 'Artyom Soldatenko',
'email': '*****@*****.**', 'bdate': '3.6.1997'}
self.db.merge_user(old_user, user)
res = self.db.contains_user('Artyom Soldatenko')
new_user = self.db.get_user_inf('Artyom Soldatenko')
res1 = new_user['email'] == '*****@*****.**' and \
new_user['bdate'] == '3.6.1997'
assert True, res and res1
