/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/*
 * 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:
 *
 * Copyright (C) 2018 Aleksander Morgado <aleksander@aleksander.es>
 */

#include <glib.h>
#include <glib-object.h>
#include <locale.h>
#include <math.h>

#include <ModemManager.h>
#define _LIBMM_INSIDE_MM
#include <libmm-glib.h>

#include "mm-log-test.h"
#include "mm-modem-helpers.h"
#include "mm-modem-helpers-xmm.h"

#define g_assert_cmpfloat_tolerance(val1, val2, tolerance)  \
    g_assert_cmpfloat (fabs (val1 - val2), <, tolerance)

/*****************************************************************************/
/* Test XACT=? responses */

static void
validate_xact_test_response (const gchar                  *response,
                             const MMModemModeCombination *expected_modes,
                             guint                         n_expected_modes,
                             const MMModemBand            *expected_bands,
                             guint                         n_expected_bands)
{
    GError   *error = NULL;
    GArray   *modes = NULL;
    GArray   *bands = NULL;
    gboolean  ret;
    guint     i;

    ret = mm_xmm_parse_xact_test_response (response, NULL, &modes, &bands, &error);
    g_assert_no_error (error);
    g_assert (ret);

    g_assert_cmpuint (modes->len, ==, n_expected_modes);
    for (i = 0; i < modes->len; i++) {
        MMModemModeCombination mode;
        guint                  j;
        gboolean               found = FALSE;

        mode = g_array_index (modes, MMModemModeCombination, i);
        for (j = 0; !found && j < n_expected_modes; j++)
            found = (mode.allowed == expected_modes[j].allowed && mode.preferred == expected_modes[j].preferred);
        g_assert (found);
    }
    g_array_unref (modes);

    g_assert_cmpuint (bands->len, ==, n_expected_bands);
    for (i = 0; i < bands->len; i++) {
        MMModemBand band;
        guint       j;
        gboolean    found = FALSE;

        band = g_array_index (bands, MMModemBand, i);
        for (j = 0; !found && j < n_expected_bands; j++)
            found = (band == expected_bands[j]);
        g_assert (found);
    }
    g_array_unref (bands);
}

static void
test_xact_test_4g_only (void)
{
    const gchar *response =
        "+XACT: "
        "(0-6),(0-2),0,"
        "101,102,103,104,105,107,108,111,112,113,117,118,119,120,121,126,128,129,130,138,139,140,141,166";

    static const MMModemModeCombination expected_modes[] = {
        { MM_MODEM_MODE_4G, MM_MODEM_MODE_NONE },
    };

    static const MMModemBand expected_bands[] = {
        MM_MODEM_BAND_EUTRAN_1,  MM_MODEM_BAND_EUTRAN_2,  MM_MODEM_BAND_EUTRAN_3,  MM_MODEM_BAND_EUTRAN_4,  MM_MODEM_BAND_EUTRAN_5,
        MM_MODEM_BAND_EUTRAN_7,  MM_MODEM_BAND_EUTRAN_8,  MM_MODEM_BAND_EUTRAN_11, MM_MODEM_BAND_EUTRAN_12, MM_MODEM_BAND_EUTRAN_13,
        MM_MODEM_BAND_EUTRAN_17, MM_MODEM_BAND_EUTRAN_18, MM_MODEM_BAND_EUTRAN_19, MM_MODEM_BAND_EUTRAN_20, MM_MODEM_BAND_EUTRAN_21,
        MM_MODEM_BAND_EUTRAN_26, MM_MODEM_BAND_EUTRAN_28, MM_MODEM_BAND_EUTRAN_29, MM_MODEM_BAND_EUTRAN_30, MM_MODEM_BAND_EUTRAN_38,
        MM_MODEM_BAND_EUTRAN_39, MM_MODEM_BAND_EUTRAN_40, MM_MODEM_BAND_EUTRAN_41, MM_MODEM_BAND_EUTRAN_66
    };

    /* NOTE: 2G and 3G modes are reported in XACT but no 2G or 3G frequencies supported */
    validate_xact_test_response (response,
                                 expected_modes, G_N_ELEMENTS (expected_modes),
                                 expected_bands, G_N_ELEMENTS (expected_bands));
}

