cNetServer.Tests.h

Go to the documentation of this file.

//
//
#ifndef _INC_cNetServerTests_H
#define _INC_cNetServerTests_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "Net/cNetServer.h"
 
namespace GrayLib
{
        class cThread;
        class cUnitTestTelnetFactory;
 
        class cNetServerUnitTest : public cNetServer
        {
        public:
                cNetAddress m_addr;     
        public:
                cNetServerUnitTest(IProtocolFactory& factory, cNetAddress addr);
                bool InitListenPorts();
                bool WaitForClientThread(cThread* pClientThread, TIMESYSD_t nWaitMax = (30 * cTimeSys::k_FREQ));
        };
 
        class cUnitTestTelnetClient
        {
        public:
                cStreamInput* m_pInp;
                cStreamOutput* m_pOut;
        public:
                cUnitTestTelnetClient(cStreamInput* pInp = nullptr, cStreamOutput* pOut = nullptr);
                HRESULT Test();
        };
 
        class cUnitTestTelnetStream : public cRefBase, public cProtocolStream
        {
 
        public:
                static const char* k_pszMsgStop;
        public:
                cUnitTestTelnetStream(cUnitTestTelnetFactory* pFactory, IUnkObject* pServerConnection, cStreamInput* pStreamInp, cStreamOutput* pStreamOut);
                virtual HRESULT ProcessStreamPacket() override;
                IUNKNOWN_DISAMBIG(cRefBase);
        };
 
        class cUnitTestTelnetFactory : public IProtocolFactory
        {
 
        public:
                cUnitTestTelnetFactory()
                {
                }
                virtual const char* get_ProtocolName() const override
                {
                        // IProtocolFactory
                        return "Telnet";
                }
                virtual HRESULT TestProtocol(const BYTE* pszPrefixData, size_t iPrefixLen) const override
                {
                        UNREFERENCED_PARAMETER(pszPrefixData);
                        UNREFERENCED_PARAMETER(iPrefixLen);
                        return S_OK;
                }
                virtual cProtocolStreamPtr CreateProtocolStream(IUnkObject* pServerConnection, cStreamInput* pStreamInp, cStreamOutput* pStreamOut) override
                {
                        return new cUnitTestTelnetStream(this, pServerConnection, pStreamInp, pStreamOut);
                }
        };
}
 
#endif