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))
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))
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))
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))
