#!/usr/bin/env python """ Unit tests for Relationship Manager (relationships/relationship_manager.py) Tests affinity updates, relationship handling, romance, dating activities, relationship types, and edge cases. Run with: pytest tests/unit/test_relationship_manager.py -v """ import pytest import sys import os from pathlib import Path from unittest.mock import patch, MagicMock, Mock import random import uuid from types import SimpleNamespace # Add ws directory to Python path for imports ws_dir = Path(__file__).parent.parent.parent / 'ws' sys.path.insert(0, str(ws_dir)) # Set test mode to avoid loading unnecessary dependencies os.environ['TEST_MODE'] = 'true' # Mock the problematic circular import modules BEFORE any code imports them sys.modules['events'] = MagicMock() sys.modules['events'].messageFunction = MagicMock(return_value=MagicMock()) sys.modules['messaging_style'] = MagicMock() sys.modules['messaging_style'].apply_event_modifiers = MagicMock() # ============================================================================ # HELPER FUNCTIONS TO CREATE TEST OBJECTS # ============================================================================ def create_test_person(person_id=None, firstname="Test", lastname="Person", affinity=50): """Create a test person object without database dependencies.""" person = SimpleNamespace() person.id = person_id or uuid.uuid4().hex person.type = "personObject" person.relationships = [] person.firstname = firstname person.lastname = lastname person.sex = "Male" person.pronoun = "He" person.message = False person.ageHours = 0 person.ageDays = 0 person.ageYears = 0 person.image = "test_image.png" person.mood = "Calm" person.money = 1000 person.weight = 55 person.weightType = "Normal" person.hunger = 0 person.thirst = 0 person.energy = 100 person.calcEnergy = 100 person.peakEnergy = 0 person.prestige = 0 person.diamonds = 35 person.stress = 0 person.social = 0 person.happiness = 50 person.location = 'home' person.health = 1 person.deathChance = 0 person.familyLevel = 0 person.spendingHabits = "normal" person.activityRecords = [] person.education = 'None' person.current_education = None person.elementary_school = None person.high_school = None person.college = None person.actScore = 0 person.occupation = 'preschool' person.job = None person.major = None person.minor = None person.activities = [] person.habits = [] person.healthConditions = [] person.dislikes = [] person.likes = [] person.items = [] person.canDrive = False person.affinity = affinity person.familiarity = 0 person.status = "alive" person.dailyPlan = [] person.schedules = [] person.oneTimeEvents = [] person.intraDayMessage = False person.tryingForChild = False person.partner = False person.relationship = None person.avatar_settings = SimpleNamespace( clothing='hoodie', skin_color='Light', hair='short', hair_color='AUBURN', facial_hair='none', accessory='none', mouth='default' ) return person def create_test_relationship(person1, person2, status="Dating", score=50): """Create a test relationship object.""" rel = SimpleNamespace() rel.id = uuid.uuid4().hex rel.person1 = person1 rel.person2 = person2 rel.startDate = "01-01" rel.anniversaryDate = "01-01" rel.relationshipStatus = status rel.relationshipNotes = f"Test {status} relationship" rel.eventsLog = [] rel.relationshipScore = score rel.commonInterests = [] rel.challenges = [] rel.futurePlans = [] rel.messaging_modifiers = { 'verbosity': 0, 'inquisitiveness': 0, } return rel def create_test_player(): """Create a test player object without database dependencies.""" player = SimpleNamespace() player.controller = 'inactive' player.connection = 'connected' player.updateClient = False player.status = 'creating' player.dayEvent = '' player.deviceToken = '' player.events = set() player.askedQuestions = set() player.conversations = [] player.activeDilemmas = [] player.messageQueue = [] player.messageLog = [] player.offlineStats = SimpleNamespace(minutesOffline=0) player.gameSpeed = 100 player.previousGameSpeed = 100 player.messageEnergyCost = 5 player.ticks = 0 player.fps = 0 player.dayOfYear = 15 player.monthOfYear = 1 player.season = "Winter" player.date = "01-15" player.minuteOfHour = 0 player.hourOfDay = 0 player.dayOfWeek = 0 player.weekDay = 0 player.weekend = False player.weekDayText = "Monday" player.daysSinceSchoolStarted = 0 player.daysUntilSchoolEnds = 0 player.summerVacation = False player.moods = ["Calm", "Stressed", "Exhausted", "Fulfilled", "Depressed", "Happy"] player.type = "playerObject" # Create main character player.c = create_test_person(person_id="player_001", firstname="TestPlayer", lastname="User") player.r = [] # characters / relationships player.l = [] # locations player.relData = [] return player # Import relationship manager functions after mocking from relationships.relationship_manager import ( updateAffinity, handleRelationships, getRelData, getActiveRelationship, breakUp, partnerGift, dateNight, getDateIdeas, DateIdea ) # Import with mocking for romance function import relationships.relationship_manager as rm # ============================================================================ # FIXTURES # ============================================================================ @pytest.fixture def player_with_relationships(): """Create a player with multiple NPCs and relationships.""" player = create_test_player() # Create multiple NPCs with different affinities friend = create_test_person("npc_friend", "Alice", "Friend", affinity=70) acquaintance = create_test_person("npc_acquaintance", "Bob", "Acquaintance", affinity=30) romantic_interest = create_test_person("npc_romance", "Charlie", "Romance", affinity=60) family_member = create_test_person("npc_family", "Diana", "User", affinity=90) player.r = [friend, acquaintance, romantic_interest, family_member] return player @pytest.fixture def player_with_active_relationship(): """Create a player with an active dating relationship.""" player = create_test_player() # Create partner NPC partner = create_test_person("npc_partner", "Sam", "Partner", affinity=80) player.r = [partner] # Create active relationship relationship = create_test_relationship( player.c.id, partner.id, status="Dating", score=65 ) player.relData.append(relationship) player.c.relationship = relationship.id partner.relationship = relationship.id return player @pytest.fixture def simple_player(): """Create a simple player without relationships.""" return create_test_player() # ============================================================================ # AFFINITY UPDATE TESTS (4 tests) # ============================================================================ def test_update_affinity_increases(player_with_relationships): """Test that positive interaction increases affinity.""" # Arrange player = player_with_relationships npc_id = "npc_friend" initial_affinity = next(p for p in player.r if p.id == npc_id).affinity # Act updateAffinity(player, npc_id, 10) # Assert updated_affinity = next(p for p in player.r if p.id == npc_id).affinity assert updated_affinity == initial_affinity + 10 assert updated_affinity == 80 def test_update_affinity_decreases(player_with_relationships): """Test that negative interaction decreases affinity.""" # Arrange player = player_with_relationships npc_id = "npc_friend" initial_affinity = next(p for p in player.r if p.id == npc_id).affinity # Act updateAffinity(player, npc_id, -20) # Assert updated_affinity = next(p for p in player.r if p.id == npc_id).affinity assert updated_affinity == initial_affinity - 20 assert updated_affinity == 50 def test_update_affinity_boundaries(player_with_relationships): """Test that affinity can go beyond typical bounds (implementation detail).""" # Arrange player = player_with_relationships npc_id = "npc_friend" # Act - increase beyond typical max updateAffinity(player, npc_id, 50) # Assert - value should be updated (no automatic clamping in updateAffinity itself) updated_affinity = next(p for p in player.r if p.id == npc_id).affinity assert updated_affinity == 120 # 70 + 50 # Act - decrease significantly updateAffinity(player, npc_id, -150) # Assert final_affinity = next(p for p in player.r if p.id == npc_id).affinity assert final_affinity == -30 # 120 - 150 def test_update_affinity_bidirectional(player_with_relationships): """Test that affinity can be updated in both directions.""" # Arrange player = player_with_relationships npc_id = "npc_acquaintance" initial_affinity = next(p for p in player.r if p.id == npc_id).affinity # Act - increase first updateAffinity(player, npc_id, 15) after_increase = next(p for p in player.r if p.id == npc_id).affinity # Then decrease updateAffinity(player, npc_id, -10) after_decrease = next(p for p in player.r if p.id == npc_id).affinity # Assert assert after_increase == initial_affinity + 15 assert after_decrease == after_increase - 10 assert after_decrease == initial_affinity + 5 # ============================================================================ # RELATIONSHIP HANDLING TESTS (3 tests) # ============================================================================ def test_handle_relationships_updates_score(player_with_active_relationship): """Test that handleRelationships processes relationship correctly.""" # Arrange player = player_with_active_relationship partner = player.r[0] relationship = player.relData[0] initial_score = relationship.relationshipScore # 65 # Act random.seed(42) handleRelationships(player, partner) # Assert - score changes based on correction mechanism # Initial score is 65 (above 60), so: +1 (weekly) -2 (correction) = -1 assert relationship.relationshipScore == initial_score - 1 assert relationship.relationshipScore == 64 def test_get_rel_data_returns_relationship(player_with_active_relationship): """Test that getRelData finds relationship by person ID.""" # Arrange player = player_with_active_relationship partner_id = "npc_partner" # Act rel_data = getRelData(player, partner_id) # Assert assert rel_data is not None assert rel_data.person2 == partner_id or rel_data.person1 == partner_id assert rel_data.relationshipStatus == "Dating" def test_get_rel_data_returns_none_if_not_exists(simple_player): """Test that getRelData returns None if relationship doesn't exist.""" # Arrange player = simple_player nonexistent_id = "npc_doesnt_exist" # Act rel_data = getRelData(player, nonexistent_id) # Assert assert rel_data is None # ============================================================================ # ROMANCE TESTS (4 tests) # ============================================================================ @patch.object(rm, 'get_person') def test_romance_initiates_relationship(mock_get_person, player_with_relationships): """Test that romance is initiated when affinity is sufficient.""" # Arrange player = player_with_relationships romantic_interest = next(p for p in player.r if p.id == "npc_romance") romantic_interest.affinity = 60 mock_get_person.return_value = romantic_interest # Act result = rm.romance(player, romantic_interest.id) # Assert assert result is True assert len(player.relData) > 0 new_rel = next((r for r in player.relData if r.person2 == romantic_interest.id), None) assert new_rel is not None assert new_rel.relationshipStatus == "Prospect" @patch.object(rm, 'get_person') def test_romance_fails_with_low_affinity(mock_get_person, player_with_relationships): """Test that romance fails when affinity is too low.""" # Arrange player = player_with_relationships acquaintance = next(p for p in player.r if p.id == "npc_acquaintance") acquaintance.affinity = 30 mock_get_person.return_value = acquaintance # Act result = rm.romance(player, acquaintance.id) # Assert assert result is False assert len(player.relData) == 0 def test_get_active_relationship_returns_partner(player_with_active_relationship): """Test that getActiveRelationship finds active romantic relationship.""" # Arrange player = player_with_active_relationship # Act active_rel = getActiveRelationship(player) # Assert assert active_rel is not None assert active_rel.relationshipStatus == "Dating" assert active_rel.person2 == "npc_partner" def test_get_active_relationship_returns_none_when_no_relationship(simple_player): """Test that getActiveRelationship returns None when no active relationship.""" # Arrange player = simple_player # Act active_rel = getActiveRelationship(player) # Assert assert active_rel is None def test_break_up_ends_relationship(player_with_active_relationship): """Test that breakup changes relationship status.""" # Arrange player = player_with_active_relationship partner_id = "npc_partner" # Act result = breakUp(player, partner_id) # Assert assert result is True relationship = getRelData(player, partner_id) assert relationship.relationshipStatus == "Broke Up" assert player.c.relationship is None def test_break_up_affects_relationship_status(player_with_active_relationship): """Test that breakup affects relationship status.""" # Arrange player = player_with_active_relationship partner_id = "npc_partner" # Act breakUp(player, partner_id) # Assert relationship = getRelData(player, partner_id) assert relationship.relationshipStatus == "Broke Up" # ============================================================================ # DATING TESTS (4 tests) # ============================================================================ @patch.object(rm, 'get_person') def test_partner_gift_increases_affinity(mock_get_person, player_with_active_relationship): """Test that giving a gift increases partner affinity.""" # Arrange player = player_with_active_relationship partner_id = "npc_partner" partner = player.r[0] initial_affinity = partner.affinity initial_money = player.c.money mock_get_person.return_value = partner random.seed(42) # Act result = partnerGift(player, partner_id) # Assert assert result is not False assert player.c.money == initial_money - 100 assert partner.affinity > initial_affinity def test_partner_gift_fails_insufficient_funds(player_with_active_relationship): """Test that gift fails when player doesn't have enough money.""" # Arrange player = player_with_active_relationship player.c.money = 50 partner_id = "npc_partner" # Act result = partnerGift(player, partner_id) # Assert assert result is False def test_date_night_creates_event(player_with_active_relationship): """Test that date night creates a message event.""" # Arrange player = player_with_active_relationship date_idea = "Picnic in the Park" random.seed(42) # Act result = dateNight(player, date_idea) # Assert assert result is not False relationship = getActiveRelationship(player) assert len(relationship.eventsLog) > 0 def test_get_date_ideas_returns_activities(): """Test that getDateIdeas returns list of date activities.""" # Act date_ideas = getDateIdeas() # Assert assert isinstance(date_ideas, list) assert len(date_ideas) > 0 assert all(isinstance(idea, DateIdea) for idea in date_ideas) idea_names = [idea.name for idea in date_ideas] assert "Picnic in the Park" in idea_names assert "Movie Night at Home" in idea_names def test_date_affects_relationship_score(player_with_active_relationship): """Test that successful date increases relationship score.""" # Arrange player = player_with_active_relationship relationship = getActiveRelationship(player) initial_score = relationship.relationshipScore date_idea = "Beach Day" random.seed(42) # Act result = dateNight(player, date_idea) # Assert assert result is not False assert relationship.relationshipScore > initial_score def test_date_night_fails_insufficient_resources(player_with_active_relationship): """Test that date fails when player lacks energy or money.""" # Arrange player = player_with_active_relationship player.c.energy = 1 player.c.money = 10 date_idea = "Fine Dining Experience" # Act result = dateNight(player, date_idea) # Assert assert result is False # ============================================================================ # RELATIONSHIP TYPES TESTS (3 tests) # ============================================================================ def test_family_relationships(): """Test family relationship creation and properties.""" # Arrange & Act relationship = create_test_relationship( "npc_mother", "player_001", status="Parent-Child", score=95 ) # Assert assert relationship.person1 == "npc_mother" assert relationship.person2 == "player_001" assert relationship.relationshipStatus == "Parent-Child" assert relationship.relationshipScore == 95 assert relationship.id is not None def test_friend_relationships(): """Test friend vs acquaintance relationships based on affinity.""" # Arrange player = create_test_player() friend = create_test_person("npc_friend", affinity=70) acquaintance = create_test_person("npc_acquaintance", affinity=30) player.r = [friend, acquaintance] # Assert assert friend.affinity > 60 # Friend threshold assert acquaintance.affinity < 50 # Acquaintance level def test_romantic_relationships(): """Test dating, engaged, and married relationship statuses.""" # Arrange & Act dating_rel = create_test_relationship("p1", "p2", status="Dating") engaged_rel = create_test_relationship("p3", "p4", status="Engaged") married_rel = create_test_relationship("p5", "p6", status="Married") # Assert assert dating_rel.relationshipStatus == "Dating" assert engaged_rel.relationshipStatus == "Engaged" assert married_rel.relationshipStatus == "Married" assert dating_rel.id != engaged_rel.id assert engaged_rel.id != married_rel.id # ============================================================================ # EDGE CASES TESTS (2 tests) # ============================================================================ @patch.object(rm, 'get_person') def test_relationship_with_self_edge_case(mock_get_person, simple_player): """Test edge case of attempting relationship with self.""" # Arrange player = simple_player player_id = player.c.id self_person = create_test_person(player_id, affinity=100) player.r = [self_person] mock_get_person.return_value = self_person # Act result = rm.romance(player, player_id) # Assert - system allows it (edge case in current implementation) # In production, this should be prevented if result: rel = next((r for r in player.relData if r.person2 == player_id), None) assert rel is not None or rel is None # Either outcome is tested def test_duplicate_relationships(player_with_relationships): """Test that duplicate relationships can exist (current behavior).""" # Arrange player = player_with_relationships person1_id = player.c.id person2_id = "npc_friend" # Act - create two relationships with same people rel1 = create_test_relationship(person1_id, person2_id, status="Friends", score=50) rel2 = create_test_relationship(person1_id, person2_id, status="Best Friends", score=75) player.relData.append(rel1) player.relData.append(rel2) # Assert - both exist with different IDs (not ideal, but current behavior) assert rel1.id != rel2.id assert len(player.relData) == 2 # ============================================================================ # ADDITIONAL TESTS # ============================================================================ def test_date_idea_class_initialization(): """Test that DateIdea class initializes correctly.""" # Arrange & Act date_idea = DateIdea( name="Test Date", energy_cost=3, money_cost=50, message="Test message", image="https://example.com/image.png" ) # Assert assert date_idea.name == "Test Date" assert date_idea.energy_cost == 3 assert date_idea.money_cost == 50 assert date_idea.message == "Test message" assert date_idea.image == "https://example.com/image.png" def test_handle_relationships_score_correction_low(player_with_active_relationship): """Test that handleRelationships corrects low relationship scores upward.""" # Arrange player = player_with_active_relationship partner = player.r[0] relationship = player.relData[0] relationship.relationshipScore = 30 # Below 40 threshold # Act random.seed(42) handleRelationships(player, partner) # Assert - score increases by 1 (weekly) + 2 (correction) = 3 assert relationship.relationshipScore == 33 def test_handle_relationships_score_correction_high(player_with_active_relationship): """Test that handleRelationships corrects high relationship scores downward.""" # Arrange player = player_with_active_relationship partner = player.r[0] relationship = player.relData[0] relationship.relationshipScore = 70 # Above 60 threshold # Act random.seed(42) handleRelationships(player, partner) # Assert - score increases by 1 (weekly) - 2 (correction) = -1 assert relationship.relationshipScore == 69