import unittest
from click.testing import CliRunner
from pypiup.cli import cli
from tests.mocks import TestMockedRequests, TestMockedDemoRequests


class TestRequirementsMockedRequests(TestMockedRequests):

    def setUp(self):
        super(TestRequirementsMockedRequests, self).setUp()

    def test_cli(self):
        runner = CliRunner()
        result = runner.invoke(cli, ['-r', 'requirements/requirements-tests.txt'])

        self.assertEqual(result.exit_code, 0)

        # Django
        self.assertIn("Could not compare. Invalid semver, From 1.7.1 to 1.9.", result.output)

        # djangorestframework
        self.assertIn("Up to date, 3.3.2.", result.output)

        # drfdocs
        self.assertIn("Needs update, From 0.0.4 to 0.0.5.", result.output)

        self.assertIn("Total Requirements: 5", result.output)
        self.assertIn("Up to date: 1", result.output)
        self.assertIn("Update available: 1", result.output)
        self.assertIn("Invalid Semver: 1", result.output)
        self.assertIn("Non PyPI Requirements: 2", result.output)


class TestRequirementsDemoMockedRequests(TestMockedDemoRequests):

    def setUp(self):
        super(TestRequirementsDemoMockedRequests, self).setUp()

    def test_cli(self):
        runner = CliRunner()
        result = runner.invoke(cli, ['--demo'])

        self.assertEqual(result.exit_code, 0)

        # Django
        self.assertIn("Could not compare. Invalid semver, From 1.7.1 to 1.9.", result.output)

        # click
        self.assertIn("Could not compare. Invalid semver, From 6.1 to 6.2.", result.output)

        # djangorestframework
        self.assertIn("Up to date, 3.3.2.", result.output)

        # django-crispy-forms
        self.assertIn("Needs update, From 1.4.0 to 1.5.0.", result.output)

        # drfdocs
        self.assertIn("Up to date, 0.0.5.", result.output)

        # apscheduler
        self.assertIn("Up to date, 3.0.3.", result.output)

        self.assertIn("Total Requirements: 8", result.output)
        self.assertIn("Up to date: 3", result.output)
        self.assertIn("Update available: 1", result.output)
        self.assertIn("Invalid Semver: 2", result.output)
        self.assertIn("Non PyPI Requirements: 2", result.output)


class TestRequirementsFileNotFound(unittest.TestCase):

    def test_cli_file_not_found(self):
        runner = CliRunner()
        result = runner.invoke(cli, ['-r', 'requirements-fake.txt'])

        self.assertEqual(result.exit_code, 0)
        self.assertIn("Could not find requirements-fake.txt. No such file or directory.", result.output)

# if __name__ == '__main__':
#     unittest.main()

@@ -2592,56 +2592,4 @@
 ut)%0A
-%0A# if __name__ == '__main__':%0A#     unittest.main()%0A

Clean up