/*
 * Copyright (c) 2017 Cisco 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.
 */


// Include the file(s) containing the functions to be tested.
// This permits internal static functions to be visible to this Test Framework.
#include "../rta_CommandTransmitStatistics.c"

#include <LongBow/unit-test.h>
#include <parc/algol/parc_SafeMemory.h>

#include <sys/time.h>

#define MAX_FILENAME 1024

typedef struct test_data {
    struct timeval period;
    char filename[MAX_FILENAME];

    RtaCommandTransmitStatistics *transmitStats;
} TestData;

static TestData *
_createTestData(void)
{
    TestData *data = parcMemory_AllocateAndClear(sizeof(TestData));
    assertNotNull(data, "parcMemory_AllocateAndClear(%zu) returned NULL", sizeof(TestData));

    data->period = (struct timeval) { .tv_sec = 1234, .tv_usec = 2389484 };
    snprintf(data->filename, MAX_FILENAME, "Miss Piggy");

    data->transmitStats = rtaCommandTransmitStatistics_Create(data->period, data->filename);

    return data;
}

static void
_destroyTestData(TestData **dataPtr)
{
    TestData *data = *dataPtr;
    rtaCommandTransmitStatistics_Release(&data->transmitStats);
    parcMemory_Deallocate((void **) &data);
    *dataPtr = NULL;
}

// =============================================================
LONGBOW_TEST_RUNNER(rta_CommandTransmitStatistics)
{
    // The following Test Fixtures will run their corresponding Test Cases.
    // Test Fixtures are run in the order specified, but all tests should be idempotent.
    // Never rely on the execution order of tests or share state between them.
    LONGBOW_RUN_TEST_FIXTURE(Global);
}

// The Test Runner calls this function once before any Test Fixtures are run.
LONGBOW_TEST_RUNNER_SETUP(rta_CommandTransmitStatistics)
{
    return LONGBOW_STATUS_SUCCEEDED;
}

// The Test Runner calls this function once after all the Test Fixtures are run.
LONGBOW_TEST_RUNNER_TEARDOWN(rta_CommandTransmitStatistics)
{
    return LONGBOW_STATUS_SUCCEEDED;
}

LONGBOW_TEST_FIXTURE(Global)
{
    LONGBOW_RUN_TEST_CASE(Global, rtaCommandTransmitStatistics_Acquire);
    LONGBOW_RUN_TEST_CASE(Global, rtaCommandTransmitStatistics_Create);
    LONGBOW_RUN_TEST_CASE(Global, rtaCommandTransmitStatistics_GetFilename);
    LONGBOW_RUN_TEST_CASE(Global, rtaCommandTransmitStatistics_GetPeriod);
    LONGBOW_RUN_TEST_CASE(Global, rtaCommandTransmitStatistics_Release);
}

LONGBOW_TEST_FIXTURE_SETUP(Global)
{
    return LONGBOW_STATUS_SUCCEEDED;
}

LONGBOW_TEST_FIXTURE_TEARDOWN(Global)
{
    uint32_t outstandingAllocations = parcSafeMemory_ReportAllocation(STDERR_FILENO);
    if (outstandingAllocations != 0) {
        printf("%s leaks memory by %d allocations\n", longBowTestCase_GetName(testCase), outstandingAllocations);
        return LONGBOW_STATUS_MEMORYLEAK;
    }
    return LONGBOW_STATUS_SUCCEEDED;
}

LONGBOW_TEST_CASE(Global, rtaCommandTransmitStatistics_Acquire)
{
    TestData *data = _createTestData();

    size_t firstRefCount = parcObject_GetReferenceCount(data->transmitStats);

    RtaCommandTransmitStatistics *second = rtaCommandTransmitStatistics_Acquire(data->transmitStats);
    size_t secondRefCount = parcObject_GetReferenceCount(second);

    assertTrue(secondRefCount == firstRefCount + 1, "Wrong refcount after acquire, got %zu expected %zu", secondRefCount, firstRefCount + 1);

    rtaCommandTransmitStatistics_Release(&second);
    _destroyTestData(&data);
}

LONGBOW_TEST_CASE(Global, rtaCommandTransmitStatistics_Create)
{
    TestData *data = _createTestData();
    assertNotNull(data->transmitStats, "Got null from create");

    assertTrue(timercmp(&data->period, &data->transmitStats->period, ==), "Period values not equal");
    assertTrue(strcmp(data->filename, data->transmitStats->filename) == 0, "Filenames not equal");
    _destroyTestData(&data);
}

LONGBOW_TEST_CASE(Global, rtaCommandTransmitStatistics_GetFilename)
{
    TestData *data = _createTestData();

    const char *testFilename = rtaCommandTransmitStatistics_GetFilename(data->transmitStats);
    assertTrue(strcmp(data->filename, testFilename) == 0, "Filenames not equal");

    _destroyTestData(&data);
}

LONGBOW_TEST_CASE(Global, rtaCommandTransmitStatistics_GetPeriod)
{
    TestData *data = _createTestData();

    struct timeval testPeriod = rtaCommandTransmitStatistics_GetPeriod(data->transmitStats);
    assertTrue(timercmp(&data->period, &testPeriod, ==), "Period values not equal");

    _destroyTestData(&data);
}

LONGBOW_TEST_CASE(Global, rtaCommandTransmitStatistics_Release)
{
    TestData *data = _createTestData();

    RtaCommandTransmitStatistics *second = rtaCommandTransmitStatistics_Acquire(data->transmitStats);
    size_t secondRefCount = parcObject_GetReferenceCount(second);

    rtaCommandTransmitStatistics_Release(&second);
    size_t thirdRefCount = parcObject_GetReferenceCount(data->transmitStats);

    assertTrue(thirdRefCount == secondRefCount - 1, "Wrong refcount after release, got %zu expected %zu", thirdRefCount, secondRefCount - 1);

    _destroyTestData(&data);
}

int
main(int argc, char *argv[])
{
    LongBowRunner *testRunner = LONGBOW_TEST_RUNNER_CREATE(rta_CommandTransmitStatistics);
    int exitStatus = longBowMain(argc, argv, testRunner, NULL);
    longBowTestRunner_Destroy(&testRunner);
    exit(exitStatus);
}