92 lines
3.2 KiB
OCaml
92 lines
3.2 KiB
OCaml
|
(**************************************************************************)
|
||
|
|
(* ocplib-resto *)
|
||
|
|
(* Copyright (C) 2016, OCamlPro. *)
|
||
|
|
(* *)
|
||
|
|
(* All rights reserved. This file is distributed under the terms *)
|
||
|
|
(* of the GNU Lesser General Public License version 2.1, with the *)
|
||
|
|
(* special exception on linking described in the file LICENSE. *)
|
||
|
|
(* *)
|
||
|
|
(**************************************************************************)
|
||
|
|
|
||
|
|
open EzServices
|
||
|
|
open EzResto
|
||
|
|
open EzDirectory
|
||
|
|
open Lwt.Infix
|
||
|
|
|
||
|
|
let () =
|
||
|
|
Lwt_main.run begin
|
||
|
|
allowed_methods dir ["foo";"3";"repeat"] >>= function
|
||
|
|
| Ok [`POST] -> Lwt.return_unit
|
||
|
|
| _ -> assert false
|
||
|
|
end
|
||
|
|
|
||
|
|
let () =
|
||
|
|
Lwt_main.run begin
|
||
|
|
allowed_methods dir ["bar";"3";"4";"add"] >>= function
|
||
|
|
| Ok [`GET;`POST] -> Lwt.return_unit
|
||
|
|
| _ -> assert false
|
||
|
|
end
|
||
|
|
|
||
|
|
module Test(Request : sig
|
||
|
|
val request:
|
||
|
|
('meth, 'params, 'query, 'input, 'output, 'error) EzResto.service ->
|
||
|
|
'params -> 'query -> 'input -> [> ('output, 'error) Answer.t ] Lwt.t
|
||
|
|
end) = struct
|
||
|
|
|
||
|
|
let () =
|
||
|
|
Lwt_main.run begin
|
||
|
|
Request.request describe_service ((), []) { Description.recurse = true } () >>= function
|
||
|
|
| `Ok dir ->
|
||
|
|
Format.printf "@[<v>%a@]@." Resto.Description.pp_print_directory dir ;
|
||
|
|
Lwt.return_unit
|
||
|
|
| _ -> assert false
|
||
|
|
end
|
||
|
|
|
||
|
|
let () =
|
||
|
|
let test service args arg expected =
|
||
|
|
Lwt_main.run (Request.request service args () arg) = `Ok expected in
|
||
|
|
assert (test repeat_service ((), 3) (`A []) (`A (repeat 3 (`A [])))) ;
|
||
|
|
assert (test add_service ((), 2) 3 5) ;
|
||
|
|
assert (test alternate_add_service (((), 1), 2.5) () 3.5) ;
|
||
|
|
assert (test alternate_add_service' (1, 2) () 3) ;
|
||
|
|
()
|
||
|
|
end
|
||
|
|
|
||
|
|
|
||
|
|
module Faked = Test(struct
|
||
|
|
(** Testing faked client/server communication. *)
|
||
|
|
let request (type i) (service: (_,_,_,i,_,_) service) params query (arg: i) =
|
||
|
|
let { meth ; path ; query ; input } = forge_request service params query in
|
||
|
|
let uri =
|
||
|
|
Uri.make
|
||
|
|
~path:(String.concat "/" path)
|
||
|
|
~query:(List.map (fun (k,v) -> k, [v]) query) () in
|
||
|
|
Format.eprintf "\nREQUEST: %a@." Uri.pp_hum uri ;
|
||
|
|
let json =
|
||
|
|
match input with
|
||
|
|
| No_input -> `O []
|
||
|
|
| Input input -> Json_encoding.construct input arg in
|
||
|
|
lookup dir meth path >>= function
|
||
|
|
| Ok (Service s) -> begin
|
||
|
|
let query = Resto.Query.parse s.types.query query in
|
||
|
|
begin
|
||
|
|
match s.types.input with
|
||
|
|
| No_input -> s.handler query ()
|
||
|
|
| Input input ->
|
||
|
|
s.handler query @@ Json_encoding.destruct input json
|
||
|
|
end >>= function
|
||
|
|
| `Ok res ->
|
||
|
|
let json = Json_encoding.construct s.types.output res in
|
||
|
|
Lwt.return (`Ok (Json_encoding.destruct (output_encoding service) json))
|
||
|
|
| _ -> failwith "Unexpected lwt result (1)"
|
||
|
|
end
|
||
|
|
| _ -> failwith "Unexpected lwt result (2)"
|
||
|
|
end)
|
||
|
|
|
||
|
|
module Transparent = Test(struct
|
||
|
|
let request x = transparent_lookup dir x
|
||
|
|
end)
|
||
|
|
|
||
|
|
let () =
|
||
|
|
Printf.printf "\n### OK EzResto ###\n\n%!"
|