stats_ut.c

Go to the documentation of this file.

/*
 * Copyright (c) 2013-2020 Seagate Technology LLC and/or its Affiliates
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * For any questions about this software or licensing,
 * please email opensource@seagate.com or cortx-questions@seagate.com.
 *
 */


#include "ut/ut.h"
#include "fop/fop.h"
#include "reqh/reqh.h"
#include "reqh/reqh_service.h"
#include "rpc/rpc_opcodes.h"   /* M0_UT_STATS_OPCODE */

#include "fop/ut/stats/stats_fom.c"
#include "ut/ut_rpc_machine.h"

#define DUMMY_DBNAME      "dummy-db"
#define DUMMY_COB_ID      20
#define DUMMY_SERVER_ADDR "0@lo:12345:34:10"

extern struct m0_fom_type            stats_fom_type;
extern const struct m0_fom_type_ops  fom_stats_type_ops;
static struct m0_ut_rpc_mach_ctx     rmach_ctx;
static struct m0_reqh_service       *service;

static void test_stats(void)
{
        test_stats_req_handle(&rmach_ctx.rmc_reqh);
}

static int ut_stats_service_start(struct m0_reqh_service *service)
{
        M0_PRE(service != NULL);
        return 0;
}

static void ut_stats_service_stop(struct m0_reqh_service *service)
{
        M0_PRE(service != NULL);
}

static void ut_stats_service_fini(struct m0_reqh_service *service)
{
        M0_PRE(service != NULL);
        m0_free(service);
}

static const struct m0_reqh_service_ops ut_stats_service_ops = {
        .rso_start = ut_stats_service_start,
        .rso_stop  = ut_stats_service_stop,
        .rso_fini  = ut_stats_service_fini
};

static int ut_stats_service_allocate(struct m0_reqh_service **service,
                                     const struct m0_reqh_service_type *stype)
{
        struct m0_reqh_service *serv;

        M0_PRE(stype != NULL && service != NULL);

        M0_ALLOC_PTR(serv);
        M0_ASSERT(serv != NULL);

        serv->rs_type = stype;
        serv->rs_ops = &ut_stats_service_ops;
        *service = serv;
        return 0;
}

static const struct m0_reqh_service_type_ops ut_stats_service_type_ops = {
        .rsto_service_allocate = ut_stats_service_allocate
};

struct m0_reqh_service_type ut_stats_service_type = {
        /*
         * "M0_CST_STATS" name is used by m0_stats_svc_type.
         * m0_reqh_service_type_register() will fail on precondition
         * if we reuse a name which is already registered.
         */
        .rst_name     = "M0_CST_STATS__UT",
        .rst_ops      = &ut_stats_service_type_ops,
        .rst_level    = M0_RS_LEVEL_NORMAL,
        .rst_typecode = M0_CST_STATS,
};

static int test_stats_init(void)
{
        int rc;

        m0_sm_conf_init(&fom_phases_conf);
        rc = m0_reqh_service_type_register(&ut_stats_service_type);
        M0_ASSERT(rc == 0);
        m0_fom_type_init(&stats_fom_type, M0_UT_STATS_OPCODE,
                         &fom_stats_type_ops, &ut_stats_service_type,
                         &fom_phases_conf);

        M0_SET0(&rmach_ctx);
        rmach_ctx.rmc_cob_id.id = DUMMY_COB_ID;
        rmach_ctx.rmc_ep_addr   = DUMMY_SERVER_ADDR;
        m0_ut_rpc_mach_init_and_add(&rmach_ctx);

        rc = m0_reqh_service_allocate(&service, &ut_stats_service_type, NULL);
        M0_ASSERT(rc == 0);
        m0_reqh_service_init(service, &rmach_ctx.rmc_reqh, NULL);
        rc = m0_reqh_service_start(service);
        M0_ASSERT(rc == 0);

        return rc;
}

static int test_stats_fini(void)
{
        m0_ut_rpc_mach_fini(&rmach_ctx);
        m0_reqh_service_type_unregister(&ut_stats_service_type);
        m0_sm_conf_fini(&fom_phases_conf);
        return 0;
}

struct m0_ut_suite m0_fom_stats_ut = {
        .ts_name = "fom-stats-ut",
        .ts_init = test_stats_init,
        .ts_fini = test_stats_fini,
        .ts_tests = {
                { "stats", test_stats },
                { NULL, NULL }
        }
};

/*
 *  Local variables:
 *  c-indentation-style: "K&R"
 *  c-basic-offset: 8
 *  tab-width: 8
 *  fill-column: 80
 *  scroll-step: 1
 *  End:
 */