test: add one for mock environment that is now shared in 2 test files

Signed-off-by: Naseem <[email protected]>
open-telemetry · Jul 27, 2020 · 03bc84d · 03bc84d
1 parent 8b313d5
commit 03bc84d
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/packages/opentelemetry-core/test/utils/environment.test.ts b/packages/opentelemetry-core/test/utils/environment.test.ts
@@ -42,12 +42,12 @@ describe('environment', () => {
       mockEnvironment({
         FOO: '1',
         OTEL_NO_PATCH_MODULES: 'a,b,c',
-        OTEL_LOG_LEVEL: 'WARN',
+        OTEL_LOG_LEVEL: 'ERROR',
         OTEL_SAMPLING_PROBABILITY: '0.5',
       });
       const env = getEnv();
       assert.strictEqual(env.OTEL_NO_PATCH_MODULES, 'a,b,c');
-      assert.strictEqual(env.OTEL_LOG_LEVEL, LogLevel.WARN);
+      assert.strictEqual(env.OTEL_LOG_LEVEL, LogLevel.ERROR);
       assert.strictEqual(env.OTEL_SAMPLING_PROBABILITY, 0.5);
     });
 
@@ -71,4 +71,17 @@ describe('environment', () => {
       });
     });
   });
+
+  describe('mockEnvironment', () => {
+    it('should remove a mock environment', () => {
+      mockEnvironment({
+        OTEL_LOG_LEVEL: 'DEBUG',
+        OTEL_SAMPLING_PROBABILITY: 0.5,
+      });
+      removeMockEnvironment();
+      const env = getEnv();
+      assert.strictEqual(env.OTEL_LOG_LEVEL, LogLevel.INFO);
+      assert.strictEqual(env.OTEL_SAMPLING_PROBABILITY, 1);
+    });
+  });
 });