Пример #1
0
 def fill_out_resources(self, resources):
     """Fill out the resources"""
     for resource in as_list(resources):
         if isinstance(resource, six.string_types):
             yield resource
         elif isinstance(resource, dict):
             for resource in self.expand_resource(resource):
                 yield resource
         else:
             raise BadPolicy("Resource should be a string or a dictionary", resource=resource)
Пример #2
0
        for val in (0, 1, None, True, False, "", "adf", lambda: 1, mock.Mock(name="blah")):
            dct = {"blah": val}
            lst = listify(dct, "blah")
            self.assertEqual(lst, [val])
            self.assertEqual(dct, {"blah": lst})

describe TestCase, "listified":
    it "yields nothing if the key isn't in the dict":
        self.assertEqual(list(listified({"meh": 1}, "blah")), [])

    it "yields the thing if the thing is not a list":
        for val in (0, 1, None, True, False, "", "adf", lambda: 1, mock.Mock(name="blah")):
            self.assertEqual(list(listified({"blah": val}, "blah")), [val])

    it "yields the items in the thing if the thing is a list":
        item1 = 3
        item2 = 1
        item3 = 2
        lst = [item1, item2, item3]
        self.assertEqual(list(listified({"blah": lst}, "blah")), [item1, item2, item3])

describe TestCase, "as_list":
    it "yields the things in the thing if it's a list":
        lst = [1, 2]
        self.assertEqual(list(as_list(lst)), lst)

    it "yields just the thing if not a list":
        for val in (0, 1, None, True, False, "", "adf", lambda: 1, mock.Mock(name="blah")):
            self.assertEqual(list(as_list(val)), [val])