Skip to content

Commit

Permalink
Add test pkg and mock server for traceectl
Browse files Browse the repository at this point in the history
  • Loading branch information
ShohamBit committed Feb 5, 2025
1 parent 91194b5 commit 8506ecf
Show file tree
Hide file tree
Showing 7 changed files with 41 additions and 235 deletions.
52 changes: 0 additions & 52 deletions cmd/traceectl/pkg/cmd/test/test.go

This file was deleted.

17 changes: 0 additions & 17 deletions cmd/traceectl/pkg/mock/diagnostic_server.go

This file was deleted.

28 changes: 0 additions & 28 deletions cmd/traceectl/pkg/mock/event_server.go

This file was deleted.

11 changes: 0 additions & 11 deletions cmd/traceectl/pkg/mock/root_server.go

This file was deleted.

63 changes: 0 additions & 63 deletions cmd/traceectl/pkg/mock/server.go

This file was deleted.

64 changes: 0 additions & 64 deletions cmd/traceectl/pkg/mock/stream_server.go

This file was deleted.

41 changes: 41 additions & 0 deletions cmd/traceectl/test/server.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
package test

import (
"fmt"
"log"
"net"
"os"
)

const DefaultSocket = "/tmp/tracee.sock"

type Server struct {
addr string
listener net.Listener
}

func SetupMockSocket() (*Server, error) {
listener, err := net.Listen("unix", DefaultSocket)
if err != nil {
return nil, err
}
return &Server{
addr: DefaultSocket,
listener: listener,
}, nil
}

func (s *Server) TeardownMockSocket() error {
if s.listener != nil {
if err := s.listener.Close(); err != nil {
return fmt.Errorf("failed to close mock socket: %w", err) // Wrap the error
}
}
// Check if the socket file still exists and remove it
if _, err := os.Stat(DefaultSocket); err == nil {
if err := os.Remove(DefaultSocket); err != nil {
log.Printf("Warning: failed to remove mock socket file: %v", err)
}
}
return nil
}

0 comments on commit 8506ecf

Please sign in to comment.