#-------------------------------------------------------------------------
# Copyright (c) Microsoft. All rights reserved.
#
# 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.
#--------------------------------------------------------------------------
require "integration/test_helper"

require "azure/service_bus/brokered_message"

class ScenarioHelper
  def ScenarioHelper.out (s)
    if (false)
      print s + "\n"
    end
  end

  def ScenarioHelper.get_custom_properties(i)
    custom_properties             = {}
    custom_properties['i']        = i;
    custom_properties['test']     = Time.local(2001, i, 1)
    custom_properties['name']     = 'Test' + i.to_s;
    custom_properties['meanname'] = "'\"Me`\\'&*<>!@#%^*)\n" + i.to_s;
    custom_properties['int']      = 50 + i;
    custom_properties['float']    = 3.14159 + i;
    custom_properties['even']     = (i % 2 == 0);
    custom_properties;
  end

  def ScenarioHelper.create_issue_message(issue_id, issue_body, label)
    message = Azure::ServiceBus::BrokeredMessage.new(issue_body)

    message.correlation_id  = "correlationid" + label
    message.delivery_count  = 1
    message.sequence_number = 12
    message.content_type    = 'text/xml'
    message.label           = label
    message.to              = '1@1.com'
    message.message_id      = issue_id

    custom_properties = ScenarioHelper.get_custom_properties issue_id.to_i
    custom_properties.each { |key, value|
      message.properties[key] = value
    }

    message
  end

  def ScenarioHelper.compare_messages(expected, actual, custom_properties = nil)
    actual.body.must_equal                       expected.body
    actual.content_type.must_equal               expected.content_type
    actual.correlation_id.must_equal             expected.correlation_id
    actual.enqueued_time_utc.must_equal          expected.enqueued_time_utc
    actual.label.must_equal                      expected.label
    actual.message_id.must_equal                 expected.message_id
    actual.reply_to.must_equal                   expected.reply_to
    actual.reply_to_session_id.must_equal        expected.reply_to_session_id
    actual.scheduled_enqueue_time_utc.must_equal expected.scheduled_enqueue_time_utc
    actual.session_id.must_equal                 expected.session_id
    actual.to.must_equal                         expected.to

    # Note: The following properties are controlled by the server,
    # so we cannot verify as much
    actual.delivery_count.must_be :kind_of?, Integer
    actual.sequence_number.must_be :kind_of?, Integer
    if actual.lock_token != nil
      actual.lock_token.must_be :kind_of?, String
    end
    if actual.locked_until_utc != nil
      actual.locked_until_utc.must_be :kind_of?, Time
    end

    if custom_properties == nil
      expected_properties = expected.properties;
      index = expected_properties["i"];
      if (index == nil)
        index = 1;
      end
      custom_properties = get_custom_properties(index.to_i);
    end

    custom_properties.each { |key, value|
      ScenarioHelper.out key + ":" + value.to_s
      ScenarioHelper.out key.downcase + ":" + (actual.properties[key.downcase] == nil ? "<nil>" : actual.properties[key.downcase].to_s)
      if value != "GUID"
        if value != nil && value.class == Time
          value.httpdate.must_equal actual.properties[key.downcase]
        else
          value.must_equal actual.properties[key.downcase]
        end
      end
    }
  end
end