def test_copy_password(self):
'''
Test to copy the password to the clipboard
'''
self.new_credentials.save_credentials()
Credentials.copy_password("instagram")
self.assertEqual(self.new_credentials.password, pyperclip.paste())
def test_delete_credential(self):
'''
Test the deletion of one credential
'''
self.new_credentials.save_credentials()
test_credentials = Credentials("twitter", "Bot", "root")
random_credentials = Credentials("Spotify", "user", "username")
test_credentials.save_credentials()
random_credentials.save_credentials()
random_credentials.delete_credential()
self.assertEqual(len(Credentials.credentials_list),2)
def test_credential_exist(self):
'''
Test that a credential exists
'''
self.new_credentials.save_credentials()
test_credentials = Credentials("twitter", "Bot", "root")
random_credentials = Credentials("Spotify", "user", "username")
test_credentials.save_credentials()
random_credentials.save_credentials()
account_exists = Credentials.account_exists("Spotify")
self.assertTrue(account_exists)
def test_find_by_account_name(self):
'''
Test finding an account by it's name
'''
self.new_credentials.save_credentials()
test_credentials = Credentials("twitter", "Bot", "root")
random_credentials = Credentials("Spotify", "user", "username")
test_credentials.save_credentials()
random_credentials.save_credentials()
found_account = Credentials.find_by_account_name("Spotify")
self.assertEqual(random_credentials.account_name, found_account.account_name)
def test_save_multiple_credentials(self):
'''
Test the saving of multiple credentials
'''
self.new_credentials.save_credentials()
test_credentials = Credentials("twitter", "Bot", "root")
random_credentials = Credentials("Spotify", "user", "username")
test_credentials.save_credentials()
random_credentials.save_credentials()
self.assertEqual(len(Credentials.credentials_list), 3)
def copy_credentials(site_name):
return Credentials.copy_credentials(site_name)
def delete_credential(credentials):
Credentials.delete_credentials()
def display_credentials(user_name):
return Credentials.display_credentials(user_name)
def rand_pass(size):
return Credentials.rand_pass(size)
def setUp(self):
'''
SetUp method that is performed everytime a test case is run
'''
self.new_credentials = Credentials("instagram","usertest","userpass")
def display_credentials():
'''
Function to display the credentials
'''
return Credentials.display_credentials()
def find_credential(account):
'''
Function to find a credential using the account name
'''
return Credentials.find_by_account_name(account)
def test_display_credentials(self):
'''
Test that credentials can be displayed
'''
self.assertEqual(Credentials.display_credentials(), Credentials.credentials_list)
class TestCredential(unittest.TestCase):
'''
Test class that defines the test functions for the Credentials class
Args:
unittest.case: Testcase class that helps in defining test cases
'''
def setUp(self):
'''
SetUp method that is performed everytime a test case is run
'''
self.new_credentials = Credentials("instagram","usertest","userpass")
def tearDown(self):
'''
TearDown method that cleans up after every test case
'''
Credentials.credentials_list = []
def test_init(self):
'''
Test if object is correctly initialized
'''
self.assertEqual(self.new_credentials.account_name, "instagram")
self.assertEqual(self.new_credentials.user_name,"usertest")
self.assertEqual(self.new_credentials.password, "userpass")
def test_save_credential(self):
'''
Test the function to save credentials
'''
self.new_credentials.save_credentials()
self.assertEqual(len(Credentials.credentials_list),1)
def test_save_multiple_credentials(self):
'''
Test the saving of multiple credentials
'''
self.new_credentials.save_credentials()
test_credentials = Credentials("twitter", "Bot", "root")
random_credentials = Credentials("Spotify", "user", "username")
test_credentials.save_credentials()
random_credentials.save_credentials()
self.assertEqual(len(Credentials.credentials_list), 3)
def test_delete_credential(self):
'''
Test the deletion of one credential
'''
self.new_credentials.save_credentials()
test_credentials = Credentials("twitter", "Bot", "root")
random_credentials = Credentials("Spotify", "user", "username")
test_credentials.save_credentials()
random_credentials.save_credentials()
random_credentials.delete_credential()
self.assertEqual(len(Credentials.credentials_list),2)
def test_find_by_account_name(self):
'''
Test finding an account by it's name
'''
self.new_credentials.save_credentials()
test_credentials = Credentials("twitter", "Bot", "root")
random_credentials = Credentials("Spotify", "user", "username")
test_credentials.save_credentials()
random_credentials.save_credentials()
found_account = Credentials.find_by_account_name("Spotify")
self.assertEqual(random_credentials.account_name, found_account.account_name)
def test_credential_exist(self):
'''
Test that a credential exists
'''
self.new_credentials.save_credentials()
test_credentials = Credentials("twitter", "Bot", "root")
random_credentials = Credentials("Spotify", "user", "username")
test_credentials.save_credentials()
random_credentials.save_credentials()
account_exists = Credentials.account_exists("Spotify")
self.assertTrue(account_exists)
def test_display_credentials(self):
'''
Test that credentials can be displayed
'''
self.assertEqual(Credentials.display_credentials(), Credentials.credentials_list)
def test_copy_password(self):
'''
Test to copy the password to the clipboard
'''
self.new_credentials.save_credentials()
Credentials.copy_password("instagram")
self.assertEqual(self.new_credentials.password, pyperclip.paste())
def find_by_site_name(site_name):
return Credentials.find_by_site_name(site_name)
def create_credential(user_namez, site_name, passwordz):
new_credential = Credentials(user_namez, site_name, passwordz)
return new_credential
def create_credentials(account_name, user_name, password):
'''
Function to create credentials
'''
new_credential = Credentials(account_name, user_name, password)
return new_credential
def save_credential(credential):
return Credentials.save_credential(credential)
def credential_exist(account):
'''
Function to check if a credential exists
'''
return Credentials.account_exists(account)
def authenticate_user(user_name, password):
checking_user = Credentials.check_user(user_name, password)
return checking_user
def copy_password(account):
'''
Function to copy password of the credentials password
'''
return Credentials.copy_password(account)
def get_credentials():
result = Credentials()
file = open("credentials.txt", "r")
result.method = file.readline().strip()
result.domain = file.readline().strip()
result.endpoint_badges = file.readline().strip()
result.endpoint_profile = file.readline().strip()
result.key_recipient = file.readline().strip()
result.key_badge = file.readline().strip()
result.key_message = file.readline().strip()
result.key_token = file.readline().strip()
result.prefix_cookie = file.readline().strip()
file.close()
return result