#!/usr/bin/env python

# -*- coding: utf-8 -*-

"""Tests for requests_kerberos."""

from mock import Mock, patch

import requests

import kerberos

import requests_kerberos

import unittest

# kerberos.authClientInit() is called with the service name (HTTP@FQDN) and

# returns 1 and a kerberos context object on success. Returns -1 on failure.

clientInit_complete = Mock(return_value=(1, "CTX"))

clientInit_error = Mock(return_value=(-1, "CTX"))

# kerberos.authGSSClientStep() is called with the kerberos context object

# returned by authGSSClientInit and the negotiate auth token provided in the

# http response's www-authenticate header. It returns 0 or 1 on success. 0

# Indicates that authentication is progressing but not complete.

clientStep_complete = Mock(return_value=1)

clientStep_continue = Mock(return_value=0)

clientStep_error = Mock(return_value=-1)

clientStep_exception = Mock(side_effect=kerberos.GSSError)

# kerberos.authGSSCLientResponse() is called with the kerberos context which

# was initially returned by authGSSClientInit and had been mutated by a call by

# authGSSClientStep. It returns a string.

clientResponse = Mock(return_value="GSSRESPONSE")

# Note: we're not using the @mock.patch decorator:

# > My only word of warning is that in the past, the patch decorator hides

# > tests when using the standard unittest library.

# > -- sigmavirus24 in https://github.com/requests/requests-kerberos/issues/1

class KerberosTestCase(unittest.TestCase):

clientResponse.assert_called_with("CTX")

if __name__ == '__main__':

unittest.main()