﻿/**
 * 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.
 */

namespace Kafka.Client.Tests.Producers
{
    using Kafka.Client.Cfg;
    using Kafka.Client.Cluster;
    using Kafka.Client.Producers.Partitioning;
    using NUnit.Framework;
    using System.Collections.Generic;

    [TestFixture]
    public class PartitioningTests
    {
        private ProducerConfiguration config;

        [TestFixtureSetUp]
        public void SetUp()
        {
            var brokers = new List<BrokerConfiguration>();
            brokers.Add(new BrokerConfiguration { BrokerId = 1, Host = "192.168.0.1", Port = 1234 });
            brokers.Add(new BrokerConfiguration { BrokerId = 2, Host = "192.168.0.2", Port = 3456 });
            config = new ProducerConfiguration(brokers);
        }

        [Test]
        public void BrokerPartitionInfoGetAllBrokerInfoTest()
        {
            IBrokerPartitionInfo brokerPartitionInfo = new ConfigBrokerPartitionInfo(config);
            var allInfo = brokerPartitionInfo.GetAllBrokerInfo();
            this.MakeAssertionsForBroker(allInfo[1], 1, "192.168.0.1", 1234);
            this.MakeAssertionsForBroker(allInfo[2], 2, "192.168.0.2", 3456);
        }

        [Test]
        public void BrokerPartitionInfoGetPartitionInfo()
        {
            IBrokerPartitionInfo brokerPartitionInfo = new ConfigBrokerPartitionInfo(config);
            var broker = brokerPartitionInfo.GetBrokerInfo(1);
            this.MakeAssertionsForBroker(broker, 1, "192.168.0.1", 1234);
        }

        [Test]
        public void BrokerPartitionInfoGetPartitionInfoReturnsNullOnNonexistingBrokerId()
        {
            IBrokerPartitionInfo brokerPartitionInfo = new ConfigBrokerPartitionInfo(config);
            var broker = brokerPartitionInfo.GetBrokerInfo(45);
            Assert.IsNull(broker);
        }

        private void MakeAssertionsForBroker(Broker broker, int expectedId, string expectedHost, int expectedPort)
        {
            Assert.IsNotNull(broker);
            Assert.AreEqual(expectedId, broker.Id);
            Assert.AreEqual(expectedHost, broker.Host);
            Assert.AreEqual(expectedPort, broker.Port);
        }
    }
}