test_app.py

import pytest
import json
from app import app, active_games


@pytest.fixture
def client():
    """Create a test client for the Flask app"""
    app.config['TESTING'] = True
    with app.test_client() as client:
        yield client
    active_games.clear()


@pytest.fixture
def sample_game(client):
    """Create a sample game and return its data"""
    response = client.get('/api/grid')
    data = json.loads(response.data)
    return data


class TestGetGrid:
    """Tests for GET /api/grid endpoint"""

    def test_get_grid_success(self, client):
        """Test that getting a new grid returns valid data"""
        response = client.get('/api/grid')
        data = json.loads(response.data)

        assert response.status_code == 200
        assert 'game_id' in data
        assert 'grid' in data
        assert 'difficulty' in data
        assert len(data['grid']) == 9
        assert all(len(row) == 9 for row in data['grid'])


class TestValidateMove:
    """Tests for POST /api/validate endpoint"""

    def test_validate_correct_move(self, client, sample_game):
        """Test validating a correct move"""
        game_id = sample_game['game_id']
        grid = sample_game['grid']
        game = active_games[game_id]

        # Find first empty cell
        row, col = None, None
        for r in range(9):
            for c in range(9):
                if grid[r][c] == 0:
                    row, col = r, c
                    break
            if row is not None:
                break

        if row is not None:
            correct_value = game['solution'][row][col]
            response = client.post('/api/validate',
                json={'game_id': game_id, 'row': row, 'col': col, 'value': correct_value})
            data = json.loads(response.data)

            assert response.status_code == 200
            assert data['valid'] is True
            assert data['is_correct'] is True

    def test_validate_missing_parameters(self, client):
        """Test that missing parameters return 400"""
        response = client.post('/api/validate', json={'game_id': 'test', 'row': 0})
        assert response.status_code == 400

    def test_validate_game_not_found(self, client):
        """Test that invalid game_id returns 404"""
        response = client.post('/api/validate',
            json={'game_id': 'nonexistent', 'row': 0, 'col': 0, 'value': 5})
        assert response.status_code == 404


class TestGetSolution:
    """Tests for GET /api/solution/<game_id> endpoint"""

    def test_get_solution_success(self, client, sample_game):
        """Test getting solution for an existing game"""
        game_id = sample_game['game_id']
        response = client.get(f'/api/solution/{game_id}')
        data = json.loads(response.data)

        assert response.status_code == 200
        assert 'solution' in data
        assert len(data['solution']) == 9