def test_partial_update(admin_client):
team = TeamFactory(name='old name')
payload = {'name': 'new name'}
url = reverse('team-detail', kwargs={'id': team.id})
response = admin_client.patch(url, data=payload)
updated_team = Team.objects.get(id=team.id)
assert_200_and_items_equal(get_data(updated_team), response)
def test_detail_pending_task(result, client):
result.status = states.PENDING
result.ready.return_value = False
expected = {'id': task_id, 'status': states.PENDING}
url = reverse('task-detail', kwargs={'task_id': task_id})
response = client.get(url)
assert_200_and_items_equal(expected, response)
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_preview(tm_mock, client):
template = 'Fake template'
manager = tm_mock.return_value
manager.render_from_string.return_value = template
payload = {'template_str': template}
response = client.post(reverse('template-preview'), data=payload)
expected = {'preview': template}
assert_200_and_items_equal(expected, response)
def test_detail_success_task(result, client):
result.status = states.SUCCESS
result.result = 1
result.ready.return_value = True
result.failed.return_value = False
expected = {'id': task_id, 'status': states.SUCCESS, 'result': 1}
url = reverse('task-detail', kwargs={'task_id': task_id})
response = client.get(url)
assert_200_and_items_equal(expected, response)
def test_list(tm_mock, admin_client):
manager = tm_mock.return_value
fake_info = [
{'name': 'repo1/t', 'data_source_url': 'http://url1.com/t'},
{'name': 'repo2/t', 'data_source_url': None},
]
manager.get_templates_info.return_value = fake_info
response = admin_client.get(reverse('template-list'))
assert_200_and_items_equal(fake_info, response)
def test_detail_failed_task(result, client):
result.status = states.FAILURE
error = ValueError('message')
traceback = 'fake_tb\njahsjhaks'
result.result = error
result.traceback = traceback
result.ready.return_value = True
result.failed.return_value = True
expected = {
'id': task_id,
'status': states.FAILURE,
'result': str(error),
'traceback': traceback,
}
url = reverse('task-detail', kwargs={'task_id': task_id})
response = client.get(url)
assert_200_and_items_equal(expected, response)
def test_preview_fetches_data_source_url(tm_mock, requests_m, client):
template = 'Fake template'
manager = tm_mock.return_value
manager.render_from_string.return_value = template
url = 'http://blah.com/data.json'
fake_json = json.dumps({'blah': 'blah'})
json_mock = Mock()
json_mock.json.return_value = fake_json
requests_m.get.return_value = json_mock
payload = {'template_str': template, 'data_source_url': url}
response = client.post(reverse('template-preview'), data=payload)
expected = {'preview': template}
assert_200_and_items_equal(expected, response)
requests_m.get.assert_called_with(url)
manager.render_from_string.assert_called_with(
template,
data_source=fake_json,
)
def test_detail(admin_client):
repo = RepositoryFactory()
url = reverse('repository-detail', kwargs={'pk': repo.id})
response = admin_client.get(url)
assert_200_and_items_equal(get_data(repo), response)
def test_detail(admin_client):
team = TeamFactory()
UserFactory(teams=[team])
url = reverse('team-detail', kwargs={'id': team.id})
response = admin_client.get(url)
assert_200_and_items_equal(get_data(team), response)
def test_list(admin_client):
team = TeamFactory()
response = admin_client.get(reverse('team-list'))
assert_200_and_items_equal([get_data(team)], response)
