opentelemetry-java-contrib/azure-resources/src/test/java/io/opentelemetry/contrib/azure/resource/AzureAppServiceResourceProviderTest.java at db031fce278b2137cf5a68958702c1ff53fd91e2 · open-telemetry/opentelemetry-java-contrib · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
/*
 * Copyright The OpenTelemetry Authors
 * SPDX-License-Identifier: Apache-2.0
 */

package io.opentelemetry.contrib.azure.resource;

import static io.opentelemetry.sdk.testing.assertj.OpenTelemetryAssertions.assertThat;
import static io.opentelemetry.semconv.DeploymentAttributes.DEPLOYMENT_ENVIRONMENT_NAME;
import static io.opentelemetry.semconv.ServiceAttributes.SERVICE_INSTANCE_ID;
import static io.opentelemetry.semconv.ServiceAttributes.SERVICE_NAME;
import static io.opentelemetry.semconv.incubating.CloudIncubatingAttributes.CLOUD_PLATFORM;
import static io.opentelemetry.semconv.incubating.CloudIncubatingAttributes.CLOUD_PROVIDER;
import static io.opentelemetry.semconv.incubating.CloudIncubatingAttributes.CLOUD_REGION;
import static io.opentelemetry.semconv.incubating.CloudIncubatingAttributes.CLOUD_RESOURCE_ID;
import static io.opentelemetry.semconv.incubating.HostIncubatingAttributes.HOST_ID;

import com.google.common.collect.ImmutableMap;
import io.opentelemetry.sdk.testing.assertj.AttributesAssert;
import java.util.HashMap;
import java.util.Map;
import org.jetbrains.annotations.NotNull;
import org.junit.jupiter.api.Test;

class AzureAppServiceResourceProviderTest {

  private static final String TEST_WEBSITE_SITE_NAME = "TEST_WEBSITE_SITE_NAME";
  private static final String TEST_REGION_NAME = "TEST_REGION_NAME";
  private static final String TEST_WEBSITE_SLOT_NAME = "TEST_WEBSITE_SLOT_NAME";
  private static final String TEST_WEBSITE_HOSTNAME = "TEST_WEBSITE_HOSTNAME";
  private static final String TEST_WEBSITE_INSTANCE_ID = "TEST_WEBSITE_INSTANCE_ID";
  private static final String TEST_WEBSITE_HOME_STAMPNAME = "TEST_WEBSITE_HOME_STAMPNAME";
  private static final String TEST_WEBSITE_RESOURCE_GROUP = "TEST_WEBSITE_RESOURCE_GROUP";
  private static final String TEST_WEBSITE_OWNER_NAME = "TEST_WEBSITE_OWNER_NAME";
  private static final ImmutableMap<String, String> DEFAULT_ENV_VARS =
      ImmutableMap.of(
          "WEBSITE_SITE_NAME", TEST_WEBSITE_SITE_NAME,
          "REGION_NAME", TEST_REGION_NAME,
          "WEBSITE_SLOT_NAME", TEST_WEBSITE_SLOT_NAME,
          "WEBSITE_HOSTNAME", TEST_WEBSITE_HOSTNAME,
          "WEBSITE_INSTANCE_ID", TEST_WEBSITE_INSTANCE_ID,
          "WEBSITE_HOME_STAMPNAME", TEST_WEBSITE_HOME_STAMPNAME,
          "WEBSITE_RESOURCE_GROUP", TEST_WEBSITE_RESOURCE_GROUP,
          "WEBSITE_OWNER_NAME", TEST_WEBSITE_OWNER_NAME);

  @Test
  void defaultValues() {
    createResource(DEFAULT_ENV_VARS)
        .containsEntry(SERVICE_NAME, TEST_WEBSITE_SITE_NAME)
        .containsEntry(CLOUD_PROVIDER, "azure")
        .containsEntry(CLOUD_PLATFORM, "azure.app_service")
        .containsEntry(
            CLOUD_RESOURCE_ID,
            "/subscriptions/TEST_WEBSITE_OWNER_NAME/resourceGroups/TEST_WEBSITE_RESOURCE_GROUP/providers/Microsoft.Web/sites/TEST_WEBSITE_SITE_NAME")
        .containsEntry(CLOUD_REGION, TEST_REGION_NAME)
        .containsEntry(DEPLOYMENT_ENVIRONMENT_NAME, TEST_WEBSITE_SLOT_NAME)
        .containsEntry(HOST_ID, TEST_WEBSITE_HOSTNAME)
        .containsEntry(SERVICE_INSTANCE_ID, TEST_WEBSITE_INSTANCE_ID)
        .containsEntry(
            AzureAppServiceResourceProvider.AZURE_APP_SERVICE_STAMP_RESOURCE_ATTRIBUTE,
            TEST_WEBSITE_HOME_STAMPNAME);
  }

  @Test
  void subscriptionFromOwner() {
    HashMap<String, String> map = new HashMap<>(DEFAULT_ENV_VARS);
    map.put("WEBSITE_OWNER_NAME", "foo+bar");

    createResource(map)
        .containsEntry(
            CLOUD_RESOURCE_ID,
            "/subscriptions/foo/resourceGroups/TEST_WEBSITE_RESOURCE_GROUP/providers/Microsoft.Web/sites/TEST_WEBSITE_SITE_NAME");
  }

  @Test
  void noResourceId() {
    HashMap<String, String> map = new HashMap<>(DEFAULT_ENV_VARS);
    map.remove("WEBSITE_RESOURCE_GROUP");

    createResource(map).doesNotContainKey(CLOUD_RESOURCE_ID);
  }

  @Test
  void noWebsite() {
    HashMap<String, String> map = new HashMap<>(DEFAULT_ENV_VARS);
    map.remove("WEBSITE_SITE_NAME");

    createResource(map).isEmpty();
  }

  @Test
  void isFunction() {
    HashMap<String, String> map = new HashMap<>(DEFAULT_ENV_VARS);
    map.put("FUNCTIONS_EXTENSION_VERSION", "3.0");

    createResource(map).isEmpty();
  }

  @NotNull
  private static AttributesAssert createResource(Map<String, String> map) {
    return assertThat(
        new AzureAppServiceResourceProvider(map).createResource(null).getAttributes());
  }
}