import unittest
from app.models.feedback_manager import FeedbackManager

class TestFeedback(unittest.TestCase):
    def setUp(self):
        self.feedback_manager = FeedbackManager()

    def test_record_feedback(self):
        self.feedback_manager.record_feedback(
            piece_id=1,
            slot_id=1,
            suggestions={'piece_id': 1, 'confidence': 0.9},
            user_choice=1,
            is_correct=True
        )
        training_data = self.feedback_manager.get_training_data()
        self.assertIn('positive', training_data)
        self.assertGreaterEqual(len(training_data['positive']), 1)

if __name__ == '__main__':
    unittest.main()