static void
test_xact_test_3g_4g (void)
{
    const gchar *response =
        "+XACT: "
        "(0-6),(0-2),0,"
        "1,2,4,5,8,"
        "101,102,103,104,105,107,108,111,112,113,117,118,119,120,121,126,128,129,130,138,139,140,141,166";

    static const MMModemModeCombination expected_modes[] = {
        { MM_MODEM_MODE_3G, MM_MODEM_MODE_NONE },
        { MM_MODEM_MODE_4G, MM_MODEM_MODE_NONE },
        { MM_MODEM_MODE_3G | MM_MODEM_MODE_4G, MM_MODEM_MODE_NONE },
        { MM_MODEM_MODE_3G | MM_MODEM_MODE_4G, MM_MODEM_MODE_3G },
        { MM_MODEM_MODE_3G | MM_MODEM_MODE_4G, MM_MODEM_MODE_4G },
    };

    static const MMModemBand expected_bands[] = {
        MM_MODEM_BAND_UTRAN_1,   MM_MODEM_BAND_UTRAN_2,   MM_MODEM_BAND_UTRAN_4,   MM_MODEM_BAND_UTRAN_5,   MM_MODEM_BAND_UTRAN_8,
        MM_MODEM_BAND_EUTRAN_1,  MM_MODEM_BAND_EUTRAN_2,  MM_MODEM_BAND_EUTRAN_3,  MM_MODEM_BAND_EUTRAN_4,  MM_MODEM_BAND_EUTRAN_5,
        MM_MODEM_BAND_EUTRAN_7,  MM_MODEM_BAND_EUTRAN_8,  MM_MODEM_BAND_EUTRAN_11, MM_MODEM_BAND_EUTRAN_12, MM_MODEM_BAND_EUTRAN_13,
        MM_MODEM_BAND_EUTRAN_17, MM_MODEM_BAND_EUTRAN_18, MM_MODEM_BAND_EUTRAN_19, MM_MODEM_BAND_EUTRAN_20, MM_MODEM_BAND_EUTRAN_21,
        MM_MODEM_BAND_EUTRAN_26, MM_MODEM_BAND_EUTRAN_28, MM_MODEM_BAND_EUTRAN_29, MM_MODEM_BAND_EUTRAN_30, MM_MODEM_BAND_EUTRAN_38,
        MM_MODEM_BAND_EUTRAN_39, MM_MODEM_BAND_EUTRAN_40, MM_MODEM_BAND_EUTRAN_41, MM_MODEM_BAND_EUTRAN_66
    };

    /* NOTE: 2G modes are reported in XACT but no 2G frequencies supported */
    validate_xact_test_response (response,
                                 expected_modes, G_N_ELEMENTS (expected_modes),
                                 expected_bands, G_N_ELEMENTS (expected_bands));
}

static void
test_xact_test_2g_3g_4g (void)
{
    const gchar *response =
        "+XACT: "
        "(0-6),(0-2),0,"
        "900,1800,1900,850,"
        "1,2,4,5,8,"
        "101,102,103,104,105,107,108,111,112,113,117,118,119,120,121,126,128,129,130,138,139,140,141,166";

    static const MMModemModeCombination expected_modes[] = {
        { MM_MODEM_MODE_2G, MM_MODEM_MODE_NONE },
        { MM_MODEM_MODE_3G, MM_MODEM_MODE_NONE },
        { MM_MODEM_MODE_4G, MM_MODEM_MODE_NONE },
        { MM_MODEM_MODE_2G | MM_MODEM_MODE_3G, MM_MODEM_MODE_NONE },
        { MM_MODEM_MODE_2G | MM_MODEM_MODE_3G, MM_MODEM_MODE_2G },
        { MM_MODEM_MODE_2G | MM_MODEM_MODE_3G, MM_MODEM_MODE_3G },
        { MM_MODEM_MODE_2G | MM_MODEM_MODE_4G, MM_MODEM_MODE_NONE },
        { MM_MODEM_MODE_2G | MM_MODEM_MODE_4G, MM_MODEM_MODE_2G },
        { MM_MODEM_MODE_2G | MM_MODEM_MODE_4G, MM_MODEM_MODE_4G },
        { MM_MODEM_MODE_3G | MM_MODEM_MODE_4G, MM_MODEM_MODE_NONE },
        { MM_MODEM_MODE_3G | MM_MODEM_MODE_4G, MM_MODEM_MODE_3G },
        { MM_MODEM_MODE_3G | MM_MODEM_MODE_4G, MM_MODEM_MODE_4G },
        { MM_MODEM_MODE_2G | MM_MODEM_MODE_3G | MM_MODEM_MODE_4G, MM_MODEM_MODE_NONE },
        { MM_MODEM_MODE_2G | MM_MODEM_MODE_3G | MM_MODEM_MODE_4G, MM_MODEM_MODE_2G },
        { MM_MODEM_MODE_2G | MM_MODEM_MODE_3G | MM_MODEM_MODE_4G, MM_MODEM_MODE_3G },
        { MM_MODEM_MODE_2G | MM_MODEM_MODE_3G | MM_MODEM_MODE_4G, MM_MODEM_MODE_4G },
    };

    static const MMModemBand expected_bands[] = {
        MM_MODEM_BAND_EGSM,      MM_MODEM_BAND_DCS,       MM_MODEM_BAND_PCS,       MM_MODEM_BAND_G850,
        MM_MODEM_BAND_UTRAN_1,   MM_MODEM_BAND_UTRAN_2,   MM_MODEM_BAND_UTRAN_4,   MM_MODEM_BAND_UTRAN_5,   MM_MODEM_BAND_UTRAN_8,
        MM_MODEM_BAND_EUTRAN_1,  MM_MODEM_BAND_EUTRAN_2,  MM_MODEM_BAND_EUTRAN_3,  MM_MODEM_BAND_EUTRAN_4,  MM_MODEM_BAND_EUTRAN_5,
        MM_MODEM_BAND_EUTRAN_7,  MM_MODEM_BAND_EUTRAN_8,  MM_MODEM_BAND_EUTRAN_11, MM_MODEM_BAND_EUTRAN_12, MM_MODEM_BAND_EUTRAN_13,
        MM_MODEM_BAND_EUTRAN_17, MM_MODEM_BAND_EUTRAN_18, MM_MODEM_BAND_EUTRAN_19, MM_MODEM_BAND_EUTRAN_20, MM_MODEM_BAND_EUTRAN_21,
        MM_MODEM_BAND_EUTRAN_26, MM_MODEM_BAND_EUTRAN_28, MM_MODEM_BAND_EUTRAN_29, MM_MODEM_BAND_EUTRAN_30, MM_MODEM_BAND_EUTRAN_38,
        MM_MODEM_BAND_EUTRAN_39, MM_MODEM_BAND_EUTRAN_40, MM_MODEM_BAND_EUTRAN_41, MM_MODEM_BAND_EUTRAN_66
    };

    validate_xact_test_response (response,
                                 expected_modes, G_N_ELEMENTS (expected_modes),
                                 expected_bands, G_N_ELEMENTS (expected_bands));
}

