From 1b48c2762cf2c970ff185a1b48a00b2664b00142 Mon Sep 17 00:00:00 2001
From: Jeff Irion <JeffLIrion@users.noreply.github.com>
Date: Wed, 4 Sep 2019 08:34:06 -0700
Subject: [PATCH] Add tests for the setup function (#85)

* Add tests for the setup function

* 'Value error' -> 'ValueError'

* Add missing quotes

* 'DEVICE_PROPERTIES_DICT2' -> 'DEVICE_PROPERTIES_DICT1'
---
 tests/test_setup.py | 72 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 72 insertions(+)
 create mode 100644 tests/test_setup.py

diff --git a/tests/test_setup.py b/tests/test_setup.py
new file mode 100644
index 00000000..270ce2aa
--- /dev/null
+++ b/tests/test_setup.py
@@ -0,0 +1,72 @@
+import sys
+import unittest
+
+try:
+    # Python3
+    from unittest.mock import patch
+except ImportError:
+    # Python2
+    from mock import patch
+
+
+sys.path.insert(0, '..')
+
+from androidtv import constants, ha_state_detection_rules_validator, setup
+from androidtv.androidtv import AndroidTV
+from androidtv.firetv import FireTV
+from . import patchers
+
+
+DEVICE_PROPERTIES_OUTPUT1 = """Amazon
+AFTT
+SERIALNO
+5.1.1
+    link/ether ab:cd:ef:gh:ij:kl brd ff:ff:ff:ff:ff:ff
+Device "eth0" does not exist.
+"""
+
+DEVICE_PROPERTIES_DICT1 = {'manufacturer': 'Amazon',
+                           'model': 'AFTT',
+                           'serialno': 'SERIALNO',
+                           'sw_version': '5.1.1',
+                           'wifimac': 'ab:cd:ef:gh:ij:kl',
+                           'ethmac': None}
+
+DEVICE_PROPERTIES_OUTPUT2 = """Not Amazon
+AFTT
+SERIALNO
+5.1.1
+Device "wlan0" does not exist.
+    link/ether ab:cd:ef:gh:ij:kl brd ff:ff:ff:ff:ff:ff
+"""
+
+DEVICE_PROPERTIES_DICT2 = {'manufacturer': 'Not Amazon',
+                           'model': 'AFTT',
+                           'serialno': 'SERIALNO',
+                           'sw_version': '5.1.1',
+                           'wifimac': None,
+                           'ethmac': 'ab:cd:ef:gh:ij:kl'}
+
+
+class TestSetup(unittest.TestCase):
+    PATCH_KEY = 'python'
+
+    def test_setup(self):
+        """Test that the ``setup`` function works correctly.
+        """
+        with self.assertRaises(ValueError):
+            setup('IP:PORT', device_class='INVALID')
+
+        with patchers.patch_connect(True)[self.PATCH_KEY], patchers.patch_shell(DEVICE_PROPERTIES_OUTPUT1)[self.PATCH_KEY]:
+            ftv = setup('IP:PORT')
+            self.assertIsInstance(ftv, FireTV)
+            self.assertDictEqual(ftv.device_properties, DEVICE_PROPERTIES_DICT1)
+
+        with patchers.patch_connect(True)[self.PATCH_KEY], patchers.patch_shell(DEVICE_PROPERTIES_OUTPUT2)[self.PATCH_KEY]:
+            atv = setup('IP:PORT')
+            self.assertIsInstance(atv, AndroidTV)
+            self.assertDictEqual(atv.device_properties, DEVICE_PROPERTIES_DICT2)
+
+
+if __name__ == "__main__":
+    unittest.main()