More tests

Author: yorch

test/unit-test/metric-middleware-koa-test.js

@@ -788,7 +788,7 @@ describe('metrics-middleware', () => {
                 Prometheus.register.clear();
             });
         });
-        describe('when excludeDefaultMetricLabels is passed', () => {
+        describe('when calling the function with excludeDefaultMetricLabels option', () => {
             it('or it\'s undefined', () => {
                 middleware();
                 expect(Prometheus.register.getSingleMetric('http_request_size_bytes').labelNames).to.have.members(['method', 'route', 'code']);
@@ -822,5 +822,43 @@ describe('metrics-middleware', () => {
                 Prometheus.register.clear();
             });
         });
+        describe('when calling the function with useCountersForRequestSizeMetric option', () => {
+            before(() => {
+                middleware({
+                    useCountersForRequestSizeMetric: true
+                });
+            });
+            it('shouldn\'t have http_request_size_bytes metric', () => {
+                expect(Prometheus.register.getSingleMetric('http_request_size_bytes')).to.equal(undefined);
+            });
+            it('should have http_request_size_bytes_sum with the right labels', () => {
+                expect(Prometheus.register.getSingleMetric('http_request_size_bytes_sum').labelNames).to.have.members(['method', 'route', 'code']);
+            });
+            it('should have http_request_size_bytes_count with the right labels', () => {
+                expect(Prometheus.register.getSingleMetric('http_request_size_bytes_count').labelNames).to.have.members(['method', 'route', 'code']);
+            });
+            after(() => {
+                Prometheus.register.clear();
+            });
+        });
+        describe('when calling the function with useCountersForResponseSizeMetric option', () => {
+            before(() => {
+                middleware({
+                    useCountersForResponseSizeMetric: true
+                });
+            });
+            it('shouldn\'t have http_response_size_bytes metric', () => {
+                expect(Prometheus.register.getSingleMetric('http_response_size_bytes')).to.equal(undefined);
+            });
+            it('should have http_response_size_bytes_sum with the right labels', () => {
+                expect(Prometheus.register.getSingleMetric('http_response_size_bytes_sum').labelNames).to.have.members(['method', 'route', 'code']);
+            });
+            it('should have http_response_size_bytes_count with the right labels', () => {
+                expect(Prometheus.register.getSingleMetric('http_response_size_bytes_count').labelNames).to.have.members(['method', 'route', 'code']);
+            });
+            after(() => {
+                Prometheus.register.clear();
+            });
+        });
     });
 });