25
1
mirror of https://github.com/processone/ejabberd.git synced 2024-11-24 16:23:40 +01:00

Fix tests, they are now running fine locally

This commit is contained in:
Mickael Remond 2016-04-01 11:13:48 +02:00
parent a8f92ae767
commit b5f1479763
6 changed files with 80 additions and 40 deletions

View File

@ -230,16 +230,16 @@ process([Call], #request{method = 'POST', data = Data, ip = IP} = Req) ->
?DEBUG("Bad Request: ~p", [_Err]), ?DEBUG("Bad Request: ~p", [_Err]),
badrequest_response(<<"Invalid JSON input">>); badrequest_response(<<"Invalid JSON input">>);
_:_Error -> _:_Error ->
?DEBUG("Bad Request: ~p ~p", [_Error, erlang:get_stacktrace()]), ?DEBUG("Bad Request: ~p ~p", [_Error, erlang:get_stacktrace()]),
badrequest_response() badrequest_response()
end; end;
process([Call], #request{method = 'GET', q = Data, ip = IP} = Req) -> process([Call], #request{method = 'GET', q = Data, ip = IP} = Req) ->
Version = get_api_version(Req), Version = get_api_version(Req),
try try
Args = case Data of Args = case Data of
[{nokey, <<>>}] -> []; [{nokey, <<>>}] -> [];
_ -> Data _ -> Data
end, end,
log(Call, Args, IP), log(Call, Args, IP),
case check_permissions(Req, Call) of case check_permissions(Req, Call) of
{allowed, Cmd, Auth} -> {allowed, Cmd, Auth} ->

View File

@ -41,7 +41,7 @@ defmodule EjabberdAuthMock do
Agent.get(@agent, fn users -> Map.get(users, {user, domain}, "") end ) Agent.get(@agent, fn users -> Map.get(users, {user, domain}, "") end )
end) end)
mock(:ejabberd_auth, :check_password, mock(:ejabberd_auth, :check_password,
fn (user, authzid, domain, password) -> fn (user, _authzid, domain, password) ->
Agent.get(@agent, fn users -> Agent.get(@agent, fn users ->
Map.get(users, {user, domain}) end) == password Map.get(users, {user, domain}) end) == password
end) end)

View File

