Initial commit

tests/test_validation.py

@@ -0,0 +1,177 @@
+#!/usr/bin/env python3
+"""
+Tests for validators.
+"""
+
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).parent.parent / 'scripts'))
+
+from utils.validators import *
+
+
+def test_validate_host():
+    """Test host validation."""
+    # Valid host
+    assert validate_host("web-01") == "web-01"
+    assert validate_host(" web-01 ") == "web-01"  # Strips whitespace
+
+    # With valid list
+    assert validate_host("web-01", ["web-01", "web-02"]) == "web-01"
+
+    # Invalid format
+    try:
+        validate_host("web@01")  # Invalid character
+        assert False, "Should have raised ValidationError"
+    except ValidationError:
+        pass
+
+    print("✓ validate_host() passed")
+    return True
+
+
+def test_validate_group():
+    """Test group validation."""
+    # Valid group
+    assert validate_group("production") == "production"
+    assert validate_group("PRODUCTION") == "production"  # Lowercase normalization
+
+    # With valid list
+    assert validate_group("production", ["production", "staging"]) == "production"
+
+    # Invalid
+    try:
+        validate_group("invalid!", ["production"])
+        assert False, "Should have raised ValidationError"
+    except ValidationError:
+        pass
+
+    print("✓ validate_group() passed")
+    return True
+
+
+def test_validate_path_exists():
+    """Test path existence validation."""
+    # Valid path
+    path = validate_path_exists("/tmp", must_be_dir=True)
+    assert isinstance(path, Path)
+
+    # Invalid path
+    try:
+        validate_path_exists("/nonexistent_12345")
+        assert False, "Should have raised ValidationError"
+    except ValidationError:
+        pass
+
+    print("✓ validate_path_exists() passed")
+    return True
+
+
+def test_validate_timeout():
+    """Test timeout validation."""
+    # Valid timeouts
+    assert validate_timeout(10) == 10
+    assert validate_timeout(1) == 1
+    assert validate_timeout(600) == 600
+
+    # Too low
+    try:
+        validate_timeout(0)
+        assert False, "Should have raised ValidationError"
+    except ValidationError:
+        pass
+
+    # Too high
+    try:
+        validate_timeout(1000)
+        assert False, "Should have raised ValidationError"
+    except ValidationError:
+        pass
+
+    print("✓ validate_timeout() passed")
+    return True
+
+
+def test_validate_command():
+    """Test command validation."""
+    # Safe commands
+    assert validate_command("ls -la") == "ls -la"
+    assert validate_command("uptime") == "uptime"
+
+    # Dangerous commands (should fail without allow_dangerous)
+    try:
+        validate_command("rm -rf /")
+        assert False, "Should have blocked dangerous command"
+    except ValidationError:
+        pass
+
+    # But should work with allow_dangerous
+    assert validate_command("rm -rf /tmp/test", allow_dangerous=True)
+
+    print("✓ validate_command() passed")
+    return True
+
+
+def test_validate_hosts_list():
+    """Test list validation."""
+    # Valid list
+    hosts = validate_hosts_list(["web-01", "web-02"])
+    assert len(hosts) == 2
+    assert "web-01" in hosts
+
+    # Empty list
+    try:
+        validate_hosts_list([])
+        assert False, "Should have raised ValidationError for empty list"
+    except ValidationError:
+        pass
+
+    print("✓ validate_hosts_list() passed")
+    return True
+
+
+def test_get_invalid_hosts():
+    """Test finding invalid hosts."""
+    # Test with mix of valid and invalid
+    # (This would require actual SSH config, so we test the function exists)
+    result = get_invalid_hosts(["web-01", "nonexistent-host-12345"])
+    assert isinstance(result, list)
+
+    print("✓ get_invalid_hosts() passed")
+    return True
+
+
+def main():
+    """Run all validation tests."""
+    print("=" * 70)
+    print("VALIDATION TESTS")
+    print("=" * 70)
+
+    tests = [
+        test_validate_host,
+        test_validate_group,
+        test_validate_path_exists,
+        test_validate_timeout,
+        test_validate_command,
+        test_validate_hosts_list,
+        test_get_invalid_hosts,
+    ]
+
+    passed = 0
+    for test in tests:
+        try:
+            if test():
+                passed += 1
+        except Exception as e:
+            print(f"✗ {test.__name__} failed: {e}")
+            import traceback
+            traceback.print_exc()
+
+    print(f"\nResults: {passed}/{len(tests)} passed")
+    return passed == len(tests)
+
+
+if __name__ == "__main__":
+    success = main()
+    sys.exit(0 if success else 1)