New Version 2 -- prolly doesn't work.

tests/test_stegasoo.py

@@ -0,0 +1,220 @@
+"""
+Basic tests for Stegasoo library.
+"""
+
+import io
+import sys
+from pathlib import Path
+
+import pytest
+
+# Add src to path for development
+sys.path.insert(0, str(Path(__file__).parent.parent / 'src'))
+
+import stegasoo
+from stegasoo import (
+    generate_credentials,
+    generate_pin,
+    generate_phrase,
+    validate_pin,
+    validate_message,
+    encode,
+    decode,
+    DAY_NAMES,
+)
+
+
+class TestKeygen:
+    """Test credential generation."""
+    
+    def test_generate_pin_default(self):
+        pin = generate_pin()
+        assert len(pin) == 6
+        assert pin.isdigit()
+        assert pin[0] != '0'
+    
+    def test_generate_pin_lengths(self):
+        for length in range(6, 10):
+            pin = generate_pin(length)
+            assert len(pin) == length
+    
+    def test_generate_phrase_default(self):
+        phrase = generate_phrase()
+        words = phrase.split()
+        assert len(words) == 3
+    
+    def test_generate_phrase_lengths(self):
+        for length in range(3, 13):
+            phrase = generate_phrase(length)
+            words = phrase.split()
+            assert len(words) == length
+    
+    def test_generate_credentials_pin_only(self):
+        creds = generate_credentials(use_pin=True, use_rsa=False)
+        assert creds.pin is not None
+        assert creds.rsa_key_pem is None
+        assert len(creds.phrases) == 7
+        assert set(creds.phrases.keys()) == set(DAY_NAMES)
+    
+    def test_generate_credentials_rsa_only(self):
+        creds = generate_credentials(use_pin=False, use_rsa=True)
+        assert creds.pin is None
+        assert creds.rsa_key_pem is not None
+        assert '-----BEGIN PRIVATE KEY-----' in creds.rsa_key_pem
+    
+    def test_generate_credentials_both(self):
+        creds = generate_credentials(use_pin=True, use_rsa=True)
+        assert creds.pin is not None
+        assert creds.rsa_key_pem is not None
+    
+    def test_generate_credentials_neither_fails(self):
+        with pytest.raises(ValueError):
+            generate_credentials(use_pin=False, use_rsa=False)
+    
+    def test_entropy_calculation(self):
+        creds = generate_credentials(
+            use_pin=True,
+            use_rsa=True,
+            pin_length=6,
+            rsa_bits=2048,
+            words_per_phrase=3
+        )
+        assert creds.phrase_entropy == 33  # 3 * 11
+        assert creds.pin_entropy == 19  # floor(6 * 3.32)
+        assert creds.rsa_entropy == 128
+        assert creds.total_entropy == 33 + 19 + 128
+
+
+class TestValidation:
+    """Test input validation."""
+    
+    def test_validate_pin_valid(self):
+        result = validate_pin("123456")
+        assert result.is_valid
+    
+    def test_validate_pin_empty_ok(self):
+        result = validate_pin("")
+        assert result.is_valid
+    
+    def test_validate_pin_too_short(self):
+        result = validate_pin("12345")
+        assert not result.is_valid
+        assert "6-9" in result.error_message
+    
+    def test_validate_pin_too_long(self):
+        result = validate_pin("1234567890")
+        assert not result.is_valid
+    
+    def test_validate_pin_leading_zero(self):
+        result = validate_pin("012345")
+        assert not result.is_valid
+        assert "zero" in result.error_message.lower()
+    
+    def test_validate_pin_non_digits(self):
+        result = validate_pin("12345a")
+        assert not result.is_valid
+    
+    def test_validate_message_valid(self):
+        result = validate_message("Hello, world!")
+        assert result.is_valid
+    
+    def test_validate_message_empty(self):
+        result = validate_message("")
+        assert not result.is_valid
+    
+    def test_validate_message_too_long(self):
+        result = validate_message("x" * 60000)
+        assert not result.is_valid
+
+
+class TestEncodeDecode:
+    """Test encoding and decoding (requires test images)."""
+    
+    @pytest.fixture
+    def test_image(self):
+        """Create a simple test image."""
+        from PIL import Image
+        img = Image.new('RGB', (100, 100), color='red')
+        buf = io.BytesIO()
+        img.save(buf, format='PNG')
+        return buf.getvalue()
+    
+    def test_encode_decode_roundtrip(self, test_image):
+        """Test full encode/decode cycle."""
+        message = "Secret message!"
+        phrase = "apple forest thunder"
+        pin = "123456"
+        
+        result = encode(
+            message=message,
+            reference_photo=test_image,
+            carrier_image=test_image,
+            day_phrase=phrase,
+            pin=pin
+        )
+        
+        assert result.stego_image is not None
+        assert len(result.stego_image) > 0
+        assert result.filename.endswith('.png')
+        
+        decoded = decode(
+            stego_image=result.stego_image,
+            reference_photo=test_image,
+            day_phrase=phrase,
+            pin=pin
+        )
+        
+        assert decoded == message
+    
+    def test_wrong_pin_fails(self, test_image):
+        """Test that wrong PIN fails to decode."""
+        result = encode(
+            message="Secret",
+            reference_photo=test_image,
+            carrier_image=test_image,
+            day_phrase="test phrase here",
+            pin="123456"
+        )
+        
+        with pytest.raises(stegasoo.DecryptionError):
+            decode(
+                stego_image=result.stego_image,
+                reference_photo=test_image,
+                day_phrase="test phrase here",
+                pin="654321"  # Wrong PIN
+            )
+    
+    def test_wrong_phrase_fails(self, test_image):
+        """Test that wrong phrase fails to decode."""
+        result = encode(
+            message="Secret",
+            reference_photo=test_image,
+            carrier_image=test_image,
+            day_phrase="correct phrase here",
+            pin="123456"
+        )
+        
+        with pytest.raises(stegasoo.DecryptionError):
+            decode(
+                stego_image=result.stego_image,
+                reference_photo=test_image,
+                day_phrase="wrong phrase here",
+                pin="123456"
+            )
+
+
+class TestVersion:
+    """Test version information."""
+    
+    def test_version_exists(self):
+        assert hasattr(stegasoo, '__version__')
+        assert stegasoo.__version__ == "2.0.0"
+    
+    def test_day_names(self):
+        assert len(stegasoo.DAY_NAMES) == 7
+        assert stegasoo.DAY_NAMES[0] == 'Monday'
+        assert stegasoo.DAY_NAMES[6] == 'Sunday'
+
+
+if __name__ == '__main__':
+    pytest.main([__file__, '-v'])