import unittest
import tempfile
import yaml

from utils.config import load_yaml_config


class TestLoadYamlConfig(unittest.TestCase):
    def test_load_valid_yaml(self):
        # Create a temporary YAML file
        with tempfile.NamedTemporaryFile(mode='w') as f:
            yaml.dump({'host': 'example.com', 'port': 8080}, f)
            f.flush()
            config = load_yaml_config(f.name)
            self.assertEqual(config, {'host': 'example.com', 'port': 8080})

    def test_load_non_existent_file(self):
        # Test loading a non-existent file
        config = load_yaml_config('non_existent_file.yaml')
        self.assertEqual(config, {'host': '0.0.0.0', 'port': 9999})


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