"""Unit tests for attendance processor.""" import unittest from datetime import datetime from attendance_app.processor import AttendanceProcessor from attendance_app.database import MockUserDatabase class TestAttendanceProcessor(unittest.TestCase): """Test AttendanceProcessor class.""" def setUp(self): """Set up test fixtures.""" # Mock user database self.mock_users = { 'user123': ('S001', 'Taro Yamada', 'ヤマダ タロウ', 'CS', '01'), 'user456': ('S002', 'Hanako Tanaka', 'タナカ ハナコ', 'AM', '01'), 'staff1': ('M001', 'Ken Sato', 'サトウ ケン', 'Staff', '02'), } self.db = MockUserDatabase(self.mock_users) self.processor = AttendanceProcessor(self.db.lookup_user) def test_simple_login_logout(self): """Test simple login and logout.""" events = { 'GRL-01': [ [datetime(2024, 11, 27, 14, 30), 'on:user123'], [datetime(2024, 11, 27, 16, 30), 'off:user123'], ] } from_time = datetime(2024, 11, 27, 14, 0) to_time = datetime(2024, 11, 27, 17, 0) results = self.processor.process_events(events, from_time, to_time) self.assertEqual(len(results), 1) record = results[0] self.assertEqual(record[0], 'GRL-01') # machine self.assertEqual(record[1], '14:30〜16:30') # time range self.assertEqual(record[4], 'user123') # user_id self.assertEqual(record[5], 'S001') # student_no self.assertEqual(record[6], 'Taro Yamada') # name def test_login_without_logout(self): """Test login without matching logout.""" events = { 'GRL-01': [ [datetime(2024, 11, 27, 14, 30), 'on:user123'], ] } from_time = datetime(2024, 11, 27, 14, 0) to_time = datetime(2024, 11, 27, 17, 0) results = self.processor.process_events(events, from_time, to_time) self.assertEqual(len(results), 1) record = results[0] self.assertEqual(record[2], '---') # logout time missing def test_orphaned_logout(self): """Test logout without matching login.""" events = { 'GRL-01': [ [datetime(2024, 11, 27, 16, 30), 'off:user123'], ] } from_time = datetime(2024, 11, 27, 14, 0) to_time = datetime(2024, 11, 27, 17, 0) results = self.processor.process_events(events, from_time, to_time) self.assertEqual(len(results), 1) record = results[0] self.assertTrue(record[1].startswith('?〜')) # login time unknown def test_double_login_rejection(self): """Test double login rejection.""" events = { 'GRL-01': [ [datetime(2024, 11, 27, 14, 30), 'on:user123'], [datetime(2024, 11, 27, 14, 35), 'reject:user123:GRL-02'], ] } from_time = datetime(2024, 11, 27, 14, 0) to_time = datetime(2024, 11, 27, 17, 0) results = self.processor.process_events(events, from_time, to_time) # Login should be rejected (removed) self.assertEqual(len(results), 0) def test_startup_as_implicit_logout(self): """Test startup event as implicit logout.""" events = { 'GRL-01': [ [datetime(2024, 11, 27, 14, 30), 'on:user123'], [datetime(2024, 11, 27, 16, 0), 'startup'], ] } from_time = datetime(2024, 11, 27, 14, 0) to_time = datetime(2024, 11, 27, 17, 0) results = self.processor.process_events(events, from_time, to_time) self.assertEqual(len(results), 1) record = results[0] self.assertEqual(record[2], '16:00') # logout from startup def test_shutdown_as_implicit_logout(self): """Test shutdown event as implicit logout.""" events = { 'GRL-01': [ [datetime(2024, 11, 27, 14, 30), 'on:user123'], [datetime(2024, 11, 27, 18, 0), 'shutdown'], ] } from_time = datetime(2024, 11, 27, 14, 0) to_time = datetime(2024, 11, 27, 19, 0) results = self.processor.process_events(events, from_time, to_time) self.assertEqual(len(results), 1) record = results[0] self.assertEqual(record[2], '18:00') def test_unknown_user(self): """Test handling of unknown user.""" events = { 'GRL-01': [ [datetime(2024, 11, 27, 14, 30), 'on:unknown'], ] } from_time = datetime(2024, 11, 27, 14, 0) to_time = datetime(2024, 11, 27, 17, 0) results = self.processor.process_events(events, from_time, to_time) self.assertEqual(len(results), 1) record = results[0] self.assertEqual(record[5], '????') # unknown student number self.assertEqual(record[6], 'unknown') # user_id as name def test_multiple_machines(self): """Test events from multiple machines.""" events = { 'GRL-01': [ [datetime(2024, 11, 27, 14, 30), 'on:user123'], [datetime(2024, 11, 27, 16, 30), 'off:user123'], ], 'ML1-05': [ [datetime(2024, 11, 27, 15, 0), 'on:user456'], [datetime(2024, 11, 27, 16, 0), 'off:user456'], ] } from_time = datetime(2024, 11, 27, 14, 0) to_time = datetime(2024, 11, 27, 17, 0) results = self.processor.process_events(events, from_time, to_time) self.assertEqual(len(results), 2) machines = [r[0] for r in results] self.assertIn('GRL-01', machines) self.assertIn('ML1-05', machines) class TestFilterByRooms(unittest.TestCase): """Test room filtering.""" def setUp(self): """Set up test fixtures.""" mock_users = {'user123': ('S001', 'Test User', 'テスト', 'CS', '01')} self.db = MockUserDatabase(mock_users) self.processor = AttendanceProcessor(self.db.lookup_user) def test_filter_grl_room(self): """Test filtering by GRL room.""" records = [ ['GRL-01', '14:30', '16:30', 'on', 'user123', 'S001', 'Test', 'テスト', 'CS', '01', ''], ['ML1-01', '14:30', '16:30', 'on', 'user123', 'S001', 'Test', 'テスト', 'CS', '01', ''], ] room_filters = {'GRL': True, 'ML1': False, 'ML2': False, 'IML-A': False, 'IML-B': False} filtered = self.processor.filter_by_rooms(records, room_filters) self.assertEqual(len(filtered), 1) self.assertEqual(filtered[0][0], 'GRL-01') def test_filter_iml_subrooms(self): """Test filtering IML sub-rooms.""" records = [ ['IML-01', '14:30', '16:30', 'on', 'user123', 'S001', 'Test', 'テスト', 'CS', '01', ''], # Room A ['IML-62', '14:30', '16:30', 'on', 'user123', 'S001', 'Test', 'テスト', 'CS', '01', ''], # Room B ] # Test IML-A only room_filters = {'GRL': False, 'ML1': False, 'ML2': False, 'IML-A': True, 'IML-B': False} filtered = self.processor.filter_by_rooms(records, room_filters) self.assertEqual(len(filtered), 1) self.assertEqual(filtered[0][0], 'IML-01') # Test IML-B only room_filters = {'GRL': False, 'ML1': False, 'ML2': False, 'IML-A': False, 'IML-B': True} filtered = self.processor.filter_by_rooms(records, room_filters) self.assertEqual(len(filtered), 1) self.assertEqual(filtered[0][0], 'IML-62') def test_unique_users_filter(self): """Test unique users filtering.""" records = [ ['GRL-01', '14:30', '15:30', 'on', 'user123', 'S001', 'Test', 'テスト', 'CS', '01', ''], ['GRL-02', '15:45', '16:30', 'on', 'user123', 'S001', 'Test', 'テスト', 'CS', '01', ''], ] room_filters = {'GRL': True, 'ML1': False, 'ML2': False, 'IML-A': False, 'IML-B': False} # Without unique filter filtered = self.processor.filter_by_rooms(records, room_filters, unique_users=False) self.assertEqual(len(filtered), 2) # With unique filter filtered = self.processor.filter_by_rooms(records, room_filters, unique_users=True) self.assertEqual(len(filtered), 1) if __name__ == '__main__': unittest.main()