Ejemplo n.º 1
0
    def test_load_from_file(self):
        path_to_file = '/path/to/file.txt'
        f = [1, 2, 3]

        with mock.patch('pyfeddic.record.open', create=True) as mock_open:
            mock_open.return_value = f

            rf = RecordFactory(path_to_file, mock.Mock())
            rf.load_db()

        self.assertTrue(mock_open.call_count)
        args, _ = mock_open.call_args
        self.assertEqual(args[0], path_to_file)
        self.assertEqual(len(rf.records), len(f))
Ejemplo n.º 2
0
    def test_load_from_url(self):
        url = 'http://www.example.com/file'
        f = [1, 2, 3]

        with mock.patch('pyfeddic.record.urllib') as urllib:
            urllib.urlopen.return_value = f

            rf = RecordFactory(url, mock.Mock())
            rf.load_db()

        self.assertEqual(urllib.urlopen.call_count, 1)
        args, _ = urllib.urlopen.call_args
        self.assertEqual(args[0], url)
        self.assertEqual(len(rf.records), len(f))
Ejemplo n.º 3
0
    def test_load_from_file(self):
        path_to_file = '/path/to/file.txt'
        f = [1, 2, 3]

        with mock.patch('pyfeddic.record.open', create=True) as mock_open:
            mock_open.return_value = f

            rf = RecordFactory(path_to_file, mock.Mock())
            rf.load_db()

        self.assertTrue(mock_open.call_count)
        args, _ = mock_open.call_args
        self.assertEqual(args[0], path_to_file)
        self.assertEqual(len(rf.records), len(f))
Ejemplo n.º 4
0
    def test_load_from_url(self):
        url = 'http://www.example.com/file'
        f = [1, 2, 3]

        with mock.patch('pyfeddic.record.urllib') as urllib:
            urllib.urlopen.return_value = f

            rf = RecordFactory(url, mock.Mock())
            rf.load_db()

        self.assertEqual(urllib.urlopen.call_count, 1)
        args, _ = urllib.urlopen.call_args
        self.assertEqual(args[0], url)
        self.assertEqual(len(rf.records), len(f))
Ejemplo n.º 5
0
    def test_dyanmic_lookup(self):
        class WithAProperty(object):
            def __init__(self, value):
                self.property = value

        records = [WithAProperty('one'), WithAProperty('two'),
                   WithAProperty('one')]
        rf = RecordFactory(None, None)
        rf.records = records

        filtered = rf.dynamic_record_lookup('property', 'one')
        self.assertEqual(len(filtered), 2)

        filtered = rf.dynamic_record_lookup('nothing', 'one')
        self.assertEqual(len(filtered), 0)
Ejemplo n.º 6
0
    def test_dyanmic_lookup(self):
        class WithAProperty(object):
            def __init__(self, value):
                self.property = value

        records = [
            WithAProperty('one'),
            WithAProperty('two'),
            WithAProperty('one')
        ]
        rf = RecordFactory(None, None)
        rf.records = records

        filtered = rf.dynamic_record_lookup('property', 'one')
        self.assertEqual(len(filtered), 2)

        filtered = rf.dynamic_record_lookup('nothing', 'one')
        self.assertEqual(len(filtered), 0)