/*****************************************************************************/
/* Test XACT? responses */

static void
validate_xact_query_response (const gchar                  *response,
                              const MMModemModeCombination *expected_mode,
                              const MMModemBand            *expected_bands,
                              guint                         n_expected_bands)
{
    GError   *error = NULL;
    GArray   *bands = NULL;
    gboolean  ret;
    guint     i;

    MMModemModeCombination mode = {
        .allowed = MM_MODEM_MODE_NONE,
        .preferred = MM_MODEM_MODE_NONE,
    };

    ret = mm_xmm_parse_xact_query_response (response, &mode, &bands, &error);
    g_assert_no_error (error);
    g_assert (ret);

    g_assert_cmpuint (mode.allowed,   ==, expected_mode->allowed);
    g_assert_cmpuint (mode.preferred, ==, expected_mode->preferred);

    g_assert_cmpuint (bands->len, ==, n_expected_bands);
    for (i = 0; i < bands->len; i++) {
        MMModemBand band;
        guint       j;
        gboolean    found = FALSE;

        band = g_array_index (bands, MMModemBand, i);
        for (j = 0; !found && j < n_expected_bands; j++)
            found = (band == expected_bands[j]);
        g_assert (found);
    }
    g_array_unref (bands);
}

static void
test_xact_query_3g_only (void)
{
    const gchar *response =
        "+XACT: "
        "1,1,,"
        "1,2,4,5,8,"
        "101,102,103,104,105,107,108,111,112,113,117,118,119,120,121,126,128,129,130,138,139,140,141,166";

    static const MMModemModeCombination expected_mode = {
        .allowed = MM_MODEM_MODE_3G,
        .preferred = MM_MODEM_MODE_NONE
    };

    static const MMModemBand expected_bands[] = {
        MM_MODEM_BAND_UTRAN_1,   MM_MODEM_BAND_UTRAN_2,   MM_MODEM_BAND_UTRAN_4,   MM_MODEM_BAND_UTRAN_5,   MM_MODEM_BAND_UTRAN_8,
        MM_MODEM_BAND_EUTRAN_1,  MM_MODEM_BAND_EUTRAN_2,  MM_MODEM_BAND_EUTRAN_3,  MM_MODEM_BAND_EUTRAN_4,  MM_MODEM_BAND_EUTRAN_5,
        MM_MODEM_BAND_EUTRAN_7,  MM_MODEM_BAND_EUTRAN_8,  MM_MODEM_BAND_EUTRAN_11, MM_MODEM_BAND_EUTRAN_12, MM_MODEM_BAND_EUTRAN_13,
        MM_MODEM_BAND_EUTRAN_17, MM_MODEM_BAND_EUTRAN_18, MM_MODEM_BAND_EUTRAN_19, MM_MODEM_BAND_EUTRAN_20, MM_MODEM_BAND_EUTRAN_21,
        MM_MODEM_BAND_EUTRAN_26, MM_MODEM_BAND_EUTRAN_28, MM_MODEM_BAND_EUTRAN_29, MM_MODEM_BAND_EUTRAN_30, MM_MODEM_BAND_EUTRAN_38,
        MM_MODEM_BAND_EUTRAN_39, MM_MODEM_BAND_EUTRAN_40, MM_MODEM_BAND_EUTRAN_41, MM_MODEM_BAND_EUTRAN_66
    };

    validate_xact_query_response (response,
                                  &expected_mode,
                                  expected_bands, G_N_ELEMENTS (expected_bands));
}

