Mining/pkg/mining/manager_test.go
Claude 3673757d2a
ax(batch): replace prose comments with usage examples across all packages
Converts comments that restate function signatures into concrete usage
examples per AX Principle 2. Affected files: database_race_test.go,
interface_test.go, errors_test.go, xmrig_test.go, service_test.go,
manager_test.go.

Co-Authored-By: Charon <charon@lethean.io>
2026-04-02 18:16:56 +01:00

140 lines
3.7 KiB
Go

package mining
import (
"context"
"os"
"path/filepath"
"runtime"
"testing"
)
// manager := setupTestManager(t)
// defer manager.Stop()
func setupTestManager(t *testing.T) *Manager {
dummyDir := t.TempDir()
executableName := "miner"
if runtime.GOOS == "windows" {
executableName += ".exe"
}
dummyPath := filepath.Join(dummyDir, executableName)
// Create a script that prints version and exits
var script []byte
if runtime.GOOS == "windows" {
script = []byte("@echo off\necho XMRig 6.24.0\n")
} else {
script = []byte("#!/bin/sh\necho 'XMRig 6.24.0'\n")
}
if err := os.WriteFile(dummyPath, script, 0755); err != nil {
t.Fatalf("Failed to create dummy miner executable: %v", err)
}
// Prepend the dummy directory to the PATH
originalPath := os.Getenv("PATH")
t.Cleanup(func() {
os.Setenv("PATH", originalPath)
})
os.Setenv("PATH", dummyDir+string(os.PathListSeparator)+originalPath)
return NewManager()
}
// miner, err := m.StartMiner(ctx, "xmrig", config)
func TestStartMiner_Good(t *testing.T) {
t.Skip("Skipping test that runs miner process as per request")
}
func TestStartMiner_Bad(t *testing.T) {
m := setupTestManager(t)
defer m.Stop()
config := &Config{
HTTPPort: 9001, // Use a different port to avoid conflict
Pool: "test:1234",
Wallet: "testwallet",
}
// Case 2: Attempt to start an unsupported miner
_, err := m.StartMiner(context.Background(), "unsupported", config)
if err == nil {
t.Error("Expected an error when starting an unsupported miner, but got nil")
}
}
func TestStartMiner_Ugly(t *testing.T) {
t.Skip("Skipping test that runs miner process")
}
// err := m.StopMiner(ctx, "xmrig"); // stops and removes the named miner
func TestStopMiner_Good(t *testing.T) {
t.Skip("Skipping test that runs miner process")
}
func TestStopMiner_Bad(t *testing.T) {
m := setupTestManager(t)
defer m.Stop()
// Case 2: Attempt to stop a non-existent miner
err := m.StopMiner(context.Background(), "nonexistent")
if err == nil {
t.Error("Expected an error when stopping a non-existent miner, but got nil")
}
}
// miner, err := m.GetMiner("xmrig"); miner.GetName() == "xmrig"
func TestGetMiner_Good(t *testing.T) {
m := setupTestManager(t)
defer m.Stop()
// Case 1: Get an existing miner (manually injected)
miner := NewXMRigMiner()
// Set name to match what StartMiner would produce usually ("xmrig")
// Since we inject it, we can use the default name or set one.
miner.Name = "xmrig-test"
m.mutex.Lock()
m.miners["xmrig-test"] = miner
m.mutex.Unlock()
retrievedMiner, err := m.GetMiner("xmrig-test")
if err != nil {
t.Fatalf("Expected to get miner, but got error: %v", err)
}
if retrievedMiner.GetName() != "xmrig-test" {
t.Errorf("Expected to get miner 'xmrig-test', but got %s", retrievedMiner.GetName())
}
}
func TestGetMiner_Bad(t *testing.T) {
m := setupTestManager(t)
defer m.Stop()
// Case 2: Attempt to get a non-existent miner
_, err := m.GetMiner("nonexistent")
if err == nil {
t.Error("Expected an error when getting a non-existent miner, but got nil")
}
}
// miners := m.ListMiners(); len(miners) == 1 after injecting one miner
func TestListMiners_Good(t *testing.T) {
m := setupTestManager(t)
defer m.Stop()
// Get initial count (may include autostarted miners from config)
initialMiners := m.ListMiners()
initialCount := len(initialMiners)
// Case 2: List miners when not empty (manually injected)
miner := NewXMRigMiner()
miner.Name = "xmrig-test"
m.mutex.Lock()
m.miners["xmrig-test"] = miner
m.mutex.Unlock()
finalMiners := m.ListMiners()
expectedCount := initialCount + 1
if len(finalMiners) != expectedCount {
t.Errorf("Expected %d miners, but got %d", expectedCount, len(finalMiners))
}
}