# Copyright (c) 2013 Rackspace, 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.

import json
from oslo_utils import uuidutils

import falcon

from zaqar.tests.unit.transport.wsgi import base


class TestValidation(base.V1Base):

    config_file = 'wsgi_mongodb_validation.conf'

    def setUp(self):
        super(TestValidation, self).setUp()

        self.project_id = '7e55e1a7e'

        self.queue_path = self.url_prefix + '/queues/noein'
        self.simulate_put(self.queue_path, self.project_id)

        self.headers = {
            'Client-ID': uuidutils.generate_uuid(),
        }

    def tearDown(self):
        self.simulate_delete(self.queue_path, self.project_id)
        super(TestValidation, self).tearDown()

    def test_metadata_deserialization(self):
        # Normal case
        self.simulate_put(self.queue_path + '/metadata',
                          self.project_id,
                          body='{"timespace": "Shangri-la"}')

        self.assertEqual(falcon.HTTP_204, self.srmock.status)

        # Too long
        max_queue_metadata = 64

        doc_tmpl = '{{"Dragon Torc":"{0}"}}'
        doc_tmpl_ws = '{{ "Dragon Torc" : "{0}" }}'  # with whitespace
        envelope_length = len(doc_tmpl.format(''))

        for tmpl in doc_tmpl, doc_tmpl_ws:
            gen = '0' * (max_queue_metadata - envelope_length + 1)
            doc = tmpl.format(gen)
            self.simulate_put(self.queue_path + '/metadata',
                              self.project_id,
                              body=doc)

            self.assertEqual(falcon.HTTP_400, self.srmock.status)

    def test_message_deserialization(self):
        # Normal case
        self.simulate_post(self.queue_path + '/messages',
                           self.project_id,
                           body='[{"body": "Dragon Knights", "ttl": 100}]',
                           headers=self.headers)

        self.assertEqual(falcon.HTTP_201, self.srmock.status)

        # Both messages' size are too long
        max_messages_post_size = 256

        obj = {'a': 0, 'b': ''}
        envelope_length = len(json.dumps(obj, separators=(',', ':')))
        obj['b'] = 'x' * (max_messages_post_size - envelope_length + 1)

        for long_body in ('a' * (max_messages_post_size - 2 + 1), obj):
            doc = json.dumps([{'body': long_body, 'ttl': 100}])
            self.simulate_post(self.queue_path + '/messages',
                               self.project_id,
                               body=doc,
                               headers=self.headers)

            self.assertEqual(falcon.HTTP_400, self.srmock.status)

    def test_request_without_client_id(self):
        # Unlike newer APIs (v1.1 and v2), there will be no error 400, because
        # of missing Client-ID in headers.
        empty_headers = {}
        self.simulate_put(self.queue_path,
                          self.project_id,
                          headers=empty_headers)
        # Queue was already created by setUp, expecting 204 response code.
        self.assertEqual(falcon.HTTP_204, self.srmock.status)

    def test_request_without_client_id_if_resource_name_contains_v2_text(self):
        empty_headers = {}
        queue_path_with_v2 = self.url_prefix + '/queues/my_name_is_v2'
        self.simulate_put(queue_path_with_v2,
                          self.project_id,
                          headers=empty_headers)
        self.addCleanup(self.simulate_delete, queue_path_with_v2,
                        self.project_id)
        self.assertEqual(falcon.HTTP_201, self.srmock.status)

    def test_queue_metadata_putting(self):
        # Ensure setting reserved queue attributes (which names start with
        # '_') is not allowed in API v1.

        # Try set real _default_message_ttl queue attribute.
        self.simulate_put(self.queue_path + '/metadata',
                          self.project_id,
                          body='{"_default_message_ttl": 60}')
        self.assertEqual(falcon.HTTP_400, self.srmock.status)

        # Try set a fictional queue attribute.
        self.simulate_put(self.queue_path + '/metadata',
                          self.project_id,
                          body='{"_min_message_niceness": 9000}')
        self.assertEqual(falcon.HTTP_400, self.srmock.status)