"""
Verification test suite for Production Backend Engine
"""
import sys
import time
import os

# Add backend directory to sys.path
sys.path.insert(0, os.path.dirname(__file__))

import engine

def test_initial_state():
    state = engine.get_initial_state()
    assert state['cash'] == 30.0
    assert 'wheat' in state['products']
    assert state['products']['wheat']['unlocked'] is True
    assert state['products']['wheat']['lines'] == 1
    assert state['products']['flour']['unlocked'] is False
    print("PASS: test_initial_state")

def test_raw_production_and_logistics():
    state = engine.get_initial_state()
    # Fast forward 10 seconds
    state['lastTick'] = time.time() - 10.0
    updated = engine.process_offline_and_ticks(state)
    
    # Wheat has cycle time 1.0s, output 1, 1 line, plus logistics selling
    assert updated['products']['wheat']['totalProduced'] > 0
    assert updated['cash'] > 30.0
    print(f"PASS: test_raw_production_and_logistics (Produced: {updated['products']['wheat']['totalProduced']}, Cash: {updated['cash']})")

def test_chained_dependency_and_bottleneck():
    state = engine.get_initial_state()
    # Turn off wheat lines to simulate 0 upstream production
    state['products']['wheat']['lines'] = 0
    state['products']['wheat']['stock'] = 0.0

    # Unlock flour and give it 1 line
    state['products']['flour']['unlocked'] = True
    state['products']['flour']['lines'] = 1

    # 1. Process with no wheat: flour should bottleneck
    state['lastTick'] = time.time() - 5.0
    state = engine.process_offline_and_ticks(state)
    assert state['products']['flour']['stock'] == 0.0
    assert state['products']['flour']['isBottlenecked'] is True
    
    # 2. Give 10 wheat: flour should consume wheat and produce flour
    state['products']['wheat']['stock'] = 10.0
    state['lastTick'] = time.time() - 4.0
    state = engine.process_offline_and_ticks(state)
    assert state['products']['flour']['totalProduced'] > 0
    assert state['products']['wheat']['stock'] < 10.0 # Consumed!
    print("PASS: test_chained_dependency_and_bottleneck")

def test_ipo_and_shares():
    state = engine.get_initial_state()
    state['lifetimeRevenue'] = 400000.0 # $400k
    earned = engine.calculate_shares_earned(state['lifetimeRevenue'])
    assert earned == 20 # 10 * sqrt(400k / 100k) = 10 * 2 = 20

    # Apply IPO action
    state = engine.apply_actions(state, [{'type': 'TRIGGER_IPO'}])
    assert state['ventureShares'] == 20
    assert state['ipoCount'] == 1
    assert state['products']['wheat']['lines'] == 1
    assert state['cash'] == 30.0
    print("PASS: test_ipo_and_shares")

if __name__ == '__main__':
    test_initial_state()
    test_raw_production_and_logistics()
    test_chained_dependency_and_bottleneck()
    test_ipo_and_shares()
    print("\nALL BACKEND ENGINE TESTS PASSED SUCCESSFULLY!")