static void
test_xact_query_3g_4g (void)
{
    const gchar *response =
        "+XACT: "
        "4,1,2,"
        "1,2,4,5,8,"
        "101,102,103,104,105,107,108,111,112,113,117,118,119,120,121,126,128,129,130,138,139,140,141,166";

    static const MMModemModeCombination expected_mode = {
        .allowed = MM_MODEM_MODE_3G | MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_3G
    };

    static const MMModemBand expected_bands[] = {
        MM_MODEM_BAND_UTRAN_1,   MM_MODEM_BAND_UTRAN_2,   MM_MODEM_BAND_UTRAN_4,   MM_MODEM_BAND_UTRAN_5,   MM_MODEM_BAND_UTRAN_8,
        MM_MODEM_BAND_EUTRAN_1,  MM_MODEM_BAND_EUTRAN_2,  MM_MODEM_BAND_EUTRAN_3,  MM_MODEM_BAND_EUTRAN_4,  MM_MODEM_BAND_EUTRAN_5,
        MM_MODEM_BAND_EUTRAN_7,  MM_MODEM_BAND_EUTRAN_8,  MM_MODEM_BAND_EUTRAN_11, MM_MODEM_BAND_EUTRAN_12, MM_MODEM_BAND_EUTRAN_13,
        MM_MODEM_BAND_EUTRAN_17, MM_MODEM_BAND_EUTRAN_18, MM_MODEM_BAND_EUTRAN_19, MM_MODEM_BAND_EUTRAN_20, MM_MODEM_BAND_EUTRAN_21,
        MM_MODEM_BAND_EUTRAN_26, MM_MODEM_BAND_EUTRAN_28, MM_MODEM_BAND_EUTRAN_29, MM_MODEM_BAND_EUTRAN_30, MM_MODEM_BAND_EUTRAN_38,
        MM_MODEM_BAND_EUTRAN_39, MM_MODEM_BAND_EUTRAN_40, MM_MODEM_BAND_EUTRAN_41, MM_MODEM_BAND_EUTRAN_66
    };

    validate_xact_query_response (response,
                                  &expected_mode,
                                  expected_bands, G_N_ELEMENTS (expected_bands));
}

static void
test_xact_query_no_match_mode (void)
{
    g_autoptr(GError)      error = NULL;
    gboolean               ret;
    MMModemModeCombination mode = {
        .allowed = MM_MODEM_MODE_NONE,
        .preferred = MM_MODEM_MODE_NONE,
    };

    ret = mm_xmm_parse_xact_query_response ("something here", &mode, NULL, &error);
    g_assert (error);
    g_assert (!ret);
}

static void
test_xact_query_no_match_bands (void)
{
    g_autoptr(GError) error = NULL;
    g_autoptr(GArray) bands = NULL;
    gboolean          ret;

    ret = mm_xmm_parse_xact_query_response ("something here", NULL, &bands, &error);
    g_assert (error);
    g_assert (!ret);
}

/*****************************************************************************/

#define XACT_SET_TEST_MAX_BANDS 6

typedef struct {
    MMModemMode  allowed;
    MMModemMode  preferred;
    MMModemBand  bands[XACT_SET_TEST_MAX_BANDS];
    const gchar *expected_command;
} XactSetTest;

