reddit/test_reddit_cache.py at main · ripred/reddit · 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
#!/usr/bin/env python3
import os
import tempfile
import shutil
import unittest

from reddit_cache import (
    remove_fenced_code,
    remove_indented_code,
    clean_text,
    is_code_line,
    has_unformatted_code,
    get_cache_folder
)

class TestRedditCache(unittest.TestCase):
    def test_remove_fenced_code(self):
        text = (
            "This is a test.\n"
            "```python\n"
            "#include <stdio.h>\n"
            "void main() {}\n"
            "```\n"
            "End of test."
        )
        expected = "This is a test.\nEnd of test."
        result = remove_fenced_code(text)
        self.assertEqual(result.strip(), expected.strip())

    def test_remove_indented_code(self):
        text = (
            "This is a test.\n"
            "    #include <stdio.h>\n"
            "    void main() {}\n"
            "End of test."
        )
        expected = "This is a test.\nEnd of test."
        result = remove_indented_code(text)
        self.assertEqual(result.strip(), expected.strip())

    def test_clean_text_removes_both(self):
        text = (
            "Intro text\n"
            "    #include <stdio.h>\n"
            "    void main() {}\n"
            "More text\n"
            "```\n"
            "def foo():\n"
            "    pass\n"
            "```\n"
            "Conclusion"
        )
        expected = "Intro text\nMore text\nConclusion"
        result = clean_text(text)
        self.assertEqual(result.strip(), expected.strip())

    def test_is_code_line_positive(self):
        self.assertTrue(is_code_line("#include <stdio.h>"))
        self.assertTrue(is_code_line("void main() {"))
        self.assertTrue(is_code_line("for (int i = 0; i < 10; i++) {"))
        self.assertTrue(is_code_line("printf(\"Hello\");"))

    def test_is_code_line_negative(self):
        self.assertFalse(is_code_line("This is just a regular sentence."))

    def test_has_unformatted_code_does_not_flag_properly_formatted(self):
        text = (
            "#include <stdio.h>\n"
            "void main() {\n"
            "    printf(\"Hello\");\n"
            "}\n"
            "Some extra text."
        )
        self.assertFalse(has_unformatted_code(text))

    def test_has_unformatted_code_flags_unformatted_code(self):
        text = (
            "This is a test post that contains unformatted source code.\n"
            "This is a test post that contains improperly formatted source code.\n"
            "include <Arduino.h>\n"
            "void setup() { Serial.begin(115200); Serial.println(F(\"This is a test post\\n\")); }\n"
            "void loop() { }\n"
            "I sure hope that\n"
            "these automated tests\n"
            "can be useful\n"
            "once they are\n"
            "debugged"
        )
        self.assertTrue(has_unformatted_code(text))

    def test_get_cache_folder_creates_directory(self):
        temp_dir = tempfile.mkdtemp()
        original_dir = os.getcwd()
        os.chdir(temp_dir)
        try:
            folder = get_cache_folder("testsub")
            self.assertTrue(os.path.isdir(folder))
        finally:
            os.chdir(original_dir)
            shutil.rmtree(temp_dir)

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