MaxScale/server/core/test/test_utils.cc

/*
 * Copyright (c) 2016 MariaDB Corporation Ab
 *
 * Use of this software is governed by the Business Source License included
 * in the LICENSE.TXT file and at www.mariadb.com/bsl11.
 *
 * Change Date: 2019-07-01
 *
 * On the date above, in accordance with the Business Source License, use
 * of this software will be governed by version 2 or later of the General
 * Public License.
 */

#include <maxbase/assert.h>
#include <maxscale/utils.h>
#include <maxscale/utils.hh>
#include <string.h>
#include <iostream>

using std::cout;
using std::endl;

namespace
{

#define TRIM_TCE(zFrom, zTo) { zFrom, zTo }

struct TRIM_TEST_CASE
{
    const char* zFrom;
    const char* zTo;
};

TRIM_TEST_CASE trim_testcases[] =
{
    TRIM_TCE("", ""),
    TRIM_TCE("a", "a"),
    TRIM_TCE(" a", "a"),
    TRIM_TCE("a ", "a"),
    TRIM_TCE(" a ", "a"),
    TRIM_TCE("  a", "a"),
    TRIM_TCE("a  ", "a"),
    TRIM_TCE("  a  ", "a"),
    TRIM_TCE("  a b  ", "a b"),
};

const int n_trim_testcases = sizeof(trim_testcases) / sizeof(trim_testcases[0]);

TRIM_TEST_CASE trim_leading_testcases[] =
{
    TRIM_TCE("", ""),
    TRIM_TCE("a", "a"),
    TRIM_TCE(" a", "a"),
    TRIM_TCE("a ", "a "),
    TRIM_TCE(" a ", "a "),
    TRIM_TCE("  a", "a"),
    TRIM_TCE("a  ", "a  "),
    TRIM_TCE("  a  ", "a  "),
    TRIM_TCE("  a b  ", "a b  "),
};

const int n_trim_leading_testcases = sizeof(trim_leading_testcases) / sizeof(trim_leading_testcases[0]);

TRIM_TEST_CASE trim_trailing_testcases[] =
{
    TRIM_TCE("", ""),
    TRIM_TCE("a", "a"),
    TRIM_TCE(" a", " a"),
    TRIM_TCE("a ", "a"),
    TRIM_TCE(" a ", " a"),
    TRIM_TCE("  a", "  a"),
    TRIM_TCE("a  ", "a"),
    TRIM_TCE("  a  ", "  a"),
    TRIM_TCE("  a b  ", "  a b"),
};

const int n_trim_trailing_testcases = sizeof(trim_trailing_testcases) / sizeof(trim_trailing_testcases[0]);


int test(TRIM_TEST_CASE* pTest_cases, int n_test_cases, char* (*p)(char*))
{
    int rv = 0;

    for (int i = 0; i < n_test_cases; ++i)
    {
        const char* zFrom = pTest_cases[i].zFrom;
        const char* zTo = pTest_cases[i].zTo;

        char copy[strlen(zFrom) + 1];
        strcpy(copy, zFrom);

        char* z = p(copy);

        if (strcmp(z, zTo) != 0)
        {
            ++rv;
        }
    }

    return rv;
}

int test_trim()
{
    cout << "trim()" << endl;
    return test(trim_testcases, n_trim_testcases, trim);
}

int test_trim_leading()
{
    cout << "trim_leading()" << endl;
    return test(trim_leading_testcases, n_trim_leading_testcases, trim_leading);
}

int test_trim_trailing()
{
    cout << "trim_trailing()" << endl;
    return test(trim_trailing_testcases, n_trim_trailing_testcases, trim_trailing);
}

}

template <typename T>
int test_checksums()
{
    uint8_t data[] =
    {
        'h', 'e', 'l', 'l', 'o', ' ', 'w', 'o', 'r', 'l', 'd', '!'
    };

    GWBUF* d1 = gwbuf_alloc_and_load(sizeof(data), data);
    GWBUF* d2 = gwbuf_alloc_and_load(sizeof(data), data);

    T sum1, sum2;
    sum1.update(d1);
    sum1.finalize();
    sum2.finalize(d1);
    mxb_assert(sum1 == sum2);
    sum1.reset();
    sum2.reset();

    // Check that the hex strings match
    mxb_assert(sum1.hex() == sum2.hex());

    std::string saved = sum1.hex();

    // The checksum must not be empty
    mxb_assert(!saved.empty());

    // Repeat the same test, should produce the same checksums
    sum1.update(d1);
    sum1.finalize();
    sum2.finalize(d1);
    mxb_assert(sum1 == sum2);
    mxb_assert(sum1.hex() == saved);
    mxb_assert(sum2.hex() == saved);
    sum1.reset();
    sum2.reset();

    // Check that different buffers but same content produce the same checksum
    sum1.finalize(d2);
    sum2.finalize(d1);
    mxb_assert(sum1 == sum2);
    mxb_assert(sum1.hex() == saved);
    mxb_assert(sum2.hex() == saved);

    gwbuf_free(d1);
    gwbuf_free(d2);

    return 0;
}

void test_http()
{
    auto res = mxs::http::get("https://mariadb.com/");
    std::cout << "https://mariadb.com/ responded with: " << res.code << std::endl;
    if (res.code == 200)
    {
        if (res.headers.count("Date"))
        {
            std::cout << "The date is: " << res.headers["Date"] << std::endl;
        }
    }
}

int main(int argc, char* argv[])
{
    int rv = 0;

    test_http(); // Just to see that it works

    rv += test_trim();
    rv += test_trim_leading();
    rv += test_trim_trailing();
    rv += test_checksums<mxs::SHA1Checksum>();
    rv += test_checksums<mxs::CRC32Checksum>();

    return rv;
}