53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
|
package mcpcmd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"dappco.re/go/mcp/pkg/mcp"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestRunServe_Good_ShutsDownService(t *testing.T) {
|
||
|
|
oldNew := newMCPService
|
||
|
|
oldRun := runMCPService
|
||
|
|
oldShutdown := shutdownMCPService
|
||
|
|
oldWorkspace := workspaceFlag
|
||
|
|
oldUnrestricted := unrestrictedFlag
|
||
|
|
|
||
|
|
t.Cleanup(func() {
|
||
|
|
newMCPService = oldNew
|
||
|
|
runMCPService = oldRun
|
||
|
|
shutdownMCPService = oldShutdown
|
||
|
|
workspaceFlag = oldWorkspace
|
||
|
|
unrestrictedFlag = oldUnrestricted
|
||
|
|
})
|
||
|
|
|
||
|
|
workspaceFlag = ""
|
||
|
|
unrestrictedFlag = false
|
||
|
|
|
||
|
|
var runCalled bool
|
||
|
|
var shutdownCalled bool
|
||
|
|
|
||
|
|
newMCPService = func(opts mcp.Options) (*mcp.Service, error) {
|
||
|
|
return mcp.New(mcp.Options{})
|
||
|
|
}
|
||
|
|
runMCPService = func(svc *mcp.Service, ctx context.Context) error {
|
||
|
|
runCalled = true
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
shutdownMCPService = func(svc *mcp.Service, ctx context.Context) error {
|
||
|
|
shutdownCalled = true
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := runServe(); err != nil {
|
||
|
|
t.Fatalf("runServe() returned error: %v", err)
|
||
|
|
}
|
||
|
|
if !runCalled {
|
||
|
|
t.Fatal("expected runMCPService to be called")
|
||
|
|
}
|
||
|
|
if !shutdownCalled {
|
||
|
|
t.Fatal("expected shutdownMCPService to be called")
|
||
|
|
}
|
||
|
|
}
|