def test_reset_credentials_file(self): # Check get_cred return all the keys tools.reset_credentials_file() reset_creds = tools.get_credentials_file() expected = ['username', 'stream_ids', 'api_key', 'proxy_username', 'proxy_password'] self.assertTrue(all(x in reset_creds for x in expected))
def test_credentials_tools(): original_creds = tls.get_credentials_file() expected = ['username', 'stream_ids', 'api_key'] assert all(x in original_creds for x in expected) # now, if that worked, we can try this! tls.reset_credentials_file() reset_creds = tls.get_credentials_file() tls.set_credentials_file(**original_creds) assert all(x in reset_creds for x in expected) creds = tls.get_credentials_file() assert all(x in creds for x in expected) assert original_creds == creds