File: test_options.py

package info (click to toggle)
grilo-plugins 0.3.18-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 8,348 kB
  • sloc: ansic: 52,656; xml: 2,879; python: 1,137; makefile: 9
file content (196 lines) | stat: -rw-r--r-- 8,799 bytes parent folder | download | duplicates (6)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
import unittest
import logging

try:
    from gi.repository import Grl
except:
    logging.warning("Unable to import Grilo's introspection bindings")
    exit()
try:
    from gi.repository import GLib
except TypeError:
    # there seems to be a bug with GVariants these days that triggers an
    # exception when importing GLib. Seems to be harmless though. We import
    # explicitly GLib in a try/except here so that we can handle that, else the
    # exception would be raised the first time it is imported by other code
    # (e.g. by using a GDateTime)
    from gi.repository import GLib

OPTION_TYPE_FILTER = "type-filter"

class TestCaps(unittest.TestCase):
    def test_creation(self):
        caps = Grl.Caps()
        self.assertFalse(caps is None)

    def test_mandatory_options(self):
        return None
        caps = Grl.Caps()
        for key in ("skip", "count", "flags"):
            self.assertTrue(caps.test_option(key, 0),
                            "test_option() returned False for %s" % key)

    def test_type_filter(self):
        caps = Grl.Caps()
        # test default value
        self.assertEqual(caps.get_type_filter(), Grl.TypeFilter.NONE)

        self.assertTrue(caps.test_option(OPTION_TYPE_FILTER, Grl.TypeFilter.NONE))
        self.assertFalse(caps.test_option(OPTION_TYPE_FILTER, Grl.TypeFilter.AUDIO))

        caps.set_type_filter(Grl.TypeFilter.VIDEO)
        self.assertEqual(caps.get_type_filter(), Grl.TypeFilter.VIDEO)
        self.assertTrue(caps.test_option(OPTION_TYPE_FILTER, Grl.TypeFilter.NONE))
        self.assertTrue(caps.test_option(OPTION_TYPE_FILTER, Grl.TypeFilter.VIDEO))
        self.assertFalse(caps.test_option(OPTION_TYPE_FILTER, Grl.TypeFilter.AUDIO))
        self.assertFalse(caps.test_option(OPTION_TYPE_FILTER,
                                          Grl.TypeFilter.AUDIO | Grl.TypeFilter.VIDEO))

    def test_key_filter(self):
        caps = Grl.Caps()
        self.assertEqual(caps.get_key_filter(), [])
        keys = [Grl.METADATA_KEY_ARTIST, Grl.METADATA_KEY_ALBUM]
        caps.set_key_filter(keys)
        self.assertEqual(caps.get_key_filter(), keys)
        self.assertTrue(caps.is_key_filter(Grl.METADATA_KEY_ARTIST))
        self.assertTrue(caps.is_key_filter(Grl.METADATA_KEY_ALBUM))
        self.assertFalse(caps.is_key_filter(Grl.METADATA_KEY_INVALID))
        self.assertFalse(caps.is_key_filter(Grl.METADATA_KEY_MODIFICATION_DATE))


