Function TestNewRestAPI

modules/api_rest/api_rest_test.go:31–90

(t *testing.T)

}

func TestNewRestAPI(t *testing.T) {
	s := createMockSession(t)
	mod := NewRestAPI(s)

	if mod == nil {
		t.Fatal("NewRestAPI returned nil")
	}

	if mod.Name() != "api.rest" {
		t.Errorf("Expected name 'api.rest', got '%s'", mod.Name())
	}

	if mod.Author() != "Simone Margaritelli <evilsocket@gmail.com>" {
		t.Errorf("Unexpected author: %s", mod.Author())
	}

	if mod.Description() == "" {
		t.Error("Empty description")
	}

	// Check handlers
	handlers := mod.Handlers()
	expectedHandlers := []string{
		"api.rest on",
		"api.rest off",
		"api.rest.record off",
		"api.rest.record FILENAME",
		"api.rest.replay off",
		"api.rest.replay FILENAME",
	}

	if len(handlers) != len(expectedHandlers) {
		t.Errorf("Expected %d handlers, got %d", len(expectedHandlers), len(handlers))
	}

	handlerNames := make(map[string]bool)
	for _, h := range handlers {
		handlerNames[h.Name] = true
	}

	for _, expected := range expectedHandlers {
		if !handlerNames[expected] {
			t.Errorf("Handler '%s' not found", expected)
		}
	}

	// Check initial state
	if mod.recording {
		t.Error("Should not be recording initially")
	}
	if mod.replaying {
		t.Error("Should not be replaying initially")
	}
	if mod.useWebsocket {
		t.Error("Should not use websocket by default")
	}
	if mod.allowOrigin != "*" {
		t.Errorf("Expected default allowOrigin '*', got '%s'", mod.allowOrigin)