Fix tests in Python 2

gmr · Feb 27, 2018 · 714e314 · 714e314
1 parent 09dab48
commit 714e314
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tests/agent_tests.py b/tests/agent_tests.py
@@ -5,6 +5,7 @@
 import uuid
 
 import consulate
+from consulate import utils
 
 from . import base
 
@@ -55,7 +56,7 @@ def test_metrics_forbidden(self):
 
     def test_monitor(self):
         for line in self.consul.agent.monitor():
-            self.assertIsInstance(line, str)
+            self.assertTrue(utils.is_string(line))
             break
 
     def test_monitor_forbidden(self):

diff --git a/tests/kv_tests.py b/tests/kv_tests.py
@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import json
 import unittest
 try: