test/gen_stage/demand_dispatcher_test.exs

15     disp = dispatcher([])
17     {:ok, 0, disp} = D.subscribe([], {pid, ref}, disp)
18     assert disp == {[{0, pid, ref}], 0, nil}
20     {:ok, 0, disp} = D.cancel({pid, ref}, disp)
21     assert disp == {[], 0, nil}
27     disp = dispatcher([])
30     {:ok, 0, disp} = D.subscribe([], {pid, ref}, disp)
31     assert disp == {[{0, pid, ref}], 0, nil}
33     {:ok, 10, disp} = D.ask(10, {pid, ref}, disp)
34     assert disp == {[{10, pid, ref}], 0, 10}
36     {:ok, 0, disp} = D.cancel({pid, ref}, disp)
37     assert disp == {[], 10, 10}
40     {:ok, 0, disp} = D.subscribe([], {pid, ref}, disp)
41     assert disp == {[{0, pid, ref}], 10, 10}
43     {:ok, 0, disp} = D.ask(5, {pid, ref}, disp)
44     assert disp == {[{5, pid, ref}], 5, 10}
46     {:ok, 0, disp} = D.cancel({pid, ref}, disp)
47     assert disp == {[], 10, 10}
53     disp = dispatcher([])
54     {:ok, 0, disp} = D.subscribe([], {pid, ref}, disp)
56     {:ok, 3, disp} = D.ask(3, {pid, ref}, disp)
57     assert disp == {[{3, pid, ref}], 0, 3}
59     {:ok, [], disp} = D.dispatch([:a], 1, disp)
60     assert disp == {[{2, pid, ref}], 0, 3}
63     {:ok, 3, disp} = D.ask(3, {pid, ref}, disp)
64     assert disp == {[{5, pid, ref}], 0, 3}
66     {:ok, [:g, :h], disp} = D.dispatch([:b, :c, :d, :e, :f, :g, :h], 7, disp)
67     assert disp == {[{0, pid, ref}], 0, 3}
70     {:ok, [:i, :j], disp} = D.dispatch([:i, :j], 2, disp)
71     assert disp == {[{0, pid, ref}], 0, 3}
80     disp = dispatcher([])
82     {:ok, 0, disp} = D.subscribe([], {pid, ref1}, disp)
83     {:ok, 0, disp} = D.subscribe([], {pid, ref2}, disp)
84     {:ok, 0, disp} = D.subscribe([], {pid, ref3}, disp)
86     {:ok, 4, disp} = D.ask(4, {pid, ref1}, disp)
87     {:ok, 2, disp} = D.ask(2, {pid, ref2}, disp)
88     {:ok, 3, disp} = D.ask(3, {pid, ref3}, disp)
89     assert disp == {[{4, pid, ref1}, {3, pid, ref3}, {2, pid, ref2}], 0, 4}
91     {:ok, 2, disp} = D.ask(2, {pid, ref3}, disp)
92     assert disp == {[{5, pid, ref3}, {4, pid, ref1}, {2, pid, ref2}], 0, 4}
94     {:ok, 4, disp} = D.ask(4, {pid, ref2}, disp)
95     assert disp == {[{6, pid, ref2}, {5, pid, ref3}, {4, pid, ref1}], 0, 4}
102     disp = dispatcher([])
104     {:ok, 0, disp} = D.subscribe([], {pid, ref1}, disp)
105     {:ok, 0, disp} = D.subscribe([], {pid, ref2}, disp)
107     {:ok, 3, disp} = D.ask(3, {pid, ref1}, disp)
108     {:ok, 2, disp} = D.ask(2, {pid, ref2}, disp)
109     assert disp == {[{3, pid, ref1}, {2, pid, ref2}], 0, 3}
112     {:ok, [], disp} = D.dispatch([:a, :b, :c, :d, :e], 5, disp)
113     assert disp == {[{0, pid, ref1}, {0, pid, ref2}], 0, 3}
117     {:ok, [:a, :b, :c], disp} = D.dispatch([:a, :b, :c], 3, disp)
118     assert disp == {[{0, pid, ref1}, {0, pid, ref2}], 0, 3}
122     {:ok, 3, disp} = D.ask(3, {pid, ref1}, disp)
123     {:ok, 3, disp} = D.ask(3, {pid, ref2}, disp)
124     assert disp == {[{3, pid, ref1}, {3, pid, ref2}], 0, 3}
126     {:ok, [], disp} = D.dispatch([:a, :b], 2, disp)
127     assert disp == {[{3, pid, ref2}, {1, pid, ref1}], 0, 3}
130     {:ok, [], disp} = D.dispatch([:c, :d], 2, disp)
131     assert disp == {[{1, pid, ref1}, {1, pid, ref2}], 0, 3}
135     {:ok, [:g], disp} = D.dispatch([:e, :f, :g], 3, disp)
136     assert disp == {[{0, pid, ref1}, {0, pid, ref2}], 0, 3}
145     disp = dispatcher([])
147     {:ok, 0, disp} = D.subscribe([], {pid, ref1}, disp)
148     {:ok, 0, disp} = D.subscribe([], {pid, ref2}, disp)
149     {:ok, 3, disp} = D.ask(3, {pid, ref1}, disp)
151     {:ok, notify_disp} = D.info(:hello, disp)
152     assert disp == notify_disp
160     disp = dispatcher([])
162     {:ok, 0, disp} = D.subscribe([], {pid, ref1}, disp)
163     {:ok, 0, disp} = D.subscribe([], {pid, ref2}, disp)
167         {:ok, 3, disp} = D.ask(3, {pid, ref1}, disp)
168         {:ok, 4, disp} = D.ask(4, {pid, ref2}, disp)
169         disp