test_main.py

import pytest
import sqlite3
from main import Database


@pytest.fixture
def database():
    
    db = Database()
    
    # In-memory database for testing 
    db.conn = sqlite3.connect(":memory:")
    db.conn.row_factory = sqlite3.Row  
    db.c = db.conn.cursor()
    
    # Create the table
    db.table()
    
    # Return the database instance so it can be tested
    yield db
    
    # Close the databse
    db.close()


# Test if adding a question to the database works 
def test_add(database):

    # Add a question to the database
    database.add("test", "What is the capital of Hungary?", "Budapest", "London", "Berlin", "Paris")
    
    # Check if the question actually exists/ get it back from the database
    database.c.execute('SELECT * FROM quiz WHERE question = ?', ("What is the capital of Hungary?",))
    result = database.c.fetchone()

    # Check for values 
    assert result["question"] == "What is the capital of Hungary?"
    assert result["answer"] == "Budapest"
    assert result["decoy_1"] == "London"
    assert result["decoy_2"] == "Berlin"
    assert result["decoy_3"] == "Paris"


# Test if deleting a question works
def test_delete(database):

    # Add a question
    database.add("test", "What is the capital of Hungary?", "Budapest", "London", "Berlin", "Paris")
    
    # Now delete the question
    database.delete("What is the capital of Hungary?")
    
    # Check if the question is still in the database
    database.c.execute('SELECT * FROM quiz WHERE question = ?', ("What is the capital of Hungary?",))
    result = database.c.fetchone()
    
    assert result is None # It should be None if it doen't exists


# Test if I can get all the questions of a certain type
def test_type_questions(database):

    # Add sample questions 
    database.add("test", "What is the capital of Hungary?", "Budapest", "London", "Berlin", "Paris")
    database.add("animal", "What is unique about chinchilla fur?", "density", "waterproof", "oily", "sticky")
    
    # Get questions of type 'test'
    questions = database.type_questions("test")

    # Test it
    assert len(questions) == 1 # There should be just one question for this type
    assert questions[0]["question"] == "What is the capital of Hungary?"
    
    # Get questions of type 'animal'
    questions = database.type_questions("animal")

    # Test it
    assert len(questions) == 1
    assert questions[0]["question"] == "What is unique about chinchilla fur?"


# Test if we can get all available types from the database
def test_types(database):

    # Add sample questions 
    database.add("test", "What is the capital of Hungary?", "Budapest", "London", "Berlin", "Paris")
    database.add("animal", "What is unique about chinchilla fur?", "density", "waterproof", "oily", "sticky")
    
    # Get types
    types = database.types()

    # Check if these types exist
    assert "test" in types
    assert "animal" in types


# Test the question count for a certain type of questions
def test_type_questions_count(database):

    
    # Add sample questions 
    database.add("test", "What is the capital of Hungary?", "Budapest", "London", "Berlin", "Paris")
    database.add("animal", "What is unique about chinchilla fur?", "density", "waterproof", "oily", "sticky")
    database.add("test", "What is the capital of Romania?", "Bucurest", "London", "Berlin", "Paris")
    
    # Get questions of type 'test'
    count = database.type_questions_count("test")
    assert count == 2
    
    # Get questions of type 'animal'
    count = database.type_questions_count("animal")
    assert count == 1