cseed
diff --git a/‎iot/api-client/codelabs/codelab_test.py
Lines changed: 70 additions & 0 deletions b/‎iot/api-client/codelabs/codelab_test.py
Lines changed: 70 additions & 0 deletions
diff --git a/‎iot/api-client/codelabs/colors.py
Lines changed: 68 additions & 0 deletions b/‎iot/api-client/codelabs/colors.py
Lines changed: 68 additions & 0 deletions
@@ -0,0 +1,70 @@
+# Copyright 2019 Google Inc. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+from colors import bcolors
+import gateway
+import ledlight
+import lightsensor
+import thermostat
+
+
+# Check colors exist
+def test_check_colors(capsys):
+    assert(bcolors.OKBLUE is not None)
+    assert(bcolors.FAIL is not None)
+    assert(bcolors.CEND is not None)
+
+
+# Check error code returns reasonable justifications
+def test_gateway_error_string(capsys):
+    print(gateway.error_str(1))
+    print(gateway.error_str(2))
+    print(gateway.error_str(3))
+    print(gateway.error_str(4))
+    out, _ = capsys.readouterr()
+
+    assert 'memory' in out
+    assert 'network' in out
+    assert 'function arguments' in out
+    assert 'currently connected' in out
+
+
+# Check gateway state is init to reasonable defaults
+def test_gateway_state(capsys):
+    assert (gateway.GatewayState.mqtt_config_topic == '')
+    assert (gateway.GatewayState.connected is False)
+    assert (gateway.GatewayState.pending_responses == {})
+    assert (gateway.GatewayState.pending_subscribes == {})
+    assert (gateway.GatewayState.mqtt_bridge_port == 8883 or
+            gateway.GatewayState.mqtt_bridge_ == 443)
+
+
+def test_check_ledlight(capsys):
+    print(ledlight.make_message('test', 'action', data='1234'))
+    out, _ = capsys.readouterr()
+
+    assert ('{ "device" : "test"' in out)
+
+
+def test_check_lightsensor(capsys):
+    print(lightsensor.print_sensor_state())
+    out, _ = capsys.readouterr()
+
+    assert ('Sensor is on, reporting lux every 1 seconds.' in out)
+
+
+def test_check_thermostat(capsys):
+    print(thermostat.make_message('test', 'action', data='1234'))
+    out, _ = capsys.readouterr()
+
+    assert ('{ "device" : "test"' in out)
@@ -0,0 +1,68 @@
+# Copyright 2019 Google Inc. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+class bcolors:
+    HEADER = '\033[95m'
+    OKBLUE = '\033[94m'
+    OKGREEN = '\033[92m'
+    WARNING = '\033[93m'
+    FAIL = '\033[91m'
+    ENDC = '\033[0m'
+    BOLD = '\033[1m'
+    UNDERLINE = '\033[4m'
+
+    CEND = '\33[0m'
+    CBOLD = '\33[1m'
+    CITALIC = '\33[3m'
+    CURL = '\33[4m'
+    CBLINK = '\33[5m'
+    CBLINK2 = '\33[6m'
+    CSELECTED = '\33[7m'
+
+    CBLACK = '\33[30m'
+    CRED = '\33[31m'
+    CGREEN = '\33[32m'
+    CYELLOW = '\33[33m'
+    CBLUE = '\33[34m'
+    CVIOLET = '\33[35m'
+    CBEIGE = '\33[36m'
+    CWHITE = '\33[37m'
+
+    CBLACKBG = '\33[40m'
+    CREDBG = '\33[41m'
+    CGREENBG = '\33[42m'
+    CYELLOWBG = '\33[43m'
+    CBLUEBG = '\33[44m'
+    CVIOLETBG = '\33[45m'
+    CBEIGEBG = '\33[46m'
+    CWHITEBG = '\33[47m'
+
+    CGREY = '\33[90m'
+    CRED2 = '\33[91m'
+    CGREEN2 = '\33[92m'
+    CYELLOW2 = '\33[93m'
+    CBLUE2 = '\33[94m'
+    CVIOLET2 = '\33[95m'
+    CBEIGE2 = '\33[96m'
+    CWHITE2 = '\33[97m'
+
+    CGREYBG = '\33[100m'
+    CREDBG2 = '\33[101m'
+    CGREENBG2 = '\33[102m'
+    CYELLOWBG2 = '\33[103m'
+    CBLUEBG2 = '\33[104m'
+    CVIOLETBG2 = '\33[105m'
+    CBEIGEBG2 = '\33[106m'
+    CWHITEBG2 = '\33[107m'