darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-02 17:31:14 -05:00
Code Issues Packages Projects Releases Wiki Activity

Created endpoints for create and delete token.

Browse Source
This commit is contained in:
Rebecca Law
2016-01-13 14:05:49 +00:00
parent 3a3f9becec
commit 725b976d31
10 changed files with 210 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
tests/app/dao/test_tokens_dao.py
View File

@@ -1,16 +1,45 @@
import uuid
from app.dao import tokens_dao
from app.dao import api_tokens_dao
from app.models import ApiToken
from app.models import Token
def test_should_create_token(notify_api, notify_db, notify_db_session, sample_service):
token = uuid.uuid4()
api_token = ApiToken(**{'token': token, 'service_id': sample_service.id})
api_token = Token(**{'token': token, 'service_id': sample_service.id})
api_tokens_dao.save_token_model(api_token)
all_tokens = api_tokens_dao.get_model_api_tokens()
tokens_dao.save_token_model(api_token)
all_tokens = tokens_dao.get_model_tokens()
assert len(all_tokens) == 1
assert all_tokens[0].token == str(token)
def test_should_delete_api_token(notify_api, notify_db, notify_db_session, sample_service):
token = uuid.uuid4()
api_token = Token(**{'token': token, 'service_id': sample_service.id})
tokens_dao.save_token_model(api_token)
all_tokens = tokens_dao.get_model_tokens()
assert len(all_tokens) == 1
tokens_dao.delete_model_token(all_tokens[0])
empty_token_list = tokens_dao.get_model_tokens()
assert len(empty_token_list) == 0
def test_should_return_token_for_service(notify_api, notify_db, notify_db_session, sample_service):
the_token = str(uuid.uuid4())
api_token = Token(**{'token': the_token, 'service_id': sample_service.id})
tokens_dao.save_token_model(api_token)
token = tokens_dao.get_model_tokens(sample_service.id)
assert token.service_id == sample_service.id
assert token.token == str(the_token)
def test_delete_model_token_should_remove_token(notify_api, notify_db, notify_db_session, sample_service):
api_token = Token(**{'token': str(uuid.uuid4()), 'service_id': sample_service.id})
tokens_dao.save_token_model(api_token)
all_tokens = tokens_dao.get_model_tokens()
assert len(all_tokens) == 1
tokens_dao.delete_model_token(all_tokens[0])
assert len(tokens_dao.get_model_tokens()) == 0