Fix workspace list so it can handle additional fields in the response (…

…#210)
databricks · Feb 13, 2019 · 57d28b5 · 57d28b5
1 parent 3cd9d58
commit 57d28b5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/databricks_cli/workspace/api.py b/databricks_cli/workspace/api.py
@@ -37,7 +37,7 @@
 
 
 class WorkspaceFileInfo(object):
-    def __init__(self, path, object_type, language=None):
+    def __init__(self, path, object_type, language=None, **kwargs): # noqa
         self.path = path
         self.object_type = object_type
         self.language = language

diff --git a/tests/workspace/test_api.py b/tests/workspace/test_api.py
@@ -33,7 +33,8 @@
 TEST_WORKSPACE_PATH = '/test/workspace/path'
 TEST_JSON_RESPONSE = {
     'path': TEST_WORKSPACE_PATH,
-    'object_type': api.DIRECTORY
+    'object_type': api.DIRECTORY,
+    'created_at': 124
 }
 TEST_LANGUAGE = 'PYTHON'
 TEST_FMT = 'SOURCE'