24
1
mirror of https://github.com/processone/ejabberd.git synced 2024-06-28 22:52:27 +02:00

Update test after roster code reorganization

This commit is contained in:
Paweł Chmielowski 2016-05-31 00:09:26 +02:00
parent 622bff23a4
commit 8e883a76e3

View File

@ -45,41 +45,32 @@ defmodule ModRosterMock do
try do try do
module_mock = :moka.start(module) module_mock = :moka.start(module)
:moka.replace(module_mock, :mod_roster, :invalidate_roster_cache, :moka.replace(module_mock, :mod_roster_mnesia, :invalidate_roster_cache,
fn (_user, _server) -> fn (_user, _server) ->
:ok :ok
end) end)
:moka.load(module_mock) :moka.load(module_mock)
roster_mock = :moka.start(:mod_roster) roster_mock0 = :moka.start(:mod_roster)
:moka.replace(roster_mock0, :gen_iq_handler, :add_iq_handler,
:moka.replace(roster_mock, :gen_mod, :db_type,
fn (_host, _opts) ->
{:none}
end)
:moka.replace(roster_mock, :gen_iq_handler, :add_iq_handler,
fn (_module, _host, _ns, _m, _f, _iqdisc) -> fn (_module, _host, _ns, _m, _f, _iqdisc) ->
:ok :ok
end) end)
:moka.replace(roster_mock, :gen_iq_handler, :remove_iq_handler, :moka.replace(roster_mock0, :gen_iq_handler, :remove_iq_handler,
fn (_module, _host, _ns) -> fn (_module, _host, _ns) ->
:ok :ok
end) end)
:moka.replace(roster_mock0, :gen_mod, :db_mod,
:moka.replace(roster_mock, :transaction, fn (_host, _mod) ->
fn (_server, function) -> :mod_roster_mnesia
{:atomic, function.()}
end) end)
:moka.replace(roster_mock0, :gen_mod, :db_mod,
:moka.replace(roster_mock, :get_roster, fn (_host, _opts, _mod) ->
fn (user, domain) -> :mod_roster_mnesia
to_records(get_roster(user, domain))
end) end)
:moka.replace(roster_mock0, :update_roster_t,
:moka.replace(roster_mock, :update_roster_t,
fn (user, domain, {u, d, _r}, item) -> fn (user, domain, {u, d, _r}, item) ->
add_roster_item(user, domain, u<>"@"<>d, add_roster_item(user, domain, u<>"@"<>d,
roster(item, :name), roster(item, :name),
@ -89,10 +80,32 @@ defmodule ModRosterMock do
roster(item, :askmessage)) roster(item, :askmessage))
end) end)
:moka.replace(roster_mock, :del_roster_t, :moka.replace(roster_mock0, :del_roster_t,
fn (user, domain, jid) -> fn (user, domain, jid) ->
remove_roster_item(user, domain, :jid.to_string(jid)) remove_roster_item(user, domain, :jid.to_string(jid))
end) end)
:moka.replace(roster_mock0, :get_roster,
fn (user, domain) ->
to_records(get_roster(user, domain))
end)
:moka.load(roster_mock0)
roster_mock = :moka.start(:mod_roster_mnesia)
:moka.replace(roster_mock, :gen_mod, :db_type,
fn (_host, _opts) ->
{:none}
end)
:moka.replace(roster_mock, :transaction,
fn (_server, function) ->
{:atomic, function.()}
end)
:moka.replace(roster_mock, :update_tables,
fn () ->
:ok
end)
:moka.load(roster_mock) :moka.load(roster_mock)
@ -104,7 +117,7 @@ defmodule ModRosterMock do
def mock_with_meck do def mock_with_meck do
# mock(:gen_mod, :db_type, # mock(:gen_mod, :db_type,
# fn (_server, :mod_roster) -> # fn (_server, :mod_roster_mnesia) ->
# :mnesia # :mnesia
# end) # end)
# #
@ -119,12 +132,16 @@ defmodule ModRosterMock do
# {:atomic, :ok} # {:atomic, :ok}
# end) # end)
mock(:mod_roster, :transaction, mock(:mod_roster_mnesia, :init,
fn (_server, _opts) ->
:ok
end)
mock(:mod_roster_mnesia, :transaction,
fn (_server, function) -> fn (_server, function) ->
{:atomic, function.()} {:atomic, function.()}
end) end)
mock(:mod_roster, :update_roster_t, mock(:mod_roster_mnesia, :update_roster_t,
fn (user, domain, {u, d, _r}, item) -> fn (user, domain, {u, d, _r}, item) ->
add_roster_item(user, domain, u<>"@"<>d, add_roster_item(user, domain, u<>"@"<>d,
roster(item, :name), roster(item, :name),
@ -134,7 +151,7 @@ defmodule ModRosterMock do
roster(item, :askmessage)) roster(item, :askmessage))
end) end)
mock(:mod_roster, :invalidate_roster_cache, mock(:mod_roster_mnesia, :invalidate_roster_cache,
fn (_user, _server) -> fn (_user, _server) ->
:ok :ok
end) end)