test_validators.py

"""
Basic sanity unit tests for validators using unittest framework.
"""
import unittest
from datetime import datetime
from app.utils.logger import setup_logger
from app.utils.validators import SensorRecord, CSVRowValidator
from pydantic import ValidationError

logger = setup_logger(__name__)


class TestSensorRecord(unittest.TestCase):
    """Test SensorRecord validation."""
    
    def test_valid_sensor_record(self):
        """Test creating a valid sensor record."""
        logger.info(f"[{self._testMethodName}] Testing valid sensor record creation")
        record = SensorRecord(
            sensor_id="sensor_001",
            timestamp="2024-01-01T12:00:00",
            value=25.5
        )
        self.assertEqual(record.sensor_id, "sensor_001")
        self.assertIsInstance(record.timestamp, datetime)
        self.assertEqual(record.value, 25.5)
        logger.info(f"[{self._testMethodName}] Valid sensor record test passed")
    
    def test_invalid_empty_sensor_id(self):
        """Test that empty sensor_id raises validation error."""
        logger.info(f"[{self._testMethodName}] Testing empty sensor_id validation")
        with self.assertRaises(ValidationError):
            SensorRecord(
                sensor_id="",
                timestamp="2024-01-01T12:00:00",
                value=25.5
            )
        logger.info(f"[{self._testMethodName}] Empty sensor_id validation test passed")
    
    def test_invalid_timestamp(self):
        """Test that invalid timestamp raises validation error."""
        logger.info(f"[{self._testMethodName}] Testing invalid timestamp validation")
        with self.assertRaises(ValidationError):
            SensorRecord(
                sensor_id="sensor_001",
                timestamp="invalid-date",
                value=25.5
            )
        logger.info(f"[{self._testMethodName}] Invalid timestamp validation test passed")


class TestCSVRowValidator(unittest.TestCase):
    """Test CSVRowValidator functionality."""
    
    def test_validate_valid_row(self):
        """Test validating a valid CSV row."""
        logger.info(f"[{self._testMethodName}] Testing valid CSV row validation")
        row = {
            'sensor_id': 'sensor_001',
            'timestamp': '2024-01-01T12:00:00',
            'value': '25.5'
        }
        is_valid, error_msg = CSVRowValidator.validate_csv_row(row, 1)
        self.assertTrue(is_valid)
        self.assertIsNone(error_msg)
        logger.info(f"[{self._testMethodName}] Valid CSV row validation test passed")
    
    def test_validate_invalid_row(self):
        """Test validating an invalid CSV row."""
        logger.info(f"[{self._testMethodName}] Testing invalid CSV row validation")
        row = {
            'sensor_id': '',
            'timestamp': '2024-01-01T12:00:00',
            'value': '25.5'
        }
        is_valid, error_msg = CSVRowValidator.validate_csv_row(row, 1)
        self.assertFalse(is_valid)
        self.assertIsNotNone(error_msg)
        logger.info(f"[{self._testMethodName}] Invalid CSV row validation test passed")


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