@ -18,7 +18,7 @@
# #
# ---------------------------------------------------------------------- # ----------------------------------------------------------------------
defmodule EjabberdCommandsTest do defmodule EjabberdCommandsMockTest do
use ExUnit.Case, async: false use ExUnit.Case, async: false
@author "jsautret@process-one.net" @author "jsautret@process-one.net"
@ -35,8 +35,7 @@ defmodule EjabberdCommandsTest do
@domain "domain" @domain "domain"
require Record require Record
Record.defrecord :ejabberd_commands, Record.extract(:ejabberd_commands, Record.defrecord :ejabberd_commands, Record.extract(:ejabberd_commands, from_lib: "ejabberd/include/ejabberd_commands.hrl")
from: "ejabberd_commands.hrl")
setup_all do setup_all do
try do try do
@ -313,7 +312,7 @@ defmodule EjabberdCommandsTest do
test "API command with admin policy" do test "API command with admin policy" do
mock_commands_config mock_commands_config
# Register a command test(user, domain) -> {user, domain} # Register a command test(user, domain) -> {user, domain}
# with policy=admin # with policy=admin
command_name = :test command_name = :test
@ -354,7 +353,7 @@ defmodule EjabberdCommandsTest do
command_name, command_name,
[@user, @domain]) [@user, @domain])
# An admin with bad password cannot call the command # An admin with bad password cannot call the command
assert {:error, :account_unprivileged} == assert {:error, :account_unprivileged} ==
catch_throw :ejabberd_commands.execute_command(:undefined, catch_throw :ejabberd_commands.execute_command(:undefined,

View File

@ -1,5 +1,22 @@
# ejabberd_oauth mock # ----------------------------------------------------------------------
###################### #
# ejabberd, Copyright (C) 2002-2016 ProcessOne
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; either version 2 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
# ----------------------------------------------------------------------
defmodule EjabberdOauthMock do defmodule EjabberdOauthMock do
@ -17,7 +34,7 @@ defmodule EjabberdOauthMock do
expire = 1000000 * megasecs + secs + expiration expire = 1000000 * megasecs + secs + expiration
:random.seed now :random.seed now
token = to_string :random.uniform(100000000) token = to_string :random.uniform(100000000)
{:ok, _} = :ejabberd_oauth.associate_access_token(token, {:ok, _} = :ejabberd_oauth.associate_access_token(token,
[{"resource_owner", [{"resource_owner",
{:user, user, domain}}, {:user, user, domain}},

View File

@ -1,15 +1,30 @@
# ejabberd_sm mock # ----------------------------------------------------------------------
###################### #
# ejabberd, Copyright (C) 2002-2016 ProcessOne
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; either version 2 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
# ----------------------------------------------------------------------
defmodule EjabberdSmMock do defmodule EjabberdSmMock do
@author "jsautret@process-one.net" @author "jsautret@process-one.net"
require Record require Record
Record.defrecord :session, Record.extract(:session, Record.defrecord :session, Record.extract(:session, from_lib: "ejabberd/include/ejabberd_sm.hrl")
from: "ejabberd_sm.hrl") Record.defrecord :jid, Record.extract(:jid, from_lib: "ejabberd/include/jlib.hrl")
Record.defrecord :jid, Record.extract(:jid,
from: "jlib.hrl")
@agent __MODULE__ @agent __MODULE__
def init do def init do
@ -20,9 +35,9 @@ defmodule EjabberdSmMock do
catch catch
:exit, _e -> :ok :exit, _e -> :ok
end end
{:ok, _pid} = Agent.start_link(fn -> [] end, name: @agent) {:ok, _pid} = Agent.start_link(fn -> [] end, name: @agent)
mock(:ejabberd_sm, :get_user_resources, mock(:ejabberd_sm, :get_user_resources,
fn (user, domain) -> for s <- get_sessions(user, domain), do: s.resource end) fn (user, domain) -> for s <- get_sessions(user, domain), do: s.resource end)
@ -35,9 +50,9 @@ defmodule EjabberdSmMock do
:ok :ok
(_, _, _) -> :ok (_, _, _) -> :ok
end) end)
end end
def connect_resource(user, domain, resource, def connect_resource(user, domain, resource,
opts \\ [priority: 1, conn: :c2s]) do opts \\ [priority: 1, conn: :c2s]) do
Agent.update(@agent, fn sessions -> Agent.update(@agent, fn sessions ->
@ -54,13 +69,13 @@ defmodule EjabberdSmMock do
end end
def disconnect_resource(user, domain, resource, timestamp) do def disconnect_resource(user, domain, resource, timestamp) do
Agent.update(@agent, fn sessions -> Agent.update(@agent, fn sessions ->
for s <- sessions, for s <- sessions,
s.user != user or s.domain != domain or s.resource != resource, do: s s.user != user or s.domain != domain or s.resource != resource, do: s
end) end)
ModLastMock.set_last user, domain, "", timestamp ModLastMock.set_last user, domain, "", timestamp
end end
def get_sessions() do def get_sessions() do
Agent.get(@agent, fn sessions -> sessions end) Agent.get(@agent, fn sessions -> sessions end)
end end
@ -77,7 +92,7 @@ defmodule EjabberdSmMock do
s.user == user, s.domain == domain, s.resource == resource, do: s s.user == user, s.domain == domain, s.resource == resource, do: s
end) end)
end end
def to_record(s) do def to_record(s) do
session(usr: {s.user, s.domain, s.ressource}, session(usr: {s.user, s.domain, s.ressource},
us: {s.user, s.domain}, us: {s.user, s.domain},
@ -91,7 +106,7 @@ defmodule EjabberdSmMock do
# Helpers # Helpers
#################################################################### ####################################################################
# TODO refactor: Move to ejabberd_test_mock # TODO refactor: Move to ejabberd_test_mock
def mock(module, function, fun) do def mock(module, function, fun) do
try do try do

View File

@ -18,7 +18,7 @@
# #
# ---------------------------------------------------------------------- # ----------------------------------------------------------------------
defmodule ModHttpApiTest do defmodule ModHttpApiMockTest do
use ExUnit.Case, async: false use ExUnit.Case, async: false
@author "jsautret@process-one.net" @author "jsautret@process-one.net"
@ -36,14 +36,17 @@ defmodule ModHttpApiTest do
@acommand String.to_atom(@command) @acommand String.to_atom(@command)
# default API version # default API version
@version 0 @version 0
require Record require Record
Record.defrecord :request, Record.extract(:request, Record.defrecord :request, Record.extract(:request, from_lib: "ejabberd/include/ejabberd_http.hrl")
from: "ejabberd_http.hrl")
setup_all do setup_all do
try do try do
:jid.start
:mnesia.start
:stringprep.start :stringprep.start
:ejabberd_config.start([@domain], [])
:ejabberd_commands.init
rescue rescue
_ -> :ok _ -> :ok
end end
@ -67,13 +70,16 @@ defmodule ModHttpApiTest do
fn (@acommand, {@user, @domain, @userpass, false}, @version) -> fn (@acommand, {@user, @domain, @userpass, false}, @version) ->
{[], {:res, :rescode}} {[], {:res, :rescode}}
end) end)
:meck.expect(:ejabberd_commands, :get_command_policy,
fn (@acommand) -> {:ok, :user} end)
:meck.expect(:ejabberd_commands, :get_commands,
fn () -> [@acommand] end)
:meck.expect(:ejabberd_commands, :execute_command, :meck.expect(:ejabberd_commands, :execute_command,
fn (:undefined, {@user, @domain, @userpass, false}, @acommand, [], @version) -> fn (:undefined, {@user, @domain, @userpass, false}, @acommand, [], @version) ->
:ok :ok
end) end)
#:ejabberd_logger.start :ejabberd_config.add_local_option(:commands, [[{:add_commands, [@acommand]}]])
#:ejabberd_logger.set 5
# Correct Basic Auth call # Correct Basic Auth call
req = request(method: :GET, req = request(method: :GET,
@ -84,6 +90,9 @@ defmodule ModHttpApiTest do
ip: {{127,0,0,1},60000}, ip: {{127,0,0,1},60000},
host: @domain) host: @domain)
result = :mod_http_api.process([@command], req) result = :mod_http_api.process([@command], req)
# history = :meck.history(:ejabberd_commands)
assert 200 == elem(result, 0) # HTTP code assert 200 == elem(result, 0) # HTTP code
assert "0" == elem(result, 2) # command result assert "0" == elem(result, 2) # command result
@ -104,10 +113,8 @@ defmodule ModHttpApiTest do
assert :meck.validate :ejabberd_auth assert :meck.validate :ejabberd_auth
assert :meck.validate :ejabberd_commands assert :meck.validate :ejabberd_commands
#assert :ok = :meck.history(:ejabberd_commands)
end end
test "HTTP GET simple command call with OAuth" do test "HTTP GET simple command call with OAuth" do
EjabberdAuthMock.create_user @user, @domain, @userpass EjabberdAuthMock.create_user @user, @domain, @userpass
@ -116,14 +123,16 @@ defmodule ModHttpApiTest do
fn (@acommand, {@user, @domain, {:oauth, _token}, false}, @version) -> fn (@acommand, {@user, @domain, {:oauth, _token}, false}, @version) ->
{[], {:res, :rescode}} {[], {:res, :rescode}}
end) end)
:meck.expect(:ejabberd_commands, :get_command_policy,
fn (@acommand) -> {:ok, :user} end)
:meck.expect(:ejabberd_commands, :get_commands,
fn () -> [@acommand] end)
:meck.expect(:ejabberd_commands, :execute_command, :meck.expect(:ejabberd_commands, :execute_command,
fn (:undefined, {@user, @domain, {:oauth, _token}, false}, fn (:undefined, {@user, @domain, {:oauth, _token}, false},
@acommand, [], @version) -> @acommand, [], @version) ->
:ok :ok
end) end)
#:ejabberd_logger.start
#:ejabberd_logger.set 5
# Correct OAuth call # Correct OAuth call
token = EjabberdOauthMock.get_token @user, @domain, @command token = EjabberdOauthMock.get_token @user, @domain, @command
@ -184,5 +193,5 @@ defmodule ModHttpApiTest do
#assert :ok = :meck.history(:ejabberd_commands) #assert :ok = :meck.history(:ejabberd_commands)
end end
end end