def test_update_status(db):
assert main.add_user("test1", "*****@*****.**", "tester", "one",
db) is True
assert main.add_status("test1_status_id", "test1", "test_status_text",
db) is True
assert main.update_status("test1_status_id", "test1", "NEW_status_text",
db) is True
def baseline():
'''
Method needed for memory profiling
'''
user_baseline = main.init_user_collection()
status_baseline = main.init_status_collection()
main.load_users('accounts.csv', user_baseline)
main.load_status_updates('status_updates.csv', status_baseline)
main.add_user('dave03', '*****@*****.**', 'Dave', 'Jones', user_baseline)
main.add_status('dave03', 'dave03_01', 'Random text', status_baseline)
main.update_user('dave03', '*****@*****.**', 'Dave', 'Rex', user_baseline)
main.update_status('dave03', 'dave03_01', 'Whos in charge now',
status_baseline)
main.search_user('dave03', user_baseline)
main.search_status('dave03_01', status_baseline)
main.delete_status('dave03_01', status_baseline)
main.delete_user('dave03', user_baseline)
def test_update_status_false(stolen, status_collection):
user_id = stolen.user_id
status_id = stolen.status_id
status_text = stolen.status_text
params = (status_id, user_id, status_text, status_collection)
result = main.update_status(*params)
assert result is False
def test_update_status(status):
'''
Test that modify_status is called correctly
'''
with patch('main.user_status.UserStatusCollection') as mocker:
update = mocker.return_value.modify_status
result = main.update_status(*status, mocker())
update.assert_called_with(*status)
assert result is update.return_value
def test_user_status_interfaces(status):
'''
Test functions that interface with user_status
'''
assert main.search_status(status[0]) is None
assert main.add_status(status[1], status[0], status[2])
assert main.update_status(*status)
assert main.search_status(status[0]) == user_status.search_status(
status[0])
assert main.delete_status(status[0])
def test_update_status_true(stolen, status_collection, status_database):
status_collection.database = status_database
user_id = 'test'
status_id = stolen.status_id
status_text = 'test'
params = (status_id, user_id, status_text, status_collection)
result = main.update_status(*params)
assert result is True
def update_status():
'''
Updates information for an existing status
'''
user_id = input('User ID: ')
status_id = input('Status ID: ')
status_text = input('Status text: ')
if not main.update_status(status_id, user_id, status_text):
print("An error occurred while trying to update status")
else:
print("Status was successfully updated")
def test_update_status_not_found(self):
self.assertEqual(
main.update_status('ldconejo_001', 'ldconejo', 'Code still works!',
self.status_collection), False)
def test_update_status(self):
main.add_status('ldconejo_001', 'ldconejo', 'Just testing my code',
self.status_collection)
self.assertEqual(
main.update_status('ldconejo_001', 'ldconejo', 'Code still works!',
self.status_collection), True)