/tests/MongoDB.Driver.Core.Tests/Core/WireProtocol/Messages/GetMoreMessageTests.cs

/* Copyright 2013-present MongoDB 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.
*/

using System;
using FluentAssertions;
using MongoDB.Bson;
using MongoDB.Driver.Core.WireProtocol.Messages;
using MongoDB.Driver.Core.WireProtocol.Messages.Encoders;
using Moq;
using Xunit;

namespace MongoDB.Driver.Core.WireProtocol.Messages
{
    public class GetMoreMessageTests
    {
        private readonly int _batchSize = 1;
        private readonly CollectionNamespace _collectionNamespace = new CollectionNamespace("database", "collection");
        private readonly long _cursorId = 2;
        private readonly int _requestId = 3;

        [Fact]
        public void Constructor_should_initialize_instance()
        {
            var subject = new GetMoreMessage(_requestId, _collectionNamespace, _cursorId, _batchSize);
            subject.BatchSize.Should().Be(_batchSize);
            subject.CursorId.Should().Be(_cursorId);
            subject.CollectionNamespace.Should().Be(_collectionNamespace);
            subject.RequestId.Should().Be(_requestId);
        }

        [Fact]
        public void Constructor_with_negative_batchSize_should_throw()
        {
            Action action = () => new GetMoreMessage(_requestId, _collectionNamespace, _cursorId, -1);
            action.ShouldThrow<ArgumentOutOfRangeException>();
        }

        [Fact]
        public void Constructor_with_null_collectionNamespace_should_throw()
        {
            Action action = () => new GetMoreMessage(_requestId, null, _cursorId, _batchSize);
            action.ShouldThrow<ArgumentNullException>();
        }

        [Fact]
        public void GetEncoder_should_return_encoder()
        {
            var subject = new GetMoreMessage(_requestId, _collectionNamespace, _cursorId, _batchSize);
            var mockEncoderFactory = new Mock<IMessageEncoderFactory>();
            var encoder = new Mock<IMessageEncoder>().Object;
            mockEncoderFactory.Setup(f => f.GetGetMoreMessageEncoder()).Returns(encoder);

            var result = subject.GetEncoder(mockEncoderFactory.Object);

            result.Should().BeSameAs(encoder);
        }
    }
}