grpc/test/core/census/resource_test.c

/*
 *
 * Copyright 2016, Google Inc.
 * 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 Google Inc. 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 THE COPYRIGHT
 * OWNER OR 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.
 *
 */

#include "src/core/ext/census/resource.h"
#include <grpc/census.h>
#include <grpc/support/log.h>
#include <grpc/support/port_platform.h>
#include <grpc/support/useful.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include "src/core/ext/census/base_resources.h"
#include "test/core/util/test_config.h"

// Test all the functionality for dealing with Resources.

// Just startup and shutdown resources subsystem.
static void test_enable_disable() {
  initialize_resources();
  shutdown_resources();
}

// A blank/empty initialization should not work.
static void test_empty_definition() {
  initialize_resources();
  int32_t rid = census_define_resource(NULL, 0);
  GPR_ASSERT(rid == -1);
  uint8_t buffer[50] = {0};
  rid = census_define_resource(buffer, 50);
  GPR_ASSERT(rid == -1);
  shutdown_resources();
}

// Given a file name, read raw proto and define the resource included within.
// Returns resource id from census_define_resource().
static int32_t define_resource_from_file(const char *file) {
#define BUF_SIZE 512
  uint8_t buffer[BUF_SIZE];
  FILE *input = fopen(file, "rb");
  GPR_ASSERT(input != NULL);
  size_t nbytes = fread(buffer, 1, BUF_SIZE, input);
  GPR_ASSERT(nbytes != 0 && nbytes < BUF_SIZE && feof(input) && !ferror(input));
  int32_t rid = census_define_resource(buffer, nbytes);
  GPR_ASSERT(fclose(input) == 0);
  return rid;
}

// Test definition of a single resource, using a proto read from a file. The
// `succeed` parameter indicates whether we expect the definition to succeed or
// fail. `name` is used to check that the returned resource can be looked up by
// name.
static void test_define_single_resource(const char *file, const char *name,
                                        bool succeed) {
  gpr_log(GPR_INFO, "Test defining resource \"%s\"\n", name);
  initialize_resources();
  int32_t rid = define_resource_from_file(file);
  if (succeed) {
    GPR_ASSERT(rid >= 0);
    int32_t rid2 = census_resource_id(name);
    GPR_ASSERT(rid == rid2);
  } else {
    GPR_ASSERT(rid < 0);
  }
  shutdown_resources();
}

// Try deleting various resources (both those that exist and those that don't).
static void test_delete_resource(char* minimal_good, char* full) {
  initialize_resources();
  // Try deleting resource before any are defined.
  census_delete_resource(0);
  // Create and check a couple of resources.
  int32_t rid1 = define_resource_from_file(minimal_good);
  int32_t rid2 = define_resource_from_file(full);
  GPR_ASSERT(rid1 >= 0 && rid2 >= 0 && rid1 != rid2);
  int32_t rid3 = census_resource_id("minimal_good");
  int32_t rid4 = census_resource_id("full_resource");
  GPR_ASSERT(rid1 == rid3 && rid2 == rid4);
  // Try deleting non-existant resources.
  census_delete_resource(-1);
  census_delete_resource(rid1 + rid2 + 1);
  census_delete_resource(10000000);
  // Delete one of the previously defined resources and check for deletion.
  census_delete_resource(rid1);
  rid3 = census_resource_id("minimal_good");
  GPR_ASSERT(rid3 < 0);
  // Check that re-adding works.
  rid1 = define_resource_from_file(minimal_good);
  GPR_ASSERT(rid1 >= 0);
  rid3 = census_resource_id("minimal_good");
  GPR_ASSERT(rid1 == rid3);
  shutdown_resources();
}

// Test define base resources.
static void test_base_resources() {
  initialize_resources();
  define_base_resources();
  int32_t rid1 = census_resource_id("client_rpc_latency");
  int32_t rid2 = census_resource_id("server_rpc_latency");
  GPR_ASSERT(rid1 >= 0 && rid2 >= 0 && rid1 != rid2);
  shutdown_resources();
}

int main(int argc, char **argv) {
  char *resource_empty_name_pb, *resource_full_pb, *resource_minimal_good_pb,
  *resource_no_name_pb, *resource_no_numerator_pb, *resource_no_unit_pb;
  if (argc >= 7) {
    resource_empty_name_pb = argv[1];
    resource_full_pb = argv[2];
    resource_minimal_good_pb = argv[3];
    resource_no_name_pb = argv[4];
    resource_no_numerator_pb = argv[5];
    resource_no_unit_pb = argv[6];
  } else {
    resource_empty_name_pb = "test/core/census/data/resource_empty_name.pb";
    resource_full_pb = "test/core/census/data/resource_full.pb";
    resource_minimal_good_pb = "test/core/census/data/resource_minimal_good.pb";
    resource_no_name_pb = "test/core/census/data/resource_no_name.pb";
    resource_no_numerator_pb = "test/core/census/data/resource_no_numerator.pb";
    resource_no_unit_pb = "test/core/census/data/resource_no_unit.pb";
  }
  grpc_test_init(argc, argv);
  test_enable_disable();
  test_empty_definition();
  test_define_single_resource(resource_minimal_good_pb, "minimal_good", true);
  test_define_single_resource(resource_full_pb, "full_resource", true);
  test_define_single_resource(resource_no_name_pb, "resource_no_name", false);
  test_define_single_resource(resource_no_numerator_pb,
                              "resource_no_numerator", false);
  test_define_single_resource(resource_no_unit_pb, "resource_no_unit", false);
  test_define_single_resource(resource_empty_name_pb, "resource_empty_name", false);
  test_delete_resource(resource_minimal_good_pb, resource_full_pb);
  test_base_resources();
  return 0;
}