/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "test_helpers.h"

#include <open62541/client_config_default.h>

#include <open62541/plugin/certificategroup_default.h>

#include <open62541/plugin/log.h>

#include <open62541/plugin/log_stdout.h>

#include <open62541/server_config_default.h>

#define TESTING_LOGLEVEL UA_LOGLEVEL_INFO

static void

testServerNotificationCallback(UA_Server *server,

                               UA_ApplicationNotificationType type,

                               const UA_KeyValueMap payload) {

}

UA_Server * UA_Server_newForUnitTest(void) {

    UA_ServerConfig sc;

    memset(&sc, 0, sizeof(UA_ServerConfig));

    sc.logging = UA_Log_Stdout_new(TESTING_LOGLEVEL);

    UA_ServerConfig_setMinimal(&sc, 4840, NULL);

    UA_Server *server = UA_Server_newWithConfig(&sc);

    if(!server)

        return NULL;

    UA_ServerConfig *config = UA_Server_getConfig(server);

    /* Manually set the eventloop clock to the fake clock */

    config->eventLoop->dateTime_now = UA_DateTime_now_fake;

    config->eventLoop->dateTime_nowMonotonic = UA_DateTime_now_fake;

    config->tcpReuseAddr = true;

    return server;

}

UA_Server *

UA_Server_newForUnitTestWithSecurityPolicies(UA_UInt16 portNumber,

                                             const UA_ByteString *certificate,

                                             const UA_ByteString *privateKey,

                                             const UA_ByteString *trustList,

                                             size_t trustListSize,

                                             const UA_ByteString *issuerList,

                                             size_t issuerListSize,

                                             const UA_ByteString *revocationList,

                                             size_t revocationListSize) {

    UA_ServerConfig config;

    memset(&config, 0, sizeof(UA_ServerConfig));

    config.logging = UA_Log_Stdout_new(TESTING_LOGLEVEL);

#ifdef UA_ENABLE_ENCRYPTION

    UA_ServerConfig_setDefaultWithSecurityPolicies(&config, portNumber,

                                                   certificate, privateKey,

                                                   trustList, trustListSize,

                                                   issuerList, issuerListSize,

                                                   revocationList, revocationListSize);

#endif

    config.eventLoop->dateTime_now = UA_DateTime_now_fake;

    config.eventLoop->dateTime_nowMonotonic = UA_DateTime_now_fake;

    config.tcpReuseAddr = true;

    config.globalNotificationCallback = testServerNotificationCallback;

    return UA_Server_newWithConfig(&config);

}

#if defined(__linux__) || defined(UA_ARCHITECTURE_WIN32)

UA_Server *

UA_Server_newForUnitTestWithSecurityPolicies_Filestore(UA_UInt16 portNumber,

                                                       const UA_ByteString *certificate,

                                                       const UA_ByteString *privateKey,

                                                       const UA_String storePath) {

    UA_ServerConfig config;

    memset(&config, 0, sizeof(UA_ServerConfig));

    config.logging = UA_Log_Stdout_new(TESTING_LOGLEVEL);

#ifdef UA_ENABLE_ENCRYPTION

    UA_ServerConfig_setDefaultWithFilestore(&config, portNumber,

                                            certificate, privateKey, storePath);

#endif

    config.eventLoop->dateTime_now = UA_DateTime_now_fake;

    config.eventLoop->dateTime_nowMonotonic = UA_DateTime_now_fake;

    config.tcpReuseAddr = true;

    config.globalNotificationCallback = testServerNotificationCallback;

    return UA_Server_newWithConfig(&config);

}

#endif /* defined(__linux__) || defined(UA_ARCHITECTURE_WIN32) */

static void

testClientNotificationCallback(UA_Client *client,

                               UA_ApplicationNotificationType type,

                               const UA_KeyValueMap payload) {

}

UA_Client *

UA_Client_newForUnitTest(void) {

    UA_ClientConfig cc;

    memset(&cc, 0, sizeof(UA_ClientConfig));

    cc.logging = UA_Log_Stdout_new(TESTING_LOGLEVEL);

    UA_ClientConfig_setDefault(&cc);

    UA_Client *client = UA_Client_newWithConfig(&cc);

    if(!client)

        return NULL;

    UA_ClientConfig *config = UA_Client_getConfig(client);

    /* Manually set the eventloop clock to the fake clock */

    config->eventLoop->dateTime_now = UA_DateTime_now_fake;

    config->eventLoop->dateTime_nowMonotonic = UA_DateTime_now_fake;

    config->tcpReuseAddr = true;

    /* Allow username/pw w/o encryption */

    config->allowNonePolicyPassword = true;

    /* Increase the timeouts (needed for valgrind CI tests) */

    config->timeout = 10 * 60 * 1000;

    config->globalNotificationCallback = testClientNotificationCallback;

    return client;

}

UA_StatusCode

UA_ClientConfig_newForUnitTestWithEncryption(UA_ClientConfig *config,

                                             const UA_ByteString certificate,

                                             const UA_ByteString privateKey) {

    memset(config, 0, sizeof(UA_ClientConfig));

    config->logging = UA_Log_Stdout_new(TESTING_LOGLEVEL);

#ifdef UA_ENABLE_ENCRYPTION

    UA_ClientConfig_setDefaultEncryption(config, certificate, privateKey,

                                         NULL, 0, NULL, 0);

#endif

    UA_CertificateGroup_AcceptAll(&config->certificateVerification);

    config->eventLoop->dateTime_now = UA_DateTime_now_fake;

    config->eventLoop->dateTime_nowMonotonic = UA_DateTime_now_fake;

    config->tcpReuseAddr = true;

    config->allowNonePolicyPassword = true;

    config->timeout = 10 * 60 * 1000; /* increased for valgrind CI */

    config->globalNotificationCallback = testClientNotificationCallback;

    return UA_STATUSCODE_GOOD;

}