twhatter/tests/test_cli.py

#!/usr/bin/env python
# coding: utf-8

"""Tests for `twhatter` package."""
import pytest

from twhatter import cli


def test_command_line_interface(cli_runner):
    """Test the CLI."""
    result = cli_runner.invoke(cli.main)
    assert result.exit_code == 0

    help_result = cli_runner.invoke(cli.main, ['--help'])
    assert help_result.exit_code == 0
    assert 'Show this message and exit.' in help_result.output


class TestMain:
    @pytest.mark.send_request
    def test_timeline_no_limit(self, cli_runner, user):
        result = cli_runner.invoke(cli.main, ['timeline', user])
        assert result.exit_code == 0

        lines = result.output.split('\n')[:-1]
        assert 100 > len(lines) > 0

        for l in lines:
            assert "Tweet" in l

    @pytest.mark.send_request
    def test_timeline_limit(self, cli_runner, user, tweet_limit):
        result = cli_runner.invoke(cli.main, ['timeline', user, '--limit', tweet_limit])
        assert result.exit_code == 0

        lines = result.output.split('\n')[:-1]
        assert len(lines) == tweet_limit


class TestDb:
    @pytest.mark.send_request
    def test_timeline_no_limit(self, cli_runner, user):
        result = cli_runner.invoke(cli.main, ['db', 'timeline', user])
        assert result.exit_code == 0