class TestOptions(unittest.TestCase):
    def test_creation(self):
        caps = Grl.Caps()
        self.assertFalse(caps is None)
        options = Grl.OperationOptions.new(caps)
        self.assertFalse(options is None)
        self.assertFalse(Grl.OperationOptions() is None)

    def test_default_values(self):
        options = Grl.OperationOptions()
        self.assertEqual(options.get_skip(), 0)
        self.assertEqual(options.get_count(), Grl.COUNT_INFINITY)
        self.assertEqual(options.get_flags(),
                         Grl.MetadataResolutionFlags.NORMAL)
        self.assertEqual(options.get_type_filter(), Grl.TypeFilter.ALL)


    def test_value_setting_no_caps(self):
        options = Grl.OperationOptions()

        options.set_skip(12)
        self.assertEqual(options.get_skip(), 12)

        options.set_count(28)
        self.assertEqual(options.get_count(), 28)

        flags =Grl.MetadataResolutionFlags.FAST_ONLY \
                | Grl.MetadataResolutionFlags.IDLE_RELAY
        options.set_flags(flags)
        self.assertEqual(options.get_flags(), flags)

        fltr = Grl.TypeFilter.AUDIO | Grl.TypeFilter.VIDEO
        options.set_type_filter(fltr)
        self.assertEqual(options.get_type_filter(), fltr)

    def test_caps(self):
        caps = Grl.Caps()
        caps.set_type_filter(Grl.TypeFilter.VIDEO)
        options = Grl.OperationOptions.new(caps)
        self.assertTrue(options.set_type_filter(Grl.TypeFilter.VIDEO))
        self.assertFalse(options.set_type_filter(Grl.TypeFilter.AUDIO))
        self.assertFalse(options.set_type_filter(Grl.TypeFilter.IMAGE))

        # now with other caps, test obey_caps()
        caps2 = Grl.Caps()
        caps2.set_type_filter(Grl.TypeFilter.AUDIO)
        ret, supported, unsupported = options.obey_caps(caps2)
        self.assertFalse(ret)
        self.assertFalse(supported.key_is_set(Grl.OPERATION_OPTION_SKIP))
        self.assertFalse(supported.key_is_set(Grl.OPERATION_OPTION_COUNT))
        self.assertFalse(supported.key_is_set(Grl.OPERATION_OPTION_FLAGS))
        self.assertFalse(supported.key_is_set(OPTION_TYPE_FILTER))

        self.assertFalse(unsupported.key_is_set(Grl.OPERATION_OPTION_SKIP))
        self.assertFalse(unsupported.key_is_set(Grl.OPERATION_OPTION_COUNT))
        self.assertFalse(unsupported.key_is_set(Grl.OPERATION_OPTION_FLAGS))
        self.assertTrue(unsupported.key_is_set(OPTION_TYPE_FILTER))
        self.assertEqual(unsupported.get_type_filter(), Grl.TypeFilter.VIDEO)

        caps3 = Grl.Caps()
        caps3.set_type_filter(Grl.TypeFilter.VIDEO)
        ret, supported, unsupported = options.obey_caps(caps3)
        self.assertTrue(ret)
        self.assertFalse(supported.key_is_set(Grl.OPERATION_OPTION_SKIP))
        self.assertFalse(supported.key_is_set(Grl.OPERATION_OPTION_COUNT))
        self.assertFalse(supported.key_is_set(Grl.OPERATION_OPTION_FLAGS))
        self.assertTrue(supported.key_is_set(OPTION_TYPE_FILTER))
        self.assertEqual(supported.get_type_filter(), Grl.TypeFilter.VIDEO)

        self.assertFalse(unsupported.key_is_set(Grl.OPERATION_OPTION_SKIP))
        self.assertFalse(unsupported.key_is_set(Grl.OPERATION_OPTION_COUNT))
        self.assertFalse(unsupported.key_is_set(Grl.OPERATION_OPTION_FLAGS))
        self.assertFalse(unsupported.key_is_set(OPTION_TYPE_FILTER))

    def test_key_filters(self):
        options = Grl.OperationOptions.new(None)
        release_date = GLib.DateTime.new_utc(1994,1,1,12,0,0)
        self.assertEqual(options.get_key_filter_list(), [])
        self.assertTrue(options.set_key_filter_value(Grl.METADATA_KEY_ARTIST, "NOFX"))
        self.assertTrue(options.set_key_filters({
                                                Grl.METADATA_KEY_ALBUM: "Punk in Drublic",
                                                Grl.METADATA_KEY_CREATION_DATE: release_date}))
        filter_list = options.get_key_filter_list()
        filter_list.sort()
        expected_filters = [Grl.METADATA_KEY_ARTIST,
                            Grl.METADATA_KEY_ALBUM,
                            Grl.METADATA_KEY_CREATION_DATE]
        expected_filters.sort()
        self.assertEqual(filter_list, expected_filters)
        self.assertEqual(options.get_key_filter(Grl.METADATA_KEY_ARTIST), "NOFX")
        self.assertEqual(options.get_key_filter(Grl.METADATA_KEY_ALBUM), "Punk in Drublic")
        filter_date = options.get_key_filter(Grl.METADATA_KEY_CREATION_DATE)
        self.assertTrue(filter_date is not None)
        self.assertEqual(filter_date.format("%FT%H:%M:%S:%Z"), release_date.format("%FT%H:%M:%S:%Z"))

    def test_key_filters_with_caps(self):
        caps = Grl.Caps()
        release_date = GLib.DateTime.new_utc(1994,1,1,12,0,0)
        caps.set_key_filter([Grl.METADATA_KEY_ARTIST, Grl.METADATA_KEY_CREATION_DATE])
        options = Grl.OperationOptions.new(caps)
        self.assertTrue(options.set_key_filter_value(Grl.METADATA_KEY_ARTIST, "NOFX"))
        self.assertFalse(options.set_key_filter_value(Grl.METADATA_KEY_ALBUM, "Punk in Drublic"))
        self.assertTrue(options.set_key_filter_value(Grl.METADATA_KEY_CREATION_DATE, release_date))
        filter_list = options.get_key_filter_list()
        filter_list.sort()
        expected_filters = [Grl.METADATA_KEY_ARTIST,
                            Grl.METADATA_KEY_CREATION_DATE]
        expected_filters.sort()
        self.assertEqual(filter_list, expected_filters)
        self.assertEqual(options.get_key_filter(Grl.METADATA_KEY_ARTIST), "NOFX")
        self.assertEqual(options.get_key_filter(Grl.METADATA_KEY_ALBUM), None)

        filter_date = options.get_key_filter(Grl.METADATA_KEY_CREATION_DATE)
        self.assertTrue(filter_date is not None)
        self.assertEqual(filter_date.format("%FT%H:%M:%S:%Z"), release_date.format("%FT%H:%M:%S:%Z"))



class TestFileSystem(unittest.TestCase):
    def __init__(self, method_name):
        super(TestFileSystem, self).__init__(method_name)
        self.registry = Grl.Registry.get_default()
        self.plugin = self.registry.lookup_source("grl-filesystem")

    def test_caps(self):
        caps = self.plugin.get_caps(Grl.SupportedOps.BROWSE)
        self.assertFalse(caps is None)


# who said "this is ugly" ?
Grl.init([])
registry = Grl.Registry.get_default()
try:
    registry.load_plugin_by_id("grl-filesystem")
except GLib.Error:
    pass