合并
石头 authored at 2024-11-24 11:23:32
2.13 KiB
X
using System;
using System.Threading;
using NewLife;
using NewLife.Log;
using NewLife.Messaging;
using NewLife.Remoting;
using NewLife.Security;
using NewLife.Serialization;
using Xunit;

namespace XUnitTest.Remoting
{
    /// <summary>RPC下行指令测试</summary>
    public class ApiDownTests : DisposeBase
    {
        private readonly ApiServer _Server;
        private readonly MyClient _Client;
        private readonly String _Uri;

        public ApiDownTests()
        {
            var port = Rand.Next(10000, 65535);

            _Server = new ApiServer(port)
            {
                Log = XTrace.Log,
                //EncoderLog = XTrace.Log,
                ShowError = true,
            };
            _Server.Start();

            _Uri = $"tcp://127.0.0.1:{port}";

            var client = new MyClient()
            {
                Servers = new[] { _Uri },
                //Log = XTrace.Log
            };
            //client.EncoderLog = XTrace.Log;
            _Client = client;
        }

        protected override void Dispose(Boolean disposing)
        {
            base.Dispose(disposing);

            _Server.TryDispose();
        }

        [Fact]
        public async void Test1()
        {
            var apis = await _Client.InvokeAsync<String[]>("api/all");
            Assert.NotNull(apis);

            // 服务端主动下发
            var ss = _Server.Server.AllSessions[0];
            var args = new { name = "Stone", age = 36 };
            ss.InvokeOneWay("CustomCommand", args);

            Thread.Sleep(200);

            var msg = _Client.LastMessage;
            Assert.NotNull(msg);

            // 解码消息
            var rs = _Client.Encoder.Decode(msg, out var action, out var code, out var packet);
            Assert.True(rs);
            Assert.Equal("CustomCommand", action);
            Assert.Equal(0, code);
            Assert.Equal(JsonHelper.ToJson(args), packet.ToStr());
        }

        private class MyClient : ApiClient
        {
            public IMessage LastMessage { get; set; }

            protected override void OnReceive(IMessage message) => LastMessage = message;
        }
    }
}