criclive/tests.py at main · aktech/criclive · 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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
import json
import unittest
from unittest.mock import patch, MagicMock

from criclive.main import get_scores, _format_score, _parse_match


SAMPLE_MATCH = {
    "match": {
        "matchInfo": {
            "matchId": 12345,
            "seriesId": 100,
            "seriesName": "Test Series 2026",
            "matchDesc": "1st Test",
            "matchFormat": "TEST",
            "state": "Complete",
            "stateTitle": "IND won",
            "team1": {
                "teamId": 1,
                "teamName": "India",
                "teamSName": "IND",
            },
            "team2": {
                "teamId": 2,
                "teamName": "Australia",
                "teamSName": "AUS",
            },
        },
        "matchScore": {
            "team1Score": {
                "inngs1": {"inningsId": 1, "runs": 350, "wickets": 10, "overs": 98.2},
                "inngs2": {"inningsId": 3, "runs": 200, "wickets": 5, "overs": 60.0},
            },
            "team2Score": {
                "inngs1": {"inningsId": 2, "runs": 280, "wickets": 10, "overs": 85.4},
            },
        },
    }
}


class TestFormatScore(unittest.TestCase):
    def test_single_innings(self):
        score = {"inngs1": {"runs": 150, "wickets": 5, "overs": 30.2}}
        self.assertEqual(_format_score(score), "150/5 (30.2 ov)")

    def test_two_innings(self):
        score = {
            "inngs1": {"runs": 350, "wickets": 10, "overs": 98.2},
            "inngs2": {"runs": 200, "wickets": 5, "overs": 60.0},
        }
        self.assertEqual(
            _format_score(score), "350/10 (98.2 ov) & 200/5 (60.0 ov)"
        )

    def test_empty_score(self):
        self.assertEqual(_format_score({}), "")


class TestParseMatch(unittest.TestCase):
    def test_parse_valid_match(self):
        result = _parse_match(SAMPLE_MATCH)
        self.assertEqual(result["match_id"], 12345)
        self.assertEqual(result["first_team"]["name"], "IND")
        self.assertEqual(result["second_team"]["name"], "AUS")
        self.assertEqual(result["format"], "TEST")
        self.assertEqual(result["status"], "IND won")
        self.assertIn("350/10", result["first_team"]["score"])
        self.assertIn("280/10", result["second_team"]["score"])

    def test_parse_match_no_teams(self):
        data = {"match": {"matchInfo": {}, "matchScore": {}}}
        self.assertIsNone(_parse_match(data))


class TestGetScores(unittest.TestCase):
    @patch("criclive.main.requests.get")
    def test_get_scores_parses_embedded_json(self, mock_get):
        # Build a minimal HTML page with embedded match data (compact JSON, no spaces)
        match_json = json.dumps(SAMPLE_MATCH, separators=(",", ":")).replace('"', '\\"')
        html = f"""
        <html><head></head><body>
        <script>
        self.__next_f.push([1,"matchesList\\":{{\\"matches\\":[{match_json}]}}"])
        </script>
        </body></html>
        """
        mock_response = MagicMock()
        mock_response.text = html
        mock_response.raise_for_status = MagicMock()
        mock_get.return_value = mock_response

        scores = get_scores()
        self.assertEqual(len(scores), 1)
        self.assertEqual(scores[0]["first_team"]["name"], "IND")

    @patch("criclive.main.requests.get")
    def test_get_scores_empty_page(self, mock_get):
        mock_response = MagicMock()
        mock_response.text = "<html><body></body></html>"
        mock_response.raise_for_status = MagicMock()
        mock_get.return_value = mock_response

        scores = get_scores()
        self.assertEqual(scores, [])


if __name__ == "__main__":
    unittest.main()