Producer_ProduceAsync_Null_Task.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.

#pragma warning disable xUnit1026

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Xunit;


namespace Confluent.Kafka.IntegrationTests
{
    /// <summary>
    ///     Test of every <see cref="Producer{TKey,TValue}.ProduceAsync" /> 
    ///     and <see cref="Producer.ProduceAsync" /> method overload.
    ///     (null key/value case)
    /// </summary>
    public partial class Tests
    {
        [Theory, MemberData(nameof(KafkaParameters))]
        public void Producer_ProduceAsync_Null_Task(string bootstrapServers)
        {
            LogToFile("start Producer_ProduceAsync_Null_Task");

            var producerConfig = new ProducerConfig { BootstrapServers = bootstrapServers };


            // serialize case

            var drs = new List<Task<DeliveryResult<Null, Null>>>();
            using (var producer = new ProducerBuilder<Null, Null>(producerConfig).Build())
            {
                drs.Add(producer.ProduceAsync(
                    new TopicPartition(partitionedTopic, 0), new Message<Null, Null> {}));
                drs.Add(producer.ProduceAsync(partitionedTopic, new Message<Null, Null> {}));
                Assert.Equal(0, producer.Flush(TimeSpan.FromSeconds(10)));
            }

            for (int i=0; i<2; ++i)
            {
                var dr = drs[i].Result;
                Assert.True(dr.Partition == 0 || dr.Partition == 1);
                Assert.Equal(partitionedTopic, dr.Topic);
                Assert.True(dr.Offset >= 0);
                Assert.Null(dr.Message.Key);
                Assert.Null(dr.Message.Value);
                Assert.Equal(TimestampType.CreateTime, dr.Message.Timestamp.Type);
                Assert.True(Math.Abs((DateTime.UtcNow - dr.Message.Timestamp.UtcDateTime).TotalMinutes) < 1.0);
            }

            Assert.Equal((Partition)0, drs[0].Result.Partition);
            

            // byte[] case
        
            var drs2 = new List<Task<DeliveryResult<byte[], byte[]>>>();
            using (var producer = new ProducerBuilder<byte[], byte[]>(producerConfig).Build())
            {
                drs2.Add(producer.ProduceAsync(new TopicPartition(partitionedTopic, 1), new Message<byte[], byte[]> {}));
                drs2.Add(producer.ProduceAsync(partitionedTopic, new Message<byte[], byte[]> {}));
                Assert.Equal(0, producer.Flush(TimeSpan.FromSeconds(10)));
            }

            for (int i=0; i<2; ++i)
            {
                var dr = drs2[i].Result;
                Assert.True(dr.Partition == 0 || dr.Partition == 1);
                Assert.Equal(partitionedTopic, dr.Topic);
                Assert.True(dr.Offset >= 0);
                Assert.Null(dr.Message.Key);
                Assert.Null(dr.Message.Value);
                Assert.Equal(TimestampType.CreateTime, dr.Message.Timestamp.Type);
                Assert.True(Math.Abs((DateTime.UtcNow - dr.Message.Timestamp.UtcDateTime).TotalMinutes) < 1.0);
            }
            
            Assert.Equal((Partition)1, drs2[0].Result.Partition);
            

            Assert.Equal(0, Library.HandleCount);
            LogToFile("end   Producer_ProduceAsync_Null_Task");
        }

    }
}