started on tests

Author: Kully

plotly/tests/test_core/test_tools/test_file_tools.py

@@ -1,6 +1,9 @@
 from plotly import tools, session
 from plotly.tests.utils import PlotlyTestCase
 
+import ipdb
+import warnings
+
 
 class FileToolsTest(PlotlyTestCase):
 
@@ -43,14 +46,23 @@ def test_set_config_file_two_entries(self):
         self.assertEqual(config['plotly_streaming_domain'], streaming_domain)
         tools.reset_config_file()
 
-
     def test_set_config_file_world_readable(self):
 
         # Return TypeError when world_readable type is not a bool
 
         kwargs = {'world_readable': 'True'}
         self.assertRaises(TypeError, tools.set_config_file, **kwargs)
 
+    def test_set_config_expected_error_msg(self):
+        kwargs = {'plotly_domain': 'http://www.foo-bar.com'}
+        #self.assertRaises(UserWarning, tools.set_config_file, **kwargs)
+        print 'abcd'
+        b = tools.set_config_file(**kwargs)
+        print type(b)
+
+
+
+
     def test_reset_config_file(self):
 
         # Check reset_config and get_config return the same values

plotly/tools.py

@@ -73,10 +73,10 @@
 )
 
 
-def validate_domains(*domains):
+def _validate_domains(*domains):
     for d in domains:
         if not d.lower().startswith('https'):
-            warnings.warn(http_msg)
+            warnings.warn(http_msg, category=UserWarning)
 
 
 # Warning format
@@ -255,7 +255,7 @@ def set_config_file(plotly_domain=None,
         list_of_domains.append(plotly_domain)
     if plotly_api_domain is not None:
         list_of_domains.append(plotly_api_domain)
-    validate_domains(*list_of_domains)
+    _validate_domains(*list_of_domains)
 
     if isinstance(world_readable, bool):
         settings['world_readable'] = world_readable
@@ -272,7 +272,7 @@ def set_config_file(plotly_domain=None,
     ensure_local_plotly_files()  # make sure what we just put there is OK
 
 
-def get_config_file(*args, validate=True):
+def get_config_file(validate=True, *args):
     """Return specified args from `~/.plotly/.config`. as tuple.
 
     Returns all if no arguments are specified.
@@ -293,13 +293,11 @@ def get_config_file(*args, validate=True):
             list_of_domains.append(returned_obj[domain])
 
     if validate:
-        validate_domains(*list_of_domains)
+        _validate_domains(*list_of_domains)
 
     return returned_obj
 
 
-
-
 def reset_config_file():
     ensure_local_plotly_files()  # make sure what's there is OK
     f = open(CONFIG_FILE, 'w')