mirror of
https://github.com/lrnselfreliance/wrolpi.git
synced 2026-10-26 19:01:39 -04:00
120 lines
4.6 KiB
Python
120 lines
4.6 KiB
Python
import json
|
|
from http import HTTPStatus
|
|
|
|
import pytest
|
|
|
|
from modules.videos.models import Channel
|
|
from modules.videos.models import Video
|
|
from wrolpi.downloader import Download, DownloadFrequency
|
|
|
|
|
|
def test_delete_channel_no_url(test_session, test_client, channel_factory):
|
|
"""
|
|
A Channel can be deleted even if it has no URL.
|
|
"""
|
|
channel = channel_factory()
|
|
channel.url = None
|
|
test_session.commit()
|
|
|
|
channel.delete_with_videos()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_delete_channel_with_videos(test_session, async_client, channel_factory, video_factory):
|
|
"""Videos are disowned when their Channel is deleted."""
|
|
channel = channel_factory()
|
|
video = video_factory(channel_id=channel.id)
|
|
video_path, video_id = video.video_path, video.id
|
|
test_session.commit()
|
|
|
|
# Delete the Channel.
|
|
request, response = await async_client.delete(f'/api/videos/channels/{channel.id}')
|
|
assert response.status_code == HTTPStatus.NO_CONTENT
|
|
|
|
# Video still exists, but has no channel.
|
|
video: Video = test_session.query(Video).one()
|
|
assert video.video_path == video_path and video.id == video_id, 'Video entry should not be changed'
|
|
assert video.video_path.is_file(), 'Video should not be deleted'
|
|
assert not video.channel_id and not video.channel, 'Video should not have a Channel'
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_nested_channel_directories(test_session, async_client, test_directory, channel_factory,
|
|
video_factory):
|
|
"""Channel directories cannot contain another Channel's directory."""
|
|
(test_directory / 'foo').mkdir()
|
|
channel1_directory = test_directory / 'foo/one'
|
|
channel_factory(directory=channel1_directory)
|
|
test_session.commit()
|
|
|
|
# Channel 2 cannot be in Channel 1's directory.
|
|
channel2_directory = channel1_directory / 'two'
|
|
channel2_directory.mkdir()
|
|
content = dict(
|
|
name='channel 2',
|
|
directory=str(channel2_directory),
|
|
)
|
|
request, response = await async_client.post('/api/videos/channels', content=json.dumps(content))
|
|
assert response.status_code == HTTPStatus.BAD_REQUEST
|
|
assert response.json['cause'] and \
|
|
response.json['cause']['message'] == 'The directory is already used by another channel.'
|
|
|
|
# Channel 3 cannot be above Channel 1's directory.
|
|
content = dict(
|
|
name='channel 3',
|
|
directory=str(test_directory),
|
|
)
|
|
request, response = await async_client.post('/api/videos/channels', content=json.dumps(content))
|
|
assert response.status_code == HTTPStatus.BAD_REQUEST
|
|
assert response.json['cause'] and \
|
|
response.json['cause']['message'] == 'The directory is already used by another channel.'
|
|
|
|
|
|
@pytest.mark.asyncio()
|
|
async def test_channel_download_relationships(test_session, download_channel):
|
|
"""Test relationships of Channel and Download."""
|
|
test_session.flush()
|
|
|
|
download = test_session.query(Download).one()
|
|
channel = test_session.query(Channel).one()
|
|
assert len(channel.downloads) == 1
|
|
assert channel.downloads
|
|
assert download in channel.downloads
|
|
|
|
test_session.commit()
|
|
|
|
download: Download = test_session.query(Download).one()
|
|
channel: Channel = test_session.query(Channel).one()
|
|
assert len(channel.downloads) == 1
|
|
assert channel.downloads[0] == download and channel.downloads[0].url == 'https://example.com/channel1'
|
|
assert download.frequency == DownloadFrequency.weekly
|
|
assert download.destination == channel.directory
|
|
|
|
# Deleting Download deletes the Download, but not the Channel.
|
|
download.delete()
|
|
assert not test_session.query(Download).all()
|
|
assert test_session.query(Channel).one(), 'Channel should not have been deleted.'
|
|
|
|
# Create a Download again.
|
|
download_channel.get_or_create_download('https://example.com/1', 60, test_session, reset_attempts=True)
|
|
assert test_session.query(Download).count() == 1
|
|
# Delete the Channel, and any relationships.
|
|
download_channel.delete_with_videos()
|
|
assert test_session.query(Download).count() == 0
|
|
|
|
|
|
def test_channel_info_json(test_session, test_directory):
|
|
channel = Channel(name='new channel')
|
|
test_session.add(channel)
|
|
|
|
with pytest.raises(FileNotFoundError) as e:
|
|
# Cannot get info json because directory is not defined.
|
|
channel.info_json_path
|
|
assert 'directory' in str(e)
|
|
|
|
# Channel's directory is created if it does not exist.
|
|
channel.directory = test_directory / 'new channel'
|
|
assert channel.info_json_path == test_directory / 'new channel/new channel.info.json'
|
|
assert channel.directory.is_dir()
|
|
# info json file is not yet created.
|
|
assert not channel.info_json_path.exists()
|