def test_get_templates_info(open_m, FileSystemLoader_m):
loader = FileSystemLoader_m.return_value
loader.list_templates.return_value = [
'r1/t/template.j2',
'r2/t/template.j2',
'r3/t/template.j2',
]
open_m.return_value = MagicMock(spec=file)
file_handle = open_m.return_value.__enter__.return_value
error = IOError()
error.errno = errno.ENOENT
file_handle.readline.side_effect = ['http://b.c/r', '', error]
manager = TemplateManager('/fake')
templates_info = manager.get_templates_info()
enc = urlsafe_b64encode
expected = [
{'id': enc('r1/t'), 'name': 'r1/t', 'data_source_url': 'http://b.c/r'},
{'id': enc('r2/t'), 'name': 'r2/t', 'data_source_url': None},
{'id': enc('r3/t'), 'name': 'r3/t', 'data_source_url': None},
]
# last call
open_m.assert_called_with('/fake/r3/t/data_source_url', 'rb')
assert_items_equal(expected, templates_info)
def test_list_teams_with_users(admin_client):
team = TeamFactory()
user1 = UserFactory(teams=[team])
user2 = UserFactory(teams=[team])
response = admin_client.get(reverse('team-list'))
assert_200_and_items_equal([get_data(team)], response)
assert_items_equal([user1.id, user2.id], response.data[0]['users'])
def test_list(self):
expected = [
os.path.join('repo1', 't1', 'template.j2'),
os.path.join('repo2', 't1', 'template.j2'),
os.path.join('repo2', 't2', 'template.j2'),
]
loader = FileSystemLoader(self.root_folder)
assert_items_equal(expected, loader.list_templates())
