def test_reads_web_if_no_file(self, get, mock_open): def raise_error(__): raise IOError() mock_open.side_effect = raise_error get.return_value.text = 'hello\nworld' assert get_vhost_db() == ['hello', 'world']
def test_reads_file_if_exists(self): with mock.patch('builtins.open', mock.mock_open()) as mock_open: lines = ['hello', 'world'] mock_open.return_value.__iter__.return_value = lines assert get_vhost_db() == lines