fix: Gracefully handle variant metrics

lib/client-metrics/index.js

@@ -92,8 +92,19 @@ module.exports = class UnleashClientMetrics {
     }
 
     createCountObject(entry) {
-        const yes = typeof entry.yes == 'number' ? entry.yes : 0;
+        let yes = typeof entry.yes == 'number' ? entry.yes : 0;
-        const no = typeof entry.no == 'number' ? entry.no : 0;
+        let no = typeof entry.no == 'number' ? entry.no : 0;
+
+        if (entry.variants) {
+            Object.entries(entry.variants).forEach(([key, value]) => {
+                if (key === 'disabled') {
+                    no += value;
+                } else {
+                    yes += value;
+                }
+            });
+        }
+
         return { yes, no };
     }
 

lib/routes/client-api/metrics-schema.js

@@ -8,14 +8,16 @@ const countSchema = joi
     .keys({
         yes: joi
             .number()
-            .required()
             .min(0)
             .default(0),
         no: joi
             .number()
-            .required()
             .min(0)
             .default(0),
+        variants: joi.object({
+            arg: joi.string(),
+            value: joi.number().min(0),
+        }),
     });
 
 const clientMetricsSchema = joi

lib/routes/client-api/metrics.test.js

@@ -70,7 +70,33 @@ test('should accept client metrics with yes/no', t => {
         .expect(202);
 });
 
-test('should not accept client metrics without yes/no', t => {
+test('should accept client metrics with variants', t => {
+    t.plan(0);
+    const { request } = getSetup();
+    return request
+        .post('/api/client/metrics')
+        .send({
+            appName: 'demo',
+            instanceId: '1',
+            bucket: {
+                start: Date.now(),
+                stop: Date.now(),
+                toggles: {
+                    toggleA: {
+                        yes: 200,
+                        no: 0,
+                        variants: {
+                            variant1: 1,
+                            variant2: 2,
+                        },
+                    },
+                },
+            },
+        })
+        .expect(202);
+});
+
+test('should accept client metrics without yes/no', t => {
     t.plan(0);
     const { request } = getSetup();
     return request
@@ -89,5 +115,5 @@ test('should not accept client metrics without yes/no', t => {
                 },
             },
         })
-        .expect(400);
+        .expect(202);
 });