ইনপুট ভ্যালিডেশন ও ইনজেকশন টেস্টিং

import re
import unittest

def is_safe_input(value):
    """একটি সরল ভ্যালিডেশন ফাংশন — সন্দেহজনক প্যাটার্নযুক্ত ইনপুট প্রত্যাখ্যান করে।"""
    if not isinstance(value, str):
        return False

    lowered = value.lower()

    # কোট + কমেন্ট মার্কার/সেপারেটর একসাথে থাকলে সন্দেহজনক
    if "'" in value and ("--" in value or ";" in value):
        return False

    # ক্লাসিক "OR 1=1"-স্টাইল প্যাটার্ন
    if re.search(r"or\s+1\s*=\s*1", lowered):
        return False

    # স্ক্রিপ্ট-ট্যাগ-স্টাইল প্যাটার্ন
    if "<script" in lowered:
        return False

    return True

class InputValidationTests(unittest.TestCase):

    def test_accepts_plain_email(self):
        self.assertTrue(is_safe_input("john.doe@example.com"))

    def test_accepts_apostrophe_in_name(self):
        # বৈধ ব্যবহারকারী — শুধু কোট থাকলেই প্রত্যাখ্যান করা উচিত নয়
        self.assertTrue(is_safe_input("O'Brien"))

    def test_accepts_semicolon_in_prose(self):
        # শুধু সেমিকোলন থাকলেই (কোট ছাড়া) প্রত্যাখ্যান করা উচিত নয়
        self.assertTrue(is_safe_input("Please review this; it looks correct"))

    def test_rejects_sql_comment_injection(self):
        self.assertFalse(is_safe_input("admin'--"))

    def test_rejects_sql_drop_table(self):
        self.assertFalse(is_safe_input("'; DROP TABLE users;--"))

    def test_rejects_or_1_equals_1(self):
        self.assertFalse(is_safe_input("' OR 1=1 --"))

    def test_rejects_script_tag(self):
        self.assertFalse(is_safe_input("<script>alert(1)</script>"))

suite = unittest.TestLoader().loadTestsFromTestCase(InputValidationTests)
unittest.TextTestRunner(verbosity=2).run(suite)