unit test

pkg/db/media_server_test.go

@@ -105,3 +105,212 @@ func TestAddMediaServerDuplicates(t *testing.T) {
 	t.Log("Test confirmed: Old AddMediaServer method creates duplicates")
 }
 
+func TestGetMediaServer(t *testing.T) {
+	dbPath := "./test_get_media_server.db"
+	defer os.Remove(dbPath)
+
+	db, err := NewMediaServerDB(dbPath)
+	if err != nil {
+		t.Fatalf("Failed to create database: %v", err)
+	}
+	defer db.Close()
+
+	// 添加一个媒体服务器
+	err = db.AddMediaServer("TestServer", "ZLM", "192.168.1.200", 8080, "admin", "pass123", "secret123", 0)
+	if err != nil {
+		t.Fatalf("Failed to add media server: %v", err)
+	}
+
+	// 获取服务器列表以获得ID
+	servers, err := db.ListMediaServers()
+	if err != nil {
+		t.Fatalf("Failed to list media servers: %v", err)
+	}
+	if len(servers) == 0 {
+		t.Fatal("No servers found")
+	}
+
+	// 通过ID获取服务器
+	server, err := db.GetMediaServer(servers[0].ID)
+	if err != nil {
+		t.Fatalf("Failed to get media server: %v", err)
+	}
+
+	// 验证数据
+	if server.Name != "TestServer" {
+		t.Errorf("Expected name 'TestServer', got '%s'", server.Name)
+	}
+	if server.Type != "ZLM" {
+		t.Errorf("Expected type 'ZLM', got '%s'", server.Type)
+	}
+	if server.IP != "192.168.1.200" {
+		t.Errorf("Expected IP '192.168.1.200', got '%s'", server.IP)
+	}
+	if server.Port != 8080 {
+		t.Errorf("Expected port 8080, got %d", server.Port)
+	}
+}
+
+func TestGetMediaServerNotFound(t *testing.T) {
+	dbPath := "./test_get_not_found.db"
+	defer os.Remove(dbPath)
+
+	db, err := NewMediaServerDB(dbPath)
+	if err != nil {
+		t.Fatalf("Failed to create database: %v", err)
+	}
+	defer db.Close()
+
+	// 尝试获取不存在的服务器
+	_, err = db.GetMediaServer(999)
+	if err == nil {
+		t.Error("Expected error when getting non-existent server, got nil")
+	}
+}
+
+func TestDeleteMediaServer(t *testing.T) {
+	dbPath := "./test_delete_media_server.db"
+	defer os.Remove(dbPath)
+
+	db, err := NewMediaServerDB(dbPath)
+	if err != nil {
+		t.Fatalf("Failed to create database: %v", err)
+	}
+	defer db.Close()
+
+	// 添加两个服务器
+	err = db.AddMediaServer("Server1", "SRS", "192.168.1.1", 1985, "", "", "", 0)
+	if err != nil {
+		t.Fatalf("Failed to add server1: %v", err)
+	}
+
+	err = db.AddMediaServer("Server2", "ZLM", "192.168.1.2", 8080, "", "", "", 0)
+	if err != nil {
+		t.Fatalf("Failed to add server2: %v", err)
+	}
+
+	// 获取服务器列表
+	servers, err := db.ListMediaServers()
+	if err != nil {
+		t.Fatalf("Failed to list servers: %v", err)
+	}
+	if len(servers) != 2 {
+		t.Fatalf("Expected 2 servers, got %d", len(servers))
+	}
+
+	// 删除第一个服务器
+	err = db.DeleteMediaServer(servers[0].ID)
+	if err != nil {
+		t.Fatalf("Failed to delete server: %v", err)
+	}
+
+	// 验证只剩一个服务器
+	servers, err = db.ListMediaServers()
+	if err != nil {
+		t.Fatalf("Failed to list servers after delete: %v", err)
+	}
+	if len(servers) != 1 {
+		t.Fatalf("Expected 1 server after delete, got %d", len(servers))
+	}
+}
+
+func TestSetDefaultMediaServer(t *testing.T) {
+	dbPath := "./test_set_default.db"
+	defer os.Remove(dbPath)
+
+	db, err := NewMediaServerDB(dbPath)
+	if err != nil {
+		t.Fatalf("Failed to create database: %v", err)
+	}
+	defer db.Close()
+
+	// 添加三个服务器
+	err = db.AddMediaServer("Server1", "SRS", "192.168.1.1", 1985, "", "", "", 1)
+	if err != nil {
+		t.Fatalf("Failed to add server1: %v", err)
+	}
+
+	err = db.AddMediaServer("Server2", "ZLM", "192.168.1.2", 8080, "", "", "", 0)
+	if err != nil {
+		t.Fatalf("Failed to add server2: %v", err)
+	}
+
+	err = db.AddMediaServer("Server3", "SRS", "192.168.1.3", 1985, "", "", "", 0)
+	if err != nil {
+		t.Fatalf("Failed to add server3: %v", err)
+	}
+
+	// 获取服务器列表
+	servers, err := db.ListMediaServers()
+	if err != nil {
+		t.Fatalf("Failed to list servers: %v", err)
+	}
+
+	// 找到 Server2 的 ID
+	var server2ID int
+	for _, s := range servers {
+		if s.Name == "Server2" {
+			server2ID = s.ID
+			break
+		}
+	}
+
+	// 设置 Server2 为默认
+	err = db.SetDefaultMediaServer(server2ID)
+	if err != nil {
+		t.Fatalf("Failed to set default server: %v", err)
+	}
+
+	// 验证只有 Server2 是默认的
+	servers, err = db.ListMediaServers()
+	if err != nil {
+		t.Fatalf("Failed to list servers: %v", err)
+	}
+
+	defaultCount := 0
+	for _, s := range servers {
+		if s.IsDefault == 1 {
+			defaultCount++
+			if s.Name != "Server2" {
+				t.Errorf("Expected Server2 to be default, got %s", s.Name)
+			}
+		}
+	}
+
+	if defaultCount != 1 {
+		t.Errorf("Expected exactly 1 default server, got %d", defaultCount)
+	}
+}
+
+func TestGetMediaServerByNameAndIP(t *testing.T) {
+	dbPath := "./test_get_by_name_ip.db"
+	defer os.Remove(dbPath)
+
+	db, err := NewMediaServerDB(dbPath)
+	if err != nil {
+		t.Fatalf("Failed to create database: %v", err)
+	}
+	defer db.Close()
+
+	// 添加服务器
+	err = db.AddMediaServer("MyServer", "SRS", "10.0.0.1", 1985, "user", "pass", "secret", 0)
+	if err != nil {
+		t.Fatalf("Failed to add server: %v", err)
+	}
+
+	// 通过名称和IP查询
+	server, err := db.GetMediaServerByNameAndIP("MyServer", "10.0.0.1")
+	if err != nil {
+		t.Fatalf("Failed to get server by name and IP: %v", err)
+	}
+
+	if server.Name != "MyServer" || server.IP != "10.0.0.1" {
+		t.Errorf("Server data mismatch: %+v", server)
+	}
+
+	// 查询不存在的组合
+	_, err = db.GetMediaServerByNameAndIP("MyServer", "10.0.0.2")
+	if err == nil {
+		t.Error("Expected error for non-existent name/IP combination, got nil")
+	}
+}