From 7a3aae1c7e8a5b276912e7d3f280e47b1c542c03 Mon Sep 17 00:00:00 2001
From: "Mr. Outis" <mroutis@protonmail.com>
Date: Wed, 4 Dec 2019 19:43:00 -0600
Subject: [PATCH] tests: add unit test for analytics.system_info

---
 tests/unit/test_analytics.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/tests/unit/test_analytics.py b/tests/unit/test_analytics.py
index c56898721..4a2388c95 100644
--- a/tests/unit/test_analytics.py
+++ b/tests/unit/test_analytics.py
@@ -1,5 +1,7 @@
 import pytest
 import mock
+import platform
+import json
 
 from voluptuous import Schema, Any
 
@@ -65,6 +67,36 @@ def test_is_enabled(dvc_repo, config, result, monkeypatch):
     assert result == analytics.is_enabled()
 
 
+def test_system_info():
+    schema = Schema({"os": Any("windows", "mac", "linux")})
+
+    system = platform.system()
+
+    if system == "Windows":
+        schema = schema.extend(
+            {
+                "windows_version_build": int,
+                "windows_version_major": int,
+                "windows_version_minor": int,
+                "windows_version_service_pack": str,
+            }
+        )
+
+    if system == "Darwin":
+        schema = schema.extend({"mac_version": str})
+
+    if system == "Linux":
+        schema = schema.extend(
+            {
+                "linux_distro": str,
+                "linux_distro_like": Any(str, None),
+                "linux_distro_version": Any(str, None),
+            }
+        )
+
+    assert schema(analytics.system_info())
+
+
 def test_find_or_create_user_id(tmp_global_config):
     created = analytics.find_or_create_user_id()
     found = analytics.find_or_create_user_id()