1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
/*
* Copyright 2013-2016, Corvusoft Ltd, All Rights Reserved.
*/
//System Includes
#include <map>
#include <thread>
#include <string>
#include <memory>
#include <ciso646>
#include <utility>
#include <stdexcept>
#include <functional>
//Project Includes
#include <restbed>
//External Includes
#include <catch.hpp>
//System Namespaces
using std::thread;
using std::string;
using std::function;
using std::multimap;
using std::exception;
using std::make_pair;
using std::shared_ptr;
using std::make_shared;
//Project Namespaces
using namespace restbed;
//External Namespaces
void error_handler( const int, const exception&, const shared_ptr< Session > session )
{
session->close( 0, "I see nothing!", { { "Content-Length", "14" } } );
}
void faulty_method_handler( const shared_ptr< Session > )
{
throw SERVICE_UNAVAILABLE;
}
SCENARIO( "custom service error handler", "[service]" )
{
auto resource = make_shared< Resource >( );
resource->set_path( "/resources/1" );
resource->set_method_handler( "GET", faulty_method_handler );
auto settings = make_shared< Settings >( );
settings->set_port( 1984 );
settings->set_default_header( "Connection", "close" );
shared_ptr< thread > worker = nullptr;
Service service;
service.publish( resource );
service.set_error_handler( error_handler );
service.set_ready_handler( [ &worker ]( Service & service )
{
worker = make_shared< thread >( [ &service ] ( )
{
GIVEN( "I publish a faulty resources and set a service error handler" )
{
WHEN( "I perform a HTTP 'GET' request to '/resources/1'" )
{
auto request = make_shared< Request >( );
request->set_port( 1984 );
request->set_host( "localhost" );
request->set_path( "/resources/1" );
auto response = Http::sync( request );
THEN( "I should see a '0' (No Appropriate Status Message Found) status code" )
{
REQUIRE( 0 == response->get_status_code( ) );
REQUIRE( "No Appropriate Status Message Found" == response->get_status_message( ) );
}
AND_THEN( "I should see a response body of 'I see nothing!'" )
{
auto actual = Http::fetch( 14, response );
Bytes expectation { 'I', ' ', 's', 'e', 'e', ' ', 'n', 'o', 't', 'h', 'i', 'n', 'g', '!' };
REQUIRE( actual == expectation );
}
multimap< string, string > headers = response->get_headers( );
AND_THEN( "I should see a 'Connection' header value of 'close'" )
{
auto header = headers.find( "Connection" );
REQUIRE( header not_eq headers.end( ) );
REQUIRE( "close" == headers.find( "Connection" )->second );
}
AND_THEN( "I should see a 'Content-Length' header value of '14'" )
{
auto header = headers.find( "Content-Length" );
REQUIRE( header not_eq headers.end( ) );
REQUIRE( "14" == headers.find( "Content-Length" )->second );
}
}
service.stop( );
}
} );
} );
service.start( settings );
worker->join( );
}
|
