setlocale.c

/**
 * Copyright (c) 2022 Huawei Device Co., Ltd.
 * 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.
 */

#include <locale.h>
#include <stdlib.h>
#include <stdio.h>
#include "functionalext.h"

#define TEST_LC_COUNT 7
#define TEST_LC_LENGTH 18
#define TEST_LC_OFFSET 6
#define PARAM_ERROR_VALUE_1 13
#define PARAM_ERROR_VALUE_2 (-1)

static const int LcArry[TEST_LC_COUNT] = {
    LC_PAPER, LC_NAME, LC_ADDRESS, LC_TELEPHONE, LC_MEASUREMENT, LC_IDENTIFICATION, LC_ALL};

static const char envforlocale[][TEST_LC_LENGTH] = {
    "LC_PAPER",
    "LC_NAME",
    "LC_ADDRESS",
    "LC_TELEPHONE",
    "LC_MEASUREMENT",
    "LC_IDENTIFICATION",
};

/**
 * @tc.name      : setlocaletest_0100
 * @tc.desc      : Determines whether setlocale returns the default value C
 *                 when the character set passed in for different data types is NULL
 * @tc.level     : Level 0
 */
void setlocale_0100(void)
{
    const int num = sizeof(LcArry) / sizeof(LcArry[0]);
    for (int i = 0; i < num; i++) {
        const char *locale = setlocale(LcArry[i], NULL);
        if (!locale) {
            t_error("[%s] failed\n", "setlocale_0100");
            return;
        }
        EXPECT_EQ("SetlocaleTest_0100", strcmp(locale, "C"), 0);
    }
}

/**
 * @tc.name      : setlocaletest_0200
 * @tc.desc      : Determines whether setlocale returns the default value "C"
 *                 when the default value "C" is passed in for different LC data types
 * @tc.level     : Level 0
 */
void setlocale_0200(void)
{
    const int num = sizeof(LcArry) / sizeof(LcArry[0]);
    for (int i = 0; i < num; i++) {
        setenv(envforlocale[i], "en-US", 1);
        const char *locale = setlocale(LcArry[i], "C");
        if (!locale) {
            t_error("[%s] failed\n", "setlocale_0200");
            return;
        }
        EXPECT_STREQ("SetlocaleTest_0200", locale, "C");
    }
}

/**
 * @tc.name      : setlocaletest_0300
 * @tc.desc      : Asserts whether the result returned is null when an exception LC data type is passed in
 * @tc.level     : Level 2
 */
void setlocale_0300(void)
{
    const char *locale1 = setlocale(PARAM_ERROR_VALUE_1, NULL);
    if (locale1) {
        t_error("[%s] failed\n", "SetlocaleTest_0300");
    }

    const char *locale2 = setlocale(PARAM_ERROR_VALUE_2, NULL);
    if (locale2) {
        t_error("[%s] failed\n", "SetlocaleTest_0300");
    }
}

/**
 * @tc.name      : setlocaletest_0400
 * @tc.desc      : Determines whether setlocale returns da_DK
 *                 when the environment variable is set to da_DK for different LC data types
 * @tc.level     : Level 0
 */
void setlocale_0400(void)
{
    for (unsigned int i = 0; i < sizeof(envforlocale) / sizeof(envforlocale[0]); i++) {
        setenv(envforlocale[i], "da_DK", 1);
        const char *locale = setlocale(LcArry[i], "");
        if (!locale) {
            t_error("[%s] failed\n", "setlocale_0400");
            return;
        }
        EXPECT_STREQ("SetlocaleTest_0400", locale, "da_DK");
    }
}

/**
 * @tc.name      : setlocaletest_0500
 * @tc.desc      : Determines whether setlocale returns en_ZA
 *                 when the character set passed in for different LC data types is set to en_ZA
 * @tc.level     : Level 0
 */
void setlocale_0500(void)
{
    char *rev = setlocale(LC_ALL, "C");
    if (!rev) {
        t_error("[%s] failed\n", "setlocale_0500");
        return;
    }
    for (unsigned int i = 0; i < sizeof(LcArry) / sizeof(LcArry[0]); i++) {
        const char *locale = setlocale(LcArry[i], "en_ZA");
        if (locale == NULL) {
            t_error("[%s] failed\n", "setlocale_0500");
            return;
        }
        EXPECT_STREQ("SetlocaleTest_0600", locale, "en_ZA");
    }
}

/**
 * @tc.name      : setlocaletest_0600
 * @tc.desc      : Verify that the environment variable of different LC data types is set to ar_QA. When the setlocale
 *                 function is used to pass LC_ALL, the corresponding field returned by the function is ar_QA
 * @tc.level     : Level 0
 */
void setlocale_0600(void)
{
    for (unsigned int i = 0; i < sizeof(envforlocale) / sizeof(envforlocale[0]); i++) {
        int count = 0;
        char *vec[LC_ALL];
        const char *flag = ";";

        setenv(envforlocale[i], "ar_QA", 1);
        char *locale = setlocale(LC_ALL, "");
        if (locale == NULL) {
            t_error("[%s] failed\n", "setlocale_0600");
            return;
        }

        char *token = strtok(locale, flag);
        while (token != NULL && count < LC_ALL) {
            vec[count] = token;
            token = strtok(NULL, flag);
            count++;
        }

        EXPECT_NE("setlocale_0600", count, 0);

        int expectPos = i + TEST_LC_OFFSET;
        EXPECT_STREQ("setlocale_0600", "ar_QA", vec[expectPos]);
    }
}

int main(void)
{
    setlocale_0100();
    setlocale_0200();
    setlocale_0300();
    setlocale_0400();
    setlocale_0500();
    setlocale_0600();

    return t_status;
}