/*******************************************************************************
 * Copyright (C) 2004-2006 Intel Corp. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *  - Redistributions of source code must retain the above copyright notice,
 *    this list of conditions and the following disclaimer.
 *
 *  - Redistributions in binary form must reproduce the above copyright notice,
 *    this list of conditions and the following disclaimer in the documentation
 *    and/or other materials provided with the distribution.
 *
 *  - Neither the name of Intel Corp. nor the names of its
 *    contributors may be used to endorse or promote products derived from this
 *    software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS''
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL Intel Corp. OR THE CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *******************************************************************************/

/**
 * @author Anas Nashif
 * @author Nathan Rakoff
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <time.h>

#include "u/libu.h"

#include "wsman-client-api.h"
#include "wsman-client-transport.h"
#include "wsman-debug.h"
#include "common.h"


static TestData pull_tests[] = {
  {
    "Pull. Enumeration with non existent Resource URI.",
    "http://schemas.dmtf.org/wbem/wscim/1/cim-schema/2/CIM_ComputerSystemxx",
    NULL,
    NULL,
    NULL,
    "/s:Envelope/s:Body/s:Fault/s:Code/s:Subcode/s:Value",
    "wsa:DestinationUnreachable",
    "/s:Envelope/s:Body/s:Fault/s:Detail/wsman:FaultDetail",
    "http://schemas.dmtf.org/wbem/wsman/1/wsman/faultDetail/InvalidResourceURI",
    400,
    FLAG_NONE,
    0
  },
  {
    "Pull. Enumeration (Optimized)",
    "http://schema.omc-project.org/wbem/wscim/1/cim-schema/2/OMC_InitdService",
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    200,
    FLAG_ENUMERATION_OPTIMIZATION,
    0
  },
  {
    "Pull. Enumeration (Optimized/EPR/Count)",
    "http://schema.omc-project.org/wbem/wscim/1/cim-schema/2/OMC_InitdService",
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    200,
    FLAG_ENUMERATION_OPTIMIZATION | FLAG_ENUMERATION_ENUM_EPR |
                                         FLAG_ENUMERATION_COUNT_ESTIMATION ,
    10
  },
  {
    "Pull. Enumeration (Optimized/EPR)",
    "http://schema.omc-project.org/wbem/wscim/1/cim-schema/2/OMC_InitdService",
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    200,
    FLAG_ENUMERATION_OPTIMIZATION | FLAG_ENUMERATION_ENUM_EPR,
    10
  },
  {
    "Pull. Enumeration (Optimized/ObjEPR)",
    "http://schema.omc-project.org/wbem/wscim/1/cim-schema/2/OMC_InitdService",
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    200,
    FLAG_ENUMERATION_OPTIMIZATION | FLAG_ENUMERATION_ENUM_OBJ_AND_EPR,
    10
  },
  {
    "Pull. Enumeration with valid Resource URI and Items Count Estimation.",
    "http://schemas.dmtf.org/wbem/wscim/1/cim-schema/2/CIM_ComputerSystem",
    NULL,
    NULL,
    NULL,
    "/s:Envelope/s:Header/wsman:TotalItemsCountEstimate",
    NULL,
    NULL,
    NULL,
    200,
    FLAG_ENUMERATION_COUNT_ESTIMATION,
    0
  }
};


static int ntests = sizeof (pull_tests) / sizeof (pull_tests[0]);


extern WsManClient *cl;

client_opt_t *options;

static void pull_test(void) {
    char *enumContext = NULL;
    static int i = 0;
    char *xp = NULL;
    int num;

    wsmc_reinit_conn(cl);
    options = wsmc_options_init();

    options->flags = pull_tests[i].flags;

    if (pull_tests[i].selectors != NULL)
         wsmc_add_selectors_from_str(options, pull_tests[i].selectors);

    options->max_elements = pull_tests[i].max_elements;
    WsXmlDocH enum_response = wsmc_action_enumerate(cl,
                                (char *)pull_tests[i].resource_uri, options, NULL);
    CU_ASSERT_TRUE(wsmc_get_response_code(cl) ==
                                                pull_tests[i].final_status);
    if (wsmc_get_response_code(cl) != pull_tests[i].final_status) {
        if (verbose) {
            printf("\nExpected = %d\nReturned = %ld       ",
                   pull_tests[i].final_status,
                   wsmc_get_response_code(cl));
        }
        goto RETURN;
    }
    CU_ASSERT_PTR_NOT_NULL(enum_response);

    if (enum_response) {
        enumContext = wsmc_get_enum_context(enum_response);
    } else {
        enumContext = NULL;
    }

    while (enumContext != NULL) {
        WsXmlDocH docp = wsmc_action_pull(cl, (char *)pull_tests[i].resource_uri,
                                     options, NULL, enumContext);
        CU_ASSERT_PTR_NOT_NULL(docp);
        if (!docp) {
            goto RETURN;
        }
        if (pull_tests[i].expr1 == NULL) {
            goto RETURN;
        }
        xp = ws_xml_get_xpath_value(docp, pull_tests[i].expr1);
        CU_ASSERT_PTR_NOT_NULL(xp);
        if (!xp) {
            goto RETURN;
        }
        if (pull_tests[i].value1) {
            CU_ASSERT_STRING_EQUAL(xp, pull_tests[i].value1);
            if (strcmp(xp, pull_tests[i].value1)) {
                if (verbose) {
                    printf("\nExpected <positive digital>\nReturned %s      ",
                            xp);
                }
            }
            goto RETURN;
        } else {
            num = atoi(xp);
            CU_ASSERT_TRUE(num > 0);
            if (num <= 0) {
                if (verbose) {
                    printf("\nExpected <positive digital>\nReturned %s      ",
                           xp);
                }
            }
            goto RETURN;
        }
        enumContext = wsmc_get_enum_context(docp);
        ws_xml_destroy_doc(docp);
    }


RETURN:
    if (enum_response) {
        ws_xml_destroy_doc(enum_response);
    }
    u_free(xp);
    wsmc_options_destroy(options);
    i++; // decrease executed test number
}


int add_pull_tests(CU_pSuite ps) {
    int found_test = 0;
    int i;

        /* add the tests to the suite */
    for (i = 0; i < ntests; i++) {
        found_test += (NULL != CU_add_test(ps,
                            pull_tests[i].explanation, pull_test));
    }
    return (found_test > 0);
}