Small test changes (#419)

PR-URL: #419
googleapis · Feb 25, 2017 · 3933142 · 3933142
1 parent eacfa15
commit 3933142
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 34 deletions.
diff --git a/test/test-trace-koa.js → test/plugins/test-trace-koa.js b/test/test-trace-koa.js → test/plugins/test-trace-koa.js
@@ -15,20 +15,20 @@
  */
 'use strict';
 
-var common = require('./plugins/common.js');
+var common = require('./common.js');
 var http = require('http');
 var assert = require('assert');
-var constants = require('../src/constants.js');
-var TraceLabels = require('../src/trace-labels.js');
+var constants = require('../../src/constants.js');
+var TraceLabels = require('../../src/trace-labels.js');
 
 var server;
 
 describe('test-trace-koa', function() {
   var agent;
   var koa;
   before(function() {
-    agent = require('..').start();
-    koa = require('./plugins/fixtures/koa1');
+    agent = require('../..').start();
+    koa = require('./fixtures/koa1');
   });
 
   afterEach(function() {

diff --git a/test/test-constants.js b/test/test-constants.js
diff --git a/test/test-trace-api.js b/test/test-trace-api.js
@@ -60,7 +60,7 @@ describe('Trace Interface', function() {
     traceAPI.enable_(agent);
     assert.ok(traceAPI.isActive(),
       'Being enabled sets operational implementation');
-    traceAPI.disable_(agent);
+    traceAPI.disable_();
     assert.ok(!traceAPI.isActive(),
       'Being disabled sets no-op implementation');
   });
@@ -70,10 +70,25 @@ describe('Trace Interface', function() {
     assertAPISurface(traceAPI);
     traceAPI.enable_(agent);
     assertAPISurface(traceAPI);
-    traceAPI.disable_(agent);
+    traceAPI.disable_();
     assertAPISurface(traceAPI);
   });
 
+  describe('constants', function() {
+    it('have correct values', function() {
+      var traceAPI = new TraceAPI('test');
+      assert.equal(traceAPI.constants.TRACE_CONTEXT_HEADER_NAME,
+        'x-cloud-trace-context');
+      assert.equal(traceAPI.constants.TRACE_AGENT_REQUEST_HEADER,
+        'x-cloud-trace-agent-request');
+      traceAPI.enable_(agent);
+      assert.equal(traceAPI.constants.TRACE_CONTEXT_HEADER_NAME,
+        'x-cloud-trace-context');
+      assert.equal(traceAPI.constants.TRACE_AGENT_REQUEST_HEADER,
+        'x-cloud-trace-agent-request');
+    });
+  });
+
   describe('behavior when initialized', function() {
     var traceAPI = new TraceAPI('test');