def test_token(self, delete_mock, get_mock, put_mock): TOKEN = '394c46b8902fb5e8fc9268f3cfd84539' SECRET = '394c46b8902fb5e8fc9268f3cfd84538' token_dict = dict(oauth_token=TOKEN, oauth_token_secret=SECRET, id=TOKEN) token = AuthToken(oauth_token=TOKEN, oauth_token_secret=SECRET) self.assertEquals(token.to_dict(), token_dict) token.save() put_mock.assert_called_with(token.TABLE_NAME, token_dict) get_mock.return_value = None self.assertEquals(AuthToken.get_auth('invalid'), None) # get_item get_mock.return_value = token_dict self.assertEquals(AuthToken.get_auth(TOKEN), token_dict) # delete AuthToken.delete(TOKEN) delete_mock.assert_called_with(token.TABLE_NAME, id=TOKEN) get_mock.return_value = None self.assertEquals(AuthToken.get_auth('invalid'), None)