/
case_template.ex
89 lines (70 loc) · 1.89 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
defmodule ExUnit.CaseTemplate do
@moduledoc """
This module allows a developer to define a test case
template to be used throughout their tests. This is useful
when there are a set of functions that should be shared
between tests or a set of setup callbacks.
By using this module, the callbacks and assertions
available for regular test cases will also be 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