500-writeback.py 6.15 KB
Newer Older
Ivan Frade's avatar
Ivan Frade committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
#!/usr/bin/python

# Copyright (C) 2010, Nokia (ivan.frade@nokia.com)
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the
# Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
# Boston, MA  02110-1301, USA.
#
"""
Write values in tracker and check the actual values are written
on the files. Note that these tests are highly platform dependant.
"""
24
import os
Ivan Frade's avatar
Ivan Frade committed
25 26
import time

27 28
from common.utils.extractor import get_tracker_extract_jsonld_output
from common.utils.helpers import log
29
from common.utils.writebacktest import CommonTrackerWritebackTest as CommonTrackerWritebackTest
30
import unittest as ut
Ivan Frade's avatar
Ivan Frade committed
31 32 33

REASONABLE_TIMEOUT = 5 # Seconds we wait for tracker-writeback to do the work

34
class WritebackBasicDataTest (CommonTrackerWritebackTest):
Ivan Frade's avatar
Ivan Frade committed
35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
    """
    Write in tracker store the properties witih writeback support and check
    that the new values are actually in the file
    """

    def __writeback_test (self, filename, mimetype, prop, expectedKey=None):
        """
        Set a value in @prop for the @filename. Then ask tracker-extractor
        for metadata and check in the results dictionary if the property is there.

        Note: given the special translation of some property-names in the dictionary
        with extracted metadata, there is an optional parameter @expectedKey
        to specify what property to check in the dictionary. If None, then
        the @prop is used.
        """

51 52 53 54
        # FIXME: filename is actually a URI! :(
        filename_real = filename[len('file://'):]
        initial_mtime = os.stat(filename_real).st_mtime

Ivan Frade's avatar
Ivan Frade committed
55 56
        TEST_VALUE = prop.replace (":","") + "test"
        SPARQL_TMPL = """
57
           DELETE { ?u %s ?v } WHERE { ?u nie:url '%s' ; %s ?v }
Ivan Frade's avatar
Ivan Frade committed
58 59 60
           INSERT { ?u %s '%s' }
           WHERE  { ?u nie:url '%s' }
        """ 
61
        self.tracker.update (SPARQL_TMPL % (prop, filename, prop, prop, TEST_VALUE, filename))
62

63
        log("Waiting for change on %s" % filename_real)
64
        self.wait_for_file_change(filename_real, initial_mtime)
65
        log("Got the change")
66

67
        results = get_tracker_extract_jsonld_output (filename, mimetype)
Ivan Frade's avatar
Ivan Frade committed
68
        keyDict = expectedKey or prop
69
        self.assertIn (TEST_VALUE, results[keyDict])
Ivan Frade's avatar
Ivan Frade committed
70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92

    def __writeback_hasTag_test (self, filename, mimetype):

        SPARQL_TMPL = """
            INSERT {
              <test://writeback-hasTag-test/1> a nao:Tag ;
                        nao:prefLabel "testTag" .

              ?u nao:hasTag <test://writeback-hasTag-test/1> .
            } WHERE {
              ?u nie:url '%s' .
            }
        """

        CLEAN_VALUE = """
           DELETE {
              <test://writeback-hasTag-test/1> a rdfs:Resource.
              ?u nao:hasTag <test://writeback-hasTag-test/1> .
           } WHERE {
              ?u nao:hasTag <test://writeback-hasTag-test/1> .
           }
        """

93
        self.tracker.update (SPARQL_TMPL % (filename))
Ivan Frade's avatar
Ivan Frade committed
94 95 96

        time.sleep (REASONABLE_TIMEOUT)

97
        results = get_tracker_extract_jsonld_output (filename, mimetype)
98
        self.assertIn ("testTag", results ["nao:hasTag"])
Ivan Frade's avatar
Ivan Frade committed
99 100 101 102


    # JPEG test
    def test_001_jpeg_title (self):
103 104
        #FILENAME = "test-writeback-monitored/writeback-test-1.jpeg"
        self.__writeback_test (self.get_test_filename_jpeg (), "image/jpeg", "nie:title")
Ivan Frade's avatar
Ivan Frade committed
105 106

    def test_002_jpeg_description (self):
107 108
        #FILENAME = "test-writeback-monitored/writeback-test-1.jpeg"
        self.__writeback_test (self.get_test_filename_jpeg (), "image/jpeg", "nie:description")
Ivan Frade's avatar
Ivan Frade committed
109

110 111 112 113
    #def test_003_jpeg_keyword (self):
    #    #FILENAME = "test-writeback-monitored/writeback-test-1.jpeg"
    #    self.__writeback_test (self.get_test_filename_jpeg (), "image/jpeg",
    #                           "nie:keyword", "nao:hasTag")
Ivan Frade's avatar
Ivan Frade committed
114

115 116 117
    #def test_004_jpeg_hasTag (self):
    #    #FILENAME = "test-writeback-monitored/writeback-test-1.jpeg"
    #    self.__writeback_hasTag_test (self.get_test_filename_jpeg (), "image/jpeg")
Ivan Frade's avatar
Ivan Frade committed
118 119 120 121

        
    # TIFF tests
    def test_011_tiff_title (self):
122 123
        #FILANAME = "test-writeback-monitored/writeback-test-2.tif"
        self.__writeback_test (self.get_test_filename_tiff (), "image/tiff", "nie:title")
Ivan Frade's avatar
Ivan Frade committed
124 125 126

    def test_012_tiff_description (self):
        FILENAME = "test-writeback-monitored/writeback-test-2.tif"
127
        self.__writeback_test (self.get_test_filename_tiff (), "image/tiff", "nie:description")
Ivan Frade's avatar
Ivan Frade committed
128
        
129 130 131 132 133 134 135 136
    #def test_013_tiff_keyword (self):
    #    FILENAME = "test-writeback-monitored/writeback-test-2.tif"
    #    self.__writeback_test (self.get_test_filename_tiff (), "image/tiff",
    #                           "nie:keyword", "nao:hasTag")

    #def test_014_tiff_hasTag (self):
    #    FILENAME = "test-writeback-monitored/writeback-test-2.tif"
    #    self.__writeback_hasTag_test (self.get_test_filename_tiff (), "image/tiff")
Ivan Frade's avatar
Ivan Frade committed
137 138 139 140 141 142
      
        

    # PNG tests
    def test_021_png_title (self):
        FILENAME = "test-writeback-monitored/writeback-test-4.png"
143
        self.__writeback_test (self.get_test_filename_png (), "image/png", "nie:title")
Ivan Frade's avatar
Ivan Frade committed
144 145 146

    def test_022_png_description (self):
        FILENAME = "test-writeback-monitored/writeback-test-4.png"
147 148
        self.__writeback_test (self.get_test_filename_png (), "image/png", "nie:description")

149 150
    #def test_023_png_keyword (self):
    #    FILENAME = "test-writeback-monitored/writeback-test-4.png"
151
    #    self.__writeback_test (self.get_test_filename_png (), "image/png", "nie:keyword", "nao:hasTag:prefLabel")
152 153 154

    #def test_024_png_hasTag (self):
    #    FILENAME = "test-writeback-monitored/writeback-test-4.png"
155
    #    self.__writeback_hasTag_test (self.get_test_filename_png (), "image/png")
Ivan Frade's avatar
Ivan Frade committed
156 157

if __name__ == "__main__":
158
    ut.main (failfast=True)