fileserver/lib/restbed-4.8/test/feature/source/http_client/close.cpp

//System Includes
#include <map>
#include <chrono>
#include <thread>
#include <string>
#include <memory>
#include <ciso646>
#include <stdexcept>
#include <functional>

//Project Includes
#include <restbed>

//External Includes
#include <catch.hpp>

//System Namespaces
using std::stoi;
using std::thread;
using std::string;
using std::multimap;
using std::to_string;
using std::shared_ptr;
using std::make_shared;
using std::chrono::milliseconds;

//Project Namespaces
using namespace restbed;

//External Namespaces

void get_handler( const shared_ptr< Session > session )
{
    session->sleep_for( milliseconds( 5 ), [ ]( const shared_ptr< Session > session )
    {
        session->close( 200, { { "Connection", "close" } } );
    } );
}

SCENARIO( "close active request", "[client]" )
{
    auto resource = make_shared< Resource >( );
    resource->set_path( "/resource" );
    resource->set_method_handler( "GET", get_handler );
    
    auto settings = make_shared< Settings >( );
    settings->set_port( 1984 );
    
    shared_ptr< thread > worker = nullptr;
    
    Service service;
    service.publish( resource );
    service.set_ready_handler( [ &worker ]( Service & service )
    {
        worker = make_shared< thread >( [ &service ] ( )
        {
            GIVEN( "I have started a service and perform a HTTP 'GET' request" )
            {
                auto request = make_shared< Request >( );
                request->set_port( 1984 );
                request->set_host( "localhost" );
                request->set_method( "GET" );
                request->set_path( "/resource" );
                
                auto response = Http::sync( request );
                
                WHEN( "I close the connection before seeing a response" )
                {
                    Http::close( request );
                    
                    THEN( "I should see the connection closed" )
                    {
                        REQUIRE( Http::is_open( request ) == false );
                        REQUIRE( Http::is_closed( request ) == true );
                    }
                }
                
                service.stop( );
            }
        } );
    } );
    
    service.start( settings );
    worker->join( );
}

SCENARIO( "close inactive request", "[client]" )
{
    GIVEN( "I have not started a service and not perform a HTTP 'GET' request" )
    {
        auto request = make_shared< Request >( );
        request->set_port( 1984 );
        request->set_host( "localhost" );
        request->set_method( "GET" );
        request->set_path( "/resource" );
        
        WHEN( "I close the request socket" )
        {
            Http::close( request );
            
            THEN( "I should see the connection closed" )
            {
                REQUIRE( Http::is_open( request ) == false );
                REQUIRE( Http::is_closed( request ) == true );
            }
        }
    }
}

SCENARIO( "close null request", "[client]" )
{
    GIVEN( "I have a null request" )
    {
        WHEN( "I close the request socket" )
        {
            shared_ptr< Request > request = nullptr;
            
            Http::close( request );
            
            THEN( "I should see the connection closed" )
            {
                REQUIRE( Http::is_open( request ) == false );
                REQUIRE( Http::is_closed( request ) == true );
            }
        }
    }
}