#include <talloc.h>

#include <tevent.h>

#include <errno.h>

#include <popt.h>

#include <security/pam_modules.h>

#include "providers/backend.h"

#include "providers/data_provider/dp_private.h"

#include "providers/data_provider/dp_builtin.h"

#include "providers/data_provider/dp.h"

#include "tests/cmocka/common_mock.h"

#include "tests/common.h"

#include "tests/cmocka/common_mock_be.h"

#include "tests/cmocka/data_provider/mock_dp.h"

#define TESTS_PATH "tp_" BASE_FILE_STEM

#define TEST_CONF_DB "test_dp_request.ldb"

#define TEST_DOM_NAME "dp_request_test"

#define TEST_ID_PROVIDER "ldap"

struct test_ctx {

struct sss_test_ctx *tctx;

struct be_ctx *be_ctx;

struct dp_req_params *params;

};

static int test_setup(void **state)

{

struct test_ctx *test_ctx;

assert_true(leak_check_setup());

test_ctx = talloc_zero(global_talloc_context, struct test_ctx);

assert_non_null(test_ctx);

test_ctx->tctx = create_dom_test_ctx(test_ctx, TESTS_PATH, TEST_CONF_DB,

TEST_DOM_NAME, TEST_ID_PROVIDER, NULL);

assert_non_null(test_ctx->tctx);

test_ctx->be_ctx = mock_be_ctx(test_ctx, test_ctx->tctx);

assert_non_null(test_ctx->be_ctx);

test_ctx->params = mock_dp_req_params(test_ctx, test_ctx->be_ctx->ev,

test_ctx->be_ctx, NULL,

DPT_ID, DPM_ACCOUNT_HANDLER);

assert_non_null(test_ctx->params);

check_leaks_push(test_ctx);

*state = test_ctx;

return 0;

}

static int test_teardown(void **state)

{

struct test_ctx *test_ctx;

test_ctx = talloc_get_type_abort(*state, struct test_ctx);

assert_true(check_leaks_pop(test_ctx));

talloc_zfree(test_ctx);

assert_true(leak_check_teardown());

return 0;

}

static void test_deny_handler(void **state)

{

errno_t ret;

struct test_ctx *test_ctx;

struct tevent_req *req;

struct pam_data *pd;

struct pam_data *out_pd;

test_ctx = talloc_get_type(*state, struct test_ctx);

pd = talloc_zero(test_ctx, struct pam_data);

assert_non_null(pd);

req = dp_access_deny_handler_send(test_ctx, NULL, pd, test_ctx->params);

assert_non_null(req);

tevent_loop_wait(test_ctx->tctx->ev);

ret = dp_access_deny_handler_recv(test_ctx, req, &out_pd);

assert_int_equal(ret, EOK);

assert_ptr_equal(pd, out_pd);

assert_int_equal(pd->pam_status, PAM_PERM_DENIED);

talloc_free(req);

talloc_free(pd);

}

static void test_permit_handler(void **state)

{

errno_t ret;

struct test_ctx *test_ctx;

struct tevent_req *req;

struct pam_data *pd;

struct pam_data *out_pd;

test_ctx = talloc_get_type(*state, struct test_ctx);

pd = talloc_zero(test_ctx, struct pam_data);

assert_non_null(pd);

req = dp_access_permit_handler_send(test_ctx, NULL, pd, test_ctx->params);

assert_non_null(req);

tevent_loop_wait(test_ctx->tctx->ev);

ret = dp_access_permit_handler_recv(test_ctx, req, &out_pd);

assert_int_equal(ret, EOK);

assert_ptr_equal(pd, out_pd);

assert_int_equal(pd->pam_status, PAM_SUCCESS);

talloc_free(req);

talloc_free(pd);

}

int main(int argc, const char *argv[])

{

poptContext pc;

int opt;

int rv;

int no_cleanup = 0;

struct poptOption long_options[] = {

POPT_AUTOHELP

SSSD_DEBUG_OPTS

{"no-cleanup", 'n', POPT_ARG_NONE, &no_cleanup, 0,

_("Do not delete the test database after a test run"), NULL },

POPT_TABLEEND

};

const struct CMUnitTest tests[] = {

cmocka_unit_test_setup_teardown(test_permit_handler,

test_setup,

test_teardown),

cmocka_unit_test_setup_teardown(test_deny_handler,

test_setup,

test_teardown),

};

/* Set debug level to invalid value so we can deside if -d 0 was used. */

debug_level = SSSDBG_INVALID;

pc = poptGetContext(argv[0], argc, argv, long_options, 0);

while((opt = poptGetNextOpt(pc)) != -1) {

switch(opt) {

default:

fprintf(stderr, "\nInvalid option %s: %s\n\n",

poptBadOption(pc, 0), poptStrerror(opt));

poptPrintUsage(pc, stderr, 0);

return 1;

}

}

poptFreeContext(pc);

DEBUG_CLI_INIT(debug_level);

/* Even though normally the tests should clean up after themselves

tests_set_cwd();

test_dom_suite_cleanup(TESTS_PATH, TEST_CONF_DB, TEST_DOM_NAME);

test_dom_suite_setup(TESTS_PATH);

rv = cmocka_run_group_tests(tests, NULL, NULL);

if (rv == 0 && !no_cleanup) {

test_dom_suite_cleanup(TESTS_PATH, TEST_CONF_DB, TEST_DOM_NAME);

}

return rv;

}