#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you 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 unittest import TestCase
from qpid.spec010 import load
from qpid.codec010 import Codec, StringCodec
from qpid.testlib import testrunner
from qpid.datatypes import Struct

class SpecTest(TestCase):

  def setUp(self):
    self.spec = load(testrunner.get_spec_file("amqp.0-10.xml"))

  def testSessionHeader(self):
    hdr = self.spec["session.header"]
    sc = StringCodec(self.spec)
    hdr.encode(sc, Struct(hdr, sync=True))
    assert sc.encoded == "\x01\x01"

    sc = StringCodec(self.spec)
    hdr.encode(sc, Struct(hdr, sync=False))
    assert sc.encoded == "\x01\x00"

  def encdec(self, type, value):
    sc = StringCodec(self.spec)
    type.encode(sc, value)
    decoded = type.decode(sc)
    return decoded

  def testMessageProperties(self):
    mp = self.spec["message.message_properties"]
    rt = self.spec["message.reply_to"]

    props = Struct(mp, content_length=0xDEADBEEF,
                   reply_to=Struct(rt, exchange="the exchange name",
                                   routing_key="the routing key"))
    dec = self.encdec(mp, props)
    assert props.content_length == dec.content_length
    assert props.reply_to.exchange == dec.reply_to.exchange
    assert props.reply_to.routing_key == dec.reply_to.routing_key

  def testMessageSubscribe(self):
    ms = self.spec["message.subscribe"]
    cmd = Struct(ms, exclusive=True, destination="this is a test")
    dec = self.encdec(self.spec["message.subscribe"], cmd)
    assert cmd.exclusive == dec.exclusive
    assert cmd.destination == dec.destination