static const XactSetTest set_tests[] = {
    {
        /* 2G-only, no explicit bands */
        .allowed   = MM_MODEM_MODE_2G,
        .preferred = MM_MODEM_MODE_NONE,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=0,,"
    },
    {
        /* 3G-only, no explicit bands */
        .allowed   = MM_MODEM_MODE_3G,
        .preferred = MM_MODEM_MODE_NONE,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=1,,"
    },
    {
        /* 4G-only, no explicit bands */
        .allowed   = MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_NONE,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=2,,"
    },
    {
        /* 2G+3G, none preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_2G | MM_MODEM_MODE_3G,
        .preferred = MM_MODEM_MODE_NONE,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=3,,"
    },
    {
        /* 2G+3G, 2G preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_2G | MM_MODEM_MODE_3G,
        .preferred = MM_MODEM_MODE_2G,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=3,0,"
    },
    {
        /* 2G+3G, 3G preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_2G | MM_MODEM_MODE_3G,
        .preferred = MM_MODEM_MODE_3G,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=3,1,"
    },
    {
        /* 3G+4G, none preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_3G | MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_NONE,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=4,,"
    },
    {
        /* 3G+4G, 3G preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_3G | MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_3G,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=4,1,"
    },
    {
        /* 3G+4G, 4G preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_3G | MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_4G,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=4,2,"
    },
    {
        /* 2G+4G, none preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_2G | MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_NONE,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=5,,"
    },
    {
        /* 2G+4G, 2G preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_2G | MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_2G,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=5,0,"
    },
    {
        /* 2G+4G, 4G preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_2G | MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_4G,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=5,2,"
    },
    {
        /* 2G+3G+4G, none preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_2G | MM_MODEM_MODE_3G | MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_NONE,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=6,,"
    },
    {
        /* 2G+3G+4G, 2G preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_2G | MM_MODEM_MODE_3G | MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_2G,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=6,0,"
    },
    {
        /* 2G+3G+4G, 3G preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_2G | MM_MODEM_MODE_3G | MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_3G,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=6,1,"
    },
    {
        /* 2G+3G+4G, 4G preferred, no explicit bands */
        .allowed   = MM_MODEM_MODE_2G | MM_MODEM_MODE_3G | MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_4G,
        .bands = { [0] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=6,2,"
    },
    {
        /* 2G bands, no explicit modes */
        .allowed = MM_MODEM_MODE_NONE,
        .preferred = MM_MODEM_MODE_NONE,
        .bands = { [0] = MM_MODEM_BAND_EGSM,
                   [1] = MM_MODEM_BAND_DCS,
                   [2] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=,,,900,1800"
    },
    {
        /* 3G bands, no explicit modes */
        .allowed = MM_MODEM_MODE_NONE,
        .preferred = MM_MODEM_MODE_NONE,
        .bands = { [0] = MM_MODEM_BAND_UTRAN_1,
                   [1] = MM_MODEM_BAND_UTRAN_2,
                   [2] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=,,,1,2"
    },
    {
        /* 4G bands, no explicit modes */
        .allowed = MM_MODEM_MODE_NONE,
        .preferred = MM_MODEM_MODE_NONE,
        .bands = { [0] = MM_MODEM_BAND_EUTRAN_1,
                   [1] = MM_MODEM_BAND_EUTRAN_2,
                   [2] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=,,,101,102"
    },
    {
        /* 2G, 3G and 4G bands, no explicit modes */
        .allowed = MM_MODEM_MODE_NONE,
        .preferred = MM_MODEM_MODE_NONE,
        .bands = { [0] = MM_MODEM_BAND_EGSM,
                   [1] = MM_MODEM_BAND_DCS,
                   [2] = MM_MODEM_BAND_UTRAN_1,
                   [3] = MM_MODEM_BAND_UTRAN_2,
                   [4] = MM_MODEM_BAND_EUTRAN_1,
                   [5] = MM_MODEM_BAND_EUTRAN_2 },
        .expected_command = "+XACT=,,,900,1800,1,2,101,102"
    },
    {
        /* Auto bands, no explicit modes */
        .allowed = MM_MODEM_MODE_NONE,
        .preferred = MM_MODEM_MODE_NONE,
        .bands = { [0] = MM_MODEM_BAND_ANY,
                   [1] = MM_MODEM_BAND_UNKNOWN },
        .expected_command = "+XACT=,,,0"
    },

    {
        /* 2G+3G+4G with 4G preferred, and 2G+3G+4G bands */
        .allowed = MM_MODEM_MODE_2G | MM_MODEM_MODE_3G | MM_MODEM_MODE_4G,
        .preferred = MM_MODEM_MODE_4G,
        .bands = { [0] = MM_MODEM_BAND_EGSM,
                   [1] = MM_MODEM_BAND_DCS,
                   [2] = MM_MODEM_BAND_UTRAN_1,
                   [3] = MM_MODEM_BAND_UTRAN_2,
                   [4] = MM_MODEM_BAND_EUTRAN_1,
                   [5] = MM_MODEM_BAND_EUTRAN_2 },
        .expected_command = "+XACT=6,2,,900,1800,1,2,101,102"
    },
};

static void
validate_xact_set_command (const MMModemMode  allowed,
                           const MMModemMode  preferred,
                           const MMModemBand *bands,
                           guint              n_bands,
                           const gchar       *expected_command)
{
    gchar                  *command;
    MMModemModeCombination  mode;
    GArray                 *bandsarray = NULL;
    GError                 *error = NULL;

    if (n_bands)
        bandsarray = g_array_append_vals (g_array_sized_new (FALSE, FALSE, sizeof (MMModemBand), n_bands), bands, n_bands);

    mode.allowed   = allowed;
    mode.preferred = preferred;

    command = mm_xmm_build_xact_set_command ((mode.allowed != MM_MODEM_MODE_NONE) ? &mode : NULL, bandsarray, &error);
    g_assert_no_error (error);
    g_assert (command);

    g_assert_cmpstr (command, == , expected_command);

    g_free (command);
    if (bandsarray)
        g_array_unref (bandsarray);
}

static void
test_xact_set (void)
{
    guint i;

    for (i = 0; i < G_N_ELEMENTS (set_tests); i++) {
        guint n_bands = 0;
        guint j;

        for (j = 0; j < XACT_SET_TEST_MAX_BANDS; j++) {
            if (set_tests[i].bands[j] != MM_MODEM_BAND_UNKNOWN)
                n_bands++;
        }

        validate_xact_set_command (set_tests[i].allowed,
                                   set_tests[i].preferred,
                                   set_tests[i].bands,
                                   n_bands,
                                   set_tests[i].expected_command);
    }
}

/*****************************************************************************/
/* Test +XCESQ responses */

typedef struct {
    const gchar *str;

    gboolean gsm_info;
    guint    rxlev;
    gdouble  rssi;
    guint    ber;

    gboolean umts_info;
    guint    rscp_level;
    gdouble  rscp;
    guint    ecn0_level;
    gdouble  ecio;

    gboolean lte_info;
    guint    rsrq_level;
    gdouble  rsrq;
    guint    rsrp_level;
    gdouble  rsrp;
    gint     rssnr_level;
    gdouble  rssnr;
} XCesqResponseTest;

static const XCesqResponseTest xcesq_response_tests[] = {
    {
        .str       = "+XCESQ: 0,99,99,255,255,19,46,32",
        .gsm_info  = FALSE, .rxlev = 99, .ber = 99,
        .umts_info = FALSE, .rscp_level = 255, .ecn0_level = 255,
        .lte_info  = TRUE,  .rsrq_level = 19, .rsrq = -10.5, .rsrp_level = 46, .rsrp = -95.0, .rssnr_level = 32, .rssnr = 16.0
    },
    {
        .str       = "+XCESQ: 0,99,99,255,255,19,46,-32",
        .gsm_info  = FALSE, .rxlev = 99, .ber = 99,
        .umts_info = FALSE, .rscp_level = 255, .ecn0_level = 255,
        .lte_info  = TRUE,  .rsrq_level = 19, .rsrq = -10.5, .rsrp_level = 46, .rsrp = -95.0, .rssnr_level = -32, .rssnr = -16.0
    },
    {
        .str       = "+XCESQ: 0,99,99,255,255,16,47,28",
        .gsm_info  = FALSE, .rxlev = 99, .ber = 99,
        .umts_info = FALSE, .rscp_level = 255, .ecn0_level = 255,
        .lte_info  = TRUE,  .rsrq_level = 16, .rsrq = -12.0, .rsrp_level = 47, .rsrp = -94.0, .rssnr_level = 28, .rssnr = 14.0
    },
    {
        .str       = "+XCESQ: 0,99,99,41,29,255,255,255",
        .gsm_info  = FALSE, .rxlev = 99, .ber = 99,
        .umts_info = TRUE,  .rscp_level = 41, .rscp = -80.0, .ecn0_level = 29, .ecio = -10.0,
        .lte_info  = FALSE, .rsrq_level = 255, .rsrp_level = 255, .rssnr_level = 255
    },
    {
        .str       = "+XCESQ: 0,10,6,255,255,255,255,255",
        .gsm_info  = TRUE,  .rxlev = 10, .rssi = -101.0, .ber = 6,
        .umts_info = FALSE, .rscp_level = 255, .ecn0_level = 255,
        .lte_info  = FALSE, .rsrq_level = 255, .rsrp_level = 255, .rssnr_level = 255
    }
};

static void
test_xcesq_response (void)
{
    guint i;

    for (i = 0; i < G_N_ELEMENTS (xcesq_response_tests); i++) {
        GError   *error = NULL;
        gboolean  success;
        guint rxlev = G_MAXUINT;
        guint ber = G_MAXUINT;
        guint rscp = G_MAXUINT;
        guint ecn0 = G_MAXUINT;
        guint rsrq = G_MAXUINT;
        guint rsrp = G_MAXUINT;
        gint rssnr = G_MAXUINT;

        success = mm_xmm_parse_xcesq_query_response (xcesq_response_tests[i].str,
                                                    &rxlev, &ber,
                                                    &rscp, &ecn0,
                                                    &rsrq, &rsrp,
                                                    &rssnr, &error);
        g_assert_no_error (error);
        g_assert (success);

        g_assert_cmpuint (xcesq_response_tests[i].rxlev,      ==, rxlev);
        g_assert_cmpuint (xcesq_response_tests[i].ber,        ==, ber);
        g_assert_cmpuint (xcesq_response_tests[i].rscp_level, ==, rscp);
        g_assert_cmpuint (xcesq_response_tests[i].ecn0_level, ==, ecn0);
        g_assert_cmpuint (xcesq_response_tests[i].rsrq_level, ==, rsrq);
        g_assert_cmpuint (xcesq_response_tests[i].rsrp_level, ==, rsrp);
        g_assert_cmpuint (xcesq_response_tests[i].rssnr_level, ==, rssnr);
    }
}

static void
test_xcesq_response_to_signal (void)
{
    guint i;

    for (i = 0; i < G_N_ELEMENTS (xcesq_response_tests); i++) {
        GError   *error = NULL;
        gboolean  success;
        MMSignal *gsm  = NULL;
        MMSignal *umts = NULL;
        MMSignal *lte  = NULL;

        success = mm_xmm_xcesq_response_to_signal_info (xcesq_response_tests[i].str,
                                                        NULL,
                                                        &gsm, &umts, &lte,
                                                        &error);
        g_assert_no_error (error);
        g_assert (success);

        if (xcesq_response_tests[i].gsm_info) {
            g_assert (gsm);
            g_assert_cmpfloat_tolerance (mm_signal_get_rssi (gsm), xcesq_response_tests[i].rssi, 0.1);
            g_object_unref (gsm);
        } else
            g_assert (!gsm);

        if (xcesq_response_tests[i].umts_info) {
            g_assert (umts);
            g_assert_cmpfloat_tolerance (mm_signal_get_rscp (umts), xcesq_response_tests[i].rscp, 0.1);
            g_assert_cmpfloat_tolerance (mm_signal_get_ecio (umts), xcesq_response_tests[i].ecio, 0.1);
            g_object_unref (umts);
        } else
            g_assert (!umts);

        if (xcesq_response_tests[i].lte_info) {
            g_assert (lte);
            g_assert_cmpfloat_tolerance (mm_signal_get_rsrq (lte), xcesq_response_tests[i].rsrq, 0.1);
            g_assert_cmpfloat_tolerance (mm_signal_get_rsrp (lte), xcesq_response_tests[i].rsrp, 0.1);
            g_assert_cmpfloat_tolerance (mm_signal_get_snr (lte), xcesq_response_tests[i].rssnr, 0.1);
            g_object_unref (lte);
        } else
            g_assert (!lte);
    }
}

/*****************************************************************************/
/* AT+XLCSLSR=? response parser */

typedef struct {
    const gchar *response;
    gboolean     expected_transport_protocol_invalid_supported;
    gboolean     expected_transport_protocol_supl_supported;
    gboolean     expected_standalone_position_mode_supported;
    gboolean     expected_ms_assisted_based_position_mode_supported;
    gboolean     expected_loc_response_type_nmea_supported;
    gboolean     expected_gnss_type_gps_glonass_supported;
} XlcslsrTest;

static XlcslsrTest xlcslsr_tests[] = {
    {
        "+XLCSLSR:(0-2),(0-3), ,(0-1), ,(0-1),(0-7200),(0-255),(0-1),(0-2),(1-256),(0-1)",
        TRUE, TRUE, TRUE, TRUE, TRUE, TRUE,
    },
    {
        "+XLCSLSR:(0,1,2),(0,1,2,3), ,(0,1), ,(0,1),(0-7200),(0-255),(0,1),(0,1,2),(1-256),(0,1)",
        TRUE, TRUE, TRUE, TRUE, TRUE, TRUE,
    },
    {
        "+XLCSLSR:(0-1),(0-2), ,(0,1), ,(0,1),(0 -7200),(0-255),(0-1),(0),(1-256),(1)",
        FALSE, TRUE, FALSE, TRUE, FALSE, FALSE
    },
};

static void
test_xlcslsr_test (void)
{
    guint i;

    for (i = 0; i < G_N_ELEMENTS (xlcslsr_tests); i++) {
        GError   *error = NULL;
        gboolean  ret;
        gboolean  transport_protocol_invalid_supported;
        gboolean  transport_protocol_supl_supported;
        gboolean  standalone_position_mode_supported;
        gboolean  ms_assisted_based_position_mode_supported;
        gboolean  loc_response_type_nmea_supported;
        gboolean  gnss_type_gps_glonass_supported;

        ret = mm_xmm_parse_xlcslsr_test_response (xlcslsr_tests[i].response,
                                                  &transport_protocol_invalid_supported,
                                                  &transport_protocol_supl_supported,
                                                  &standalone_position_mode_supported,
                                                  &ms_assisted_based_position_mode_supported,
                                                  &loc_response_type_nmea_supported,
                                                  &gnss_type_gps_glonass_supported,
                                                  &error);
        g_assert_no_error (error);
        g_assert (ret);

        g_assert (transport_protocol_invalid_supported       == xlcslsr_tests[i].expected_transport_protocol_invalid_supported);
        g_assert (transport_protocol_supl_supported          == xlcslsr_tests[i].expected_transport_protocol_supl_supported);
        g_assert (standalone_position_mode_supported         == xlcslsr_tests[i].expected_standalone_position_mode_supported);
        g_assert (ms_assisted_based_position_mode_supported  == xlcslsr_tests[i].expected_ms_assisted_based_position_mode_supported);
        g_assert (loc_response_type_nmea_supported           == xlcslsr_tests[i].expected_loc_response_type_nmea_supported);
        g_assert (gnss_type_gps_glonass_supported            == xlcslsr_tests[i].expected_gnss_type_gps_glonass_supported);
    }
}

/*****************************************************************************/
/* AT+XLCSSLP? response parser */

typedef struct {
    const gchar *response;
    const gchar *expected;
} XlcsslpQuery;

static XlcsslpQuery xlcsslp_queries[] = {
    {
        "+XLCSSLP:1,\"www.spirent-lcs.com\",7275",
        "www.spirent-lcs.com:7275"
    },
    {
        "+XLCSSLP:0,\"123.123.123.123\",7275",
        "123.123.123.123:7275"
    },
};

static void
test_xlcsslp_queries (void)
{
    guint i;

    for (i = 0; i < G_N_ELEMENTS (xlcsslp_queries); i++) {
        GError   *error = NULL;
        gchar    *supl_server = NULL;
        gboolean  ret;

        ret = mm_xmm_parse_xlcsslp_query_response (xlcsslp_queries[i].response,
                                                   &supl_server,
                                                   &error);
        g_assert_no_error (error);
        g_assert (ret);

        g_assert_cmpstr (supl_server, ==, xlcsslp_queries[i].expected);
        g_free (supl_server);
    }
}

/*****************************************************************************/

int main (int argc, char **argv)
{
    setlocale (LC_ALL, "");

    g_test_init (&argc, &argv, NULL);

    g_test_add_func ("/MM/xmm/xact/test/4g-only",  test_xact_test_4g_only);
    g_test_add_func ("/MM/xmm/xact/test/3g-4g",    test_xact_test_3g_4g);
    g_test_add_func ("/MM/xmm/xact/test/2g-3g-4g", test_xact_test_2g_3g_4g);

    g_test_add_func ("/MM/xmm/xact/query/3g-only",        test_xact_query_3g_only);
    g_test_add_func ("/MM/xmm/xact/query/3g-4g",          test_xact_query_3g_4g);
    g_test_add_func ("/MM/xmm/xact/query/no-match/mode",  test_xact_query_no_match_mode);
    g_test_add_func ("/MM/xmm/xact/query/no-match/bands", test_xact_query_no_match_bands);

    g_test_add_func ("/MM/xmm/xact/set", test_xact_set);

    g_test_add_func ("/MM/xmm/xcesq/query_response", test_xcesq_response);
    g_test_add_func ("/MM/xmm/xcesq/query_response_to_signal", test_xcesq_response_to_signal);

    g_test_add_func ("/MM/xmm/xlcslsr/test", test_xlcslsr_test);

    g_test_add_func ("/MM/xmm/xlcsslp/query", test_xlcsslp_queries);

    return g_test_run ();
}
