#include <gtest/gtest.h>
#include "displayflow/core/session/session_manager.h"

using namespace displayflow::core;

class SessionManagerTest : public ::testing::Test {
protected:
    void SetUp() override {
        manager_ = std::make_shared<SessionManager>();
    }
    
    void TearDown() override {
        manager_.reset();
    }
    
    std::shared_ptr<SessionManager> manager_;
};

TEST_F(SessionManagerTest, CreateSession) {
    auto session = manager_->CreateSession(RoleType::Host);
    ASSERT_NE(session, nullptr);
    EXPECT_EQ(session->GetRole(), RoleType::Host);
}

TEST_F(SessionManagerTest, GetSession) {
    auto session = manager_->CreateSession(RoleType::Client);
    ASSERT_NE(session, nullptr);
    
    SessionId id = session->GetId();
    auto retrieved = manager_->GetSession(id);
    ASSERT_NE(retrieved, nullptr);
    EXPECT_EQ(retrieved->GetId(), id);
}

// 添加更多测试...