view dictation/config_test.py @ 279:b3b547563ec7

Add Google connector service and agent wiki Implement the C/Seobeo Google Drive and Gmail connector with encrypted OAuth storage, Zenbu authentication, browser testing, AI tool discovery, chunked HTTP decoding, and Bazel coverage. Consolidate repository guidance into progressive wiki documentation and enforce arena-first allocation for new first-party C code. Co-authored-by: Copilot <[email protected]> Copilot-Session: 84c338fd-0939-4bb3-b7f3-1062eb213e5d
author MrJuneJune <me@mrjunejune.com>
date Mon, 17 Aug 2026 22:22:36 -0700
parents 78699f810817
children 49e9e591c9bb
line wrap: on
line source

import os
from pathlib import Path
import unittest
from unittest.mock import patch

from dictation.config import DictationConfig


class DictationConfigTest(unittest.TestCase):
    def test_defaults_are_local_and_bounded(self):
        with patch.dict(os.environ, {}, clear=True):
            config = DictationConfig.from_environment()
        self.assertEqual(config.host, "127.0.0.1")
        self.assertEqual(config.port, 8090)
        self.assertEqual(config.max_sessions, 1)
        self.assertEqual(config.compute_type, "int8_float16")
        self.assertTrue(str(config.model_dir).endswith("faster-whisper-small"))

    def test_rejects_invalid_session_limit(self):
        with patch.dict(
            os.environ,
            {"DICTATION_MAX_SESSIONS": "0"},
            clear=True,
        ):
            with self.assertRaisesRegex(ValueError, "between 1 and 8"):
                DictationConfig.from_environment()

    def test_accepts_model_override(self):
        with patch.dict(
            os.environ,
            {"DICTATION_MODEL_DIR": "/tmp/dictation-model"},
            clear=True,
        ):
            config = DictationConfig.from_environment()
        self.assertEqual(config.model_dir, Path("/tmp/dictation-model"))


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