/
case_template.ex
90 lines (70 loc) · 1.9 KB
/
case_template.ex
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
defmodule ExUnit.CaseTemplate do
@moduledoc """
Defines a module template to be used throughout your test suite.
This is useful when there are a set of setup callbacks or a set
of functions that should be shared between test modules.
Once a case template is used, the regular functionality in
`ExUnit.Case` plus the functionality defined in the template
will become available.
## Example
defmodule MyCase do
use ExUnit.CaseTemplate
setup do
IO.puts("This will run before each test that uses this case")
end
end
defmodule MyTest do
use MyCase, async: true
test "truth" do
assert true
end
end
"""
@doc false
defmacro __using__(_) do
quote do
use ExUnit.Callbacks
import ExUnit.Assertions
import unquote(__MODULE__)
defmacro __using__(opts) do
unquote(__MODULE__).__proxy__(__MODULE__, opts)
end
defoverridable __using__: 1
end
end
@doc false
def __proxy__(module, opts) do
quote do
use ExUnit.Case, unquote(opts)
setup_all context do
unquote(module).__ex_unit__(:setup_all, context)
end
setup context do
unquote(module).__ex_unit__(:setup, context)
end
end
end
@doc """
Allows a developer to customize the using block
when the case template is used.
## Example
defmodule MyCase do
use ExUnit.CaseTemplate
using do
quote do
# This code is injected into every case that calls "use MyCase"
alias MyApp.FunModule
end
end
end
"""
defmacro using(var \\ quote(do: _), do: block) do
quote do
defmacro __using__(unquote(var) = opts) do
parent = unquote(__MODULE__).__proxy__(__MODULE__, opts)
result = unquote(block)
{:__block__, [], [parent, result]}
end
end
end
end