X-Git-Url: http://woldlab.caltech.edu/gitweb/?p=htsworkflow.git;a=blobdiff_plain;f=htsworkflow%2Fsubmission%2Ftest%2Ftest_submission.py;fp=htsworkflow%2Fsubmission%2Ftest%2Ftest_submission.py;h=90852ceb766f56a871fc199d675982fc095b98c2;hp=f362cea0f2e2a7235024e05296aed127fbf9f1ea;hb=2816ef120f54daaf9ef5d6f70bb19e379f5ab955;hpb=fbebf34fa2f5842a2ab1eb8319176c82587fe3bb

diff --git a/htsworkflow/submission/test/test_submission.py b/htsworkflow/submission/test/test_submission.py
index f362cea..90852ce 100644
--- a/htsworkflow/submission/test/test_submission.py
+++ b/htsworkflow/submission/test/test_submission.py
@@ -7,15 +7,40 @@ from unittest2 import TestCase, TestSuite, defaultTestLoader
 from htsworkflow.submission import daf, results
 from htsworkflow.util.rdfhelp import \
      dafTermOntology, \
+     dump_model, \
      fromTypedNode, \
+     get_turtle_header, \
      load_string_into_model, \
      rdfNS, \
      submissionLog, \
      submissionOntology, \
      get_model, \
      get_serializer
-from htsworkflow.submission.submission import list_submissions
+from htsworkflow.submission.submission import list_submissions, Submission
+from htsworkflow.submission.results import ResultMap
+from submission_test_common import *
+
 import RDF
+#import logging
+#logging.basicConfig(level=logging.DEBUG)
+
+def generate_sample_results_tree(obj):
+    obj.tempdir = tempfile.mkdtemp(prefix="submission_test")
+    obj.sourcedir = os.path.join(obj.tempdir, 'source')
+    obj.resultdir = os.path.join(obj.tempdir, 'results')
+
+    for d in [os.path.join(obj.tempdir, S1_NAME),
+              os.path.join(obj.tempdir, S2_NAME),
+              ]:
+        os.mkdir(os.path.join(obj.tempdir, d))
+
+    tomake = []
+    tomake.extend(S1_FILES)
+    tomake.extend(S2_FILES)
+    for f in tomake:
+        stream = open(os.path.join(obj.tempdir, f), 'w')
+        stream.write(f)
+        stream.close()
 
 class TestSubmissionModule(TestCase):
     def test_empty_list_submission(self):
@@ -57,10 +82,110 @@ class TestSubmissionModule(TestCase):
             testset.add(name)
         self.assertEqual(testset, truth)
 
+class TestSubmission(TestCase):
+    def setUp(self):
+        generate_sample_results_tree(self)
+        self.model = get_model()
+
+    def tearDown(self):
+        shutil.rmtree(self.tempdir)
+
+    def test_create_submission(self):
+        model = get_model()
+        s = Submission('foo', self.model, 'http://localhost')
+        self.assertEqual(str(s.submissionSet),
+                         "http://jumpgate.caltech.edu/wiki/SubmissionsLog/foo")
+        self.assertEqual(str(s.submissionSetNS['']),
+                         str(RDF.NS(str(s.submissionSet) + '#')['']))
+        self.assertEqual(str(s.libraryNS['']),
+                         str(RDF.NS('http://localhost/library/')['']))
+
+    def test_scan_submission_dirs(self):
+        turtle = get_turtle_header() + """
+@prefix thisView: <http://jumpgate.caltech.edu/wiki/SubmissionsLog/test/view/> .
+thisView:Fastq ucscDaf:filename_re ".*[^12]\\.fastq\\.bz2$" ;
+               a geoSoft:raw ;
+               geoSoft:fileTypeLabel "fastq" ;
+               ucscDaf:output_type "read" .
+thisView:FastqRead1 ucscDaf:filename_re ".*r1\\.fastq\\.bz2$" ;
+               a geoSoft:raw ;
+               geoSoft:fileTypeLabel "fastq" ;
+               ucscDaf:output_type "read1" .
+thisView:FastqRead2 ucscDaf:filename_re ".*r2\\.fastq\\.bz2$" ;
+               a geoSoft:raw ;
+               geoSoft:fileTypeLabel "fastq" ;
+               ucscDaf:output_type "read2" .
+thisView:alignments ucscDaf:filename_re ".*\\.bam$" ;
+               a geoSoft:supplemental ;
+               geoSoft:fileTypeLabel "bam" ;
+               ucscDaf:output_type "alignments" .
+
+        """
+        map = ResultMap()
+        print self.tempdir
+        print os.listdir(self.tempdir)
+        map['1000'] = os.path.join(self.tempdir, S1_NAME)
+        map['2000'] = os.path.join(self.tempdir, S2_NAME)
+
+        s = Submission('foo', self.model, 'http://localhost')
+        mock = MockAddDetails(self.model, turtle)
+        mock.add_turtle(S1_TURTLE)
+        mock.add_turtle(S2_TURTLE)
+        s._add_library_details_to_model =  mock
+        s.scan_submission_dirs(map)
+
+        nodes = list(s.analysis_nodes(map))
+        self.assertEqual(len(nodes), 2)
+        expected = set((
+            'http://jumpgate.caltech.edu/wiki/SubmissionsLog/foo#1000-sample',
+            'http://jumpgate.caltech.edu/wiki/SubmissionsLog/foo#2000-sample',
+        ))
+        got = set((str(nodes[0]), str(nodes[1])))
+        self.assertEqual(expected, got)
+
+    def test_find_best_match(self):
+        turtle = get_turtle_header() + """
+@prefix thisView: <http://jumpgate.caltech.edu/wiki/SubmissionsLog/test/view/> .
+thisView:Fastq ucscDaf:filename_re ".*[^12]\\.fastq\\.bz2$" ;
+               a geoSoft:raw ;
+               geoSoft:fileTypeLabel "fastq" ;
+               ucscDaf:output_type "read" .
+thisView:FastqRead1 ucscDaf:filename_re ".*r1\\.fastq\\.bz2$" ;
+               a geoSoft:raw ;
+               geoSoft:fileTypeLabel "fastq" ;
+               ucscDaf:output_type "read1" .
+thisView:FastqRead2 ucscDaf:filename_re ".*r2\\.fastq\\.bz2$" ;
+               a geoSoft:raw ;
+               geoSoft:fileTypeLabel "fastq" ;
+               ucscDaf:output_type "read2" .
+thisView:alignments ucscDaf:filename_re ".*\\.bam$" ;
+               a geoSoft:supplemental ;
+               geoSoft:fileTypeLabel "bam" ;
+               ucscDaf:output_type "alignments" .
+
+        """
+        load_string_into_model(self.model, 'turtle', turtle)
+        s = Submission('foo', self.model, 'http://localhost')
+        q = RDF.Statement(None, dafTermOntology['filename_re'], None)
+        view_map = s._get_filename_view_map()
+        self.assertEqual(len(view_map), 4)
+
+        fastq = s.find_best_match("asdf.fastq.bz2")
+        self.assertEqual(
+            str(fastq),
+            "http://jumpgate.caltech.edu/wiki/SubmissionsLog/test/view/Fastq")
+
+        fastq = s.find_best_match("asdf.r2.fastq.bz2")
+        self.assertEqual(
+            str(fastq),
+            "http://jumpgate.caltech.edu/wiki/SubmissionsLog/test/view/FastqRead2")
+
 def suite():
     suite = TestSuite()
     suite.addTests(
         defaultTestLoader.loadTestsFromTestCase(TestSubmissionModule))
+    suite.addTests(
+        defaultTestLoader.loadTestsFromTestCase(TestSubmission))
     return suite
 
 if __name__ == "__main__":