class TokenProviderTest(unittest.TestCase): def setUp(self): self.tokenProvider = TokenProvider() remove_storage() def tearDown(self): remove_storage() def test_should_be_none_access_token_when_file_doesnt_exist(self): self.assertEquals(None, self.tokenProvider.get_access_token(), "should be none access token when file doesnt exist") def test_should_set_access_token(self): self.tokenProvider.set_access_token('asd') access_token = self.tokenProvider.get_access_token() print access_token self.assertEquals('asd', access_token, "should set access token") def test_should_delete_file_when_remove_access_token(self): self.tokenProvider.remove_token_storage() isFileExist = os.path.isfile(token_provider.STORAGE_PATH) self.assertFalse(isFileExist)
from token_provider import TokenProvider import sys def terminate_if_newer_API_available(is_version_ok): if not is_version_ok: print("There's newer API version available, script will terminate.") exit(1) IS_SANDBOX = sys.argv[1] in ['true', "True"] SAVED_NOTES_FOLDER_PATH = sys.argv[2] TOKEN_FILEPATH = sys.argv[3] NOTEBOOK_NAME = sys.argv[4] token_provider = TokenProvider(filepath=TOKEN_FILEPATH) auth_token = token_provider.get_token(IS_SANDBOX) evernote_client = EvernoteClient(token=auth_token, sandbox=IS_SANDBOX, china=False) is_version_ok = APIVersionValidator().validate_api_version(evernote_client) terminate_if_newer_API_available(is_version_ok) def notes_filter(note): return True #not note.title.startswith('_') notes = NotesProvider(NOTEBOOK_NAME).save_notes_content( evernote_client, notes_filter, SAVED_NOTES_FOLDER_PATH)
def setUp(self): self.tokenProvider = TokenProvider() remove_storage()