pfff/h_program-lang/test_program_lang.ml at master · facebookarchive/pfff · GitHub

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
open Common

module Db = Database_code
module E = Entity_code

(*****************************************************************************)
(* Helpers *)
(*****************************************************************************)

(*****************************************************************************)
(* Subsystem testing *)
(*****************************************************************************)

let test_load_light_db file =
  let _db = Db.load_database file in
  ()

let test_big_grep file =
  let db = Db.load_database file in
  let entities =
    Db.files_and_dirs_and_sorted_entities_for_completion
      ~threshold_too_many_entities:300000
      db in
  let idx = Big_grep.build_index entities in
  let query = "old_le" in
  let top_n = 10 in

  let xs = Big_grep.top_n_search ~top_n ~query idx in

  xs +> List.iter (fun e ->
(*
    let json = Db.json_of_entity e in
    let s = Json_io.string_of_json json in
    pr2 s
*)
    pr2_gen e;
  );

  (* naive search *)
  let xs = Big_grep.naive_top_n_search ~top_n ~query entities in
  xs +> List.iter (fun e ->
(*
    let json = Db.json_of_entity e in
    let s = Json_io.string_of_json json in
    pr2 s
*)
    pr2_gen e
  );
  ()

let test_layer file =
  let layer = Layer_code.load_layer file in
  let json = Layer_code.json_of_layer layer in
  let s = Json_out.string_of_json json in
  pr2 s

let layer_stat file =
  let layer = Layer_code.load_layer file in
  let stats = Layer_code.stat_of_layer layer in
  stats +> List.iter (fun (k, v) ->
    pr (spf " %s = %d" k v)
  )

let test_refactoring file =
  let xs = Refactoring_code.load file in
  xs +> List.iter pr2_gen;
  ()


(*****************************************************************************)
(* Main entry for Arg *)
(*****************************************************************************)

let actions () = [
  "-test_load_db",  " <file>",
  Common.mk_action_1_arg test_load_light_db;
  "-test_big_grep", " <file>",
  Common.mk_action_1_arg test_big_grep;
  "-test_layer", " <file>",
  Common.mk_action_1_arg test_layer;
  "-test_refactoring", " <file>",
  Common.mk_action_1_arg test_refactoring;
]