PySqlit/tests/test_main.py at master · Python51888/PySqlit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
"""Unit tests for main.py module."""

import os
import sys
import tempfile
import shutil
from unittest.mock import patch, MagicMock
import pytest

from main import main


class TestMain:
    """Test cases for main module."""

    def test_main_with_database_argument(self):
        """Test main function with database file argument."""
        test_db = "test_database.db"

        with patch('sys.argv', ['main.py', test_db]):
            with patch('pysqlit.repl.EnhancedREPL') as mock_repl:
                mock_instance = MagicMock()
                mock_repl.return_value = mock_instance

                main()

                mock_repl.assert_called_once_with(os.path.abspath(test_db))
                mock_instance.run.assert_called_once()

    def test_main_without_database_argument(self):
        """Test main function without database file argument."""
        with patch('sys.argv', ['main.py']):
            with patch('pysqlit.repl.EnhancedREPL') as mock_repl:
                mock_instance = MagicMock()
                mock_repl.return_value = mock_instance

                main()

                # Should use default "test.db"
                expected_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..', 'test.db')
                mock_repl.assert_called_once_with(os.path.abspath(expected_path))
                mock_instance.run.assert_called_once()

    def test_main_with_relative_path(self):
        """Test main function with relative database path."""
        relative_path = "data/mydb.db"

        with patch('sys.argv', ['main.py', relative_path]):
            with patch('pysqlit.repl.EnhancedREPL') as mock_repl:
                mock_instance = MagicMock()
                mock_repl.return_value = mock_instance

                main()

                expected_path = os.path.abspath(relative_path)
                mock_repl.assert_called_once_with(expected_path)
                mock_instance.run.assert_called_once()

    def test_main_with_absolute_path(self):
        """Test main function with absolute database path."""
        absolute_path = "/tmp/test.db"

        with patch('sys.argv', ['main.py', absolute_path]):
            with patch('pysqlit.repl.EnhancedREPL') as mock_repl:
                mock_instance = MagicMock()
                mock_repl.return_value = mock_instance

                main()

                mock_repl.assert_called_once_with(absolute_path)
                mock_instance.run.assert_called_once()

    def test_main_import_as_module(self):
        """Test that main can be imported without executing."""
        # This test ensures the module can be imported
        import main
        assert hasattr(main, 'main')
        assert callable(main.main)