import unittest from aider_gitea.seen_issues_db import SeenIssuesDB class TestSeenIssuesDB(unittest.TestCase): def setUp(self): self.db = SeenIssuesDB(':memory:') def test_mark_and_check_seen_issue(self): issue_text = "Test issue" self.assertFalse(self.db.has_seen(issue_text)) self.db.mark_as_seen(issue_text) self.assertTrue(self.db.has_seen(issue_text)) def test_unseen_issue(self): issue_text = "Unseen issue" self.assertFalse(self.db.has_seen(issue_text)) if __name__ == '__main__': unittest.main()