# Copyright 2009 ITA Software, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from twisted.internet import reactor
from nagcat.unittests.queries import QueryTestCase
from nagcat.unittests import dummy_server
from nagcat import errors


class TCPQueryTestCase(QueryTestCase):

    def setUp(self):
        super(TCPQueryTestCase, self).setUp()
        self.server = reactor.listenTCP(0, dummy_server.TCP())
        self.port = self.server.getHost().port
        self.config = {'type': "tcp", 'host': "localhost", 'port': self.port}

    def testBasic(self):
        d = self.startQuery(self.config)
        d.addCallback(self.assertEquals, "hello\n")
        return d

    def testPost(self):
        d = self.startQuery(self.config, data="post data")
        d.addCallback(self.assertEquals, "post data")
        return d

    def tearDown(self):
        return self.server.loseConnection()

class SSLQueryTestCase(QueryTestCase):

    def start(self, *args, **kwargs):
        context = dummy_server.ssl_context(*args, **kwargs)
        self.server = reactor.listenSSL(0, dummy_server.TCP(), context)
        self.port = self.server.getHost().port
        self.config = {'type': "ssl", 'host': "localhost", 'port': self.port}

    def testPost(self):
        self.start("localhost-a.key", "localhost-a.cert")
        d = self.startQuery(self.config, data="post data")
        d.addCallback(self.assertEquals, "post data")
        return d

    def testVerifyClientBad(self):
        self.start("localhost-a.key", "localhost-a.cert", ["ca.cert"], verify=True)

        def check(result):
            self.assertIsInstance(result, errors.Failure)
            self.assertIsInstance(result.value, errors.TestCritical)

        d = self.startQuery(self.config, data="post data")
        d.addBoth(check)
        return d

    def testVerifyClientGood(self):
        self.start("localhost-a.key", "localhost-a.cert", ["ca.cert"], verify=True)
        config = self.config.copy()
        config['ssl_key'] = dummy_server.ssl_path("localhost-b.key")
        config['ssl_cert'] = dummy_server.ssl_path("localhost-b.cert")
        d = self.startQuery(config, data="post data")
        d.addCallback(self.assertEquals, "post data")
        return d

    def testVerifyServerBad(self):
        self.start("localhost-a.key", "localhost-a.cert")
        config = self.config.copy()
        config['ssl_cacert'] = dummy_server.ssl_path("bogus-ca.cert")

        def check(result):
            self.assertIsInstance(result, errors.Failure)
            self.assertIsInstance(result.value, errors.TestCritical)

        d = self.startQuery(config, data="post data")
        d.addBoth(check)
        return d

    def testVerifyServerGood(self):
        self.start("localhost-a.key", "localhost-a.cert")
        config = self.config.copy()
        config['ssl_cacert'] = dummy_server.ssl_path("ca.cert")
        d = self.startQuery(config, data="post data")
        d.addCallback(self.assertEquals, "post data")
        return d

    def tearDown(self):
        return self.server.loseConnection()