Producer_ProduceAsync_Null_DeliveryHandler.cs

// Copyright 2016-2017 Confluent 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.
//
// Refer to LICENSE for more information.

using System;
using System.Collections.Generic;
using Xunit;


namespace Confluent.Kafka.IntegrationTests
{
    /// <summary>
    ///     Test every Producer.ProduceAsync method overload that provides
    ///     delivery reports via an IDeliveryHandler instance.
    ///     (null key/value case)
    /// </summary>
    public static partial class Tests
    {
        class DeliveryHandler_PN : IDeliveryHandler
        {
            public DeliveryHandler_PN(string topic)
            {
                Topic = topic;
            }

            public bool MarshalData { get { return true; } }

            public int Count { get; private set; }

            public string Topic { get; }

            public void HandleDeliveryReport(Message dr)
            {
                Assert.Equal(ErrorCode.NoError, dr.Error.Code);
                Assert.Equal(0, dr.Partition);
                Assert.Equal(Topic, dr.Topic);
                Assert.True(dr.Offset >= 0);
                Assert.Null(dr.Key);
                Assert.Null(dr.Value);
                Assert.Equal(TimestampType.CreateTime, dr.Timestamp.Type);
                Assert.True(Math.Abs((DateTime.UtcNow - dr.Timestamp.UtcDateTime).TotalMinutes) < 1.0);
                Count += 1;
            }
        }

        [Theory, MemberData(nameof(KafkaParameters))]
        public static void Producer_ProduceAsync_Null_DeliveryHandler(string bootstrapServers, string singlePartitionTopic, string partitionedTopic)
        {
            var producerConfig = new Dictionary<string, object> 
            { 
                { "bootstrap.servers", bootstrapServers },
                { "api.version.request", true }
            };

            var dh = new DeliveryHandler_PN(singlePartitionTopic);

            using (var producer = new Producer(producerConfig))
            {
                producer.ProduceAsync(singlePartitionTopic, null, 0, 0, null, 0, 0, 0, true, dh);
                producer.ProduceAsync(singlePartitionTopic, null, 0, 0, null, 0, 0, 0, dh);
                producer.ProduceAsync(singlePartitionTopic, null, 0, 0, null, 0, 0, true, dh);
                producer.ProduceAsync(singlePartitionTopic, null, 0, 0, null, 0, 0, dh);
                producer.ProduceAsync(singlePartitionTopic, null, null, dh);
                Assert.Throws<ArgumentException>(() => producer.ProduceAsync(singlePartitionTopic, null, -123, int.MinValue, null, int.MaxValue, 44, dh));
                producer.Flush(TimeSpan.FromSeconds(10));
            }

            Assert.Equal(5, dh.Count);
        }
    }
}