Features/dbfs

common/resultreports.py

@@ -32,6 +32,10 @@ def has_data(self):
     def write(self):
         pass
 
+    @abstractmethod
+    def write_to_dbfs(self, dbfs_path):
+        pass
+
     def _validate_add_results(self, notebook_path, test_result):
         if not isinstance(test_result, TestResults):
             raise ValueError('Expected an instance of TestResults')
@@ -85,6 +89,10 @@ def write(self):
 
         return report_name
 
+    def write_to_dbfs(self, dbfs_path):
+        self.to_file(dbfs_path)
+        return dbfs_path
+
     def to_file(self, path):
         file = open(path, 'w')
         try:
@@ -137,6 +145,10 @@ def write(self):
 
         return report_name
 
+    def write_to_dbfs(self, dbfs_path):
+        self.to_file(dbfs_path)
+        return dbfs_path
+
     def to_file(self, path):
         file = open(path, 'w')
         try:

common/testexecresults.py

@@ -4,6 +4,7 @@
 """
 
 from .apiclientresults import ExecuteNotebookResult, NotebookOutputResult
+from .resultreports import JunitXMLReportWriter, TagsReportWriter
 from .resultsview import RunCommandResultsView
 from .testresult import TestResults
 
@@ -31,3 +32,17 @@ def get_ExecuteNotebookResult(self, notebook_path, test_results):
             'N/A', None, test_results)
 
         return ExecuteNotebookResult('N/A', 'N/A', notebook_result, 'N/A')
+
+    def write_junit_xml_report_to_dbfs(self, notebook_path, dbfs_report_path):
+        junit_xml_report_writer = JunitXMLReportWriter()
+        junit_xml_report_writer.add_result(notebook_path, self.test_results)
+        junit_xml_report_writer.write_to_dbfs(dbfs_report_path)
+
+        return dbfs_report_path
+
+    def write_tags_report_to_dbfs(self, notebook_path, dbfs_report_path):
+        tags_report_writer = TagsReportWriter()
+        tags_report_writer.add_result(notebook_path, self.test_results)
+        tags_report_writer.write_to_dbfs(dbfs_report_path)
+
+        return dbfs_report_path