add test for License

Author: elchananarb

test/integration/pre_commit_test.go

@@ -26,33 +26,6 @@ func TestHooksPreCommitFullIntegration(t *testing.T) {
 	hookPath := filepath.Join(tmpDir, ".git", "hooks", "pre-commit")
 	assert.FileExists(t, hookPath, "Hook should be installed")
 
-	//	// Create file with mock secrets
-	//	fileContent := `MOCK SECRET
-	//ghp_mocksecretAAAAAAAAAAAAAAAAAAAAAAAAA
-	//ANOTHER MOCK SECRET`
-	//	secretFilePath := filepath.Join(tmpDir, "secret.txt")
-	//	assert.NoError(t, os.WriteFile(secretFilePath, []byte(fileContent), 0644))
-	//
-	//	// Stage the file
-	//	execCmd(t, tmpDir, "git", "add", "secret.txt")
-	//
-	//	// Run secrets scan (expect failure due to secret detection)
-	//	err, output := executeCommand(t, "hooks", "pre-commit", "secrets-scan")
-	//	assert.Error(t, err, "Scan should fail due to secret")
-	//	assert.Contains(t, output.String(), "Secret detected")
-	//
-	//	// Extract result IDs from output
-	//	resultIds := parseResultIDs(output.String())
-	//	assert.NotEmpty(t, resultIds, "Should detect result IDs")
-	//
-	//	// Ignore detected secrets by ID
-	//	output = executeCmdNilAssertion(t, "Ignoring detected secrets", "hooks", "pre-commit", "secrets-ignore", "--resultIds", strings.Join(resultIds, ","))
-	//	assert.Contains(t, output.String(), "Added new IDs to .checkmarx_ignore")
-	//
-	//	// Run secrets scan again (expect success after ignoring)
-	//	output = executeCmdNilAssertion(t, "Running secrets scan after ignoring", "hooks", "pre-commit", "secrets-scan")
-	//	assert.Contains(t, output.String(), "No secrets detected")
-
 	// Uninstall pre-commit hook
 	_ = executeCmdNilAssertion(t, "Uninstalling pre-commit hook", "hooks", "pre-commit", "secrets-uninstall-git-hook")
 
@@ -62,6 +35,59 @@ func TestHooksPreCommitFullIntegration(t *testing.T) {
 	assert.NotContains(t, string(data), "cx-secret-detection", "Hook content should be cleared after uninstall")
 }
 
+func TestHooksPreCommitLicenseValidation(t *testing.T) {
+	tmpDir, cleanup := setupTempDir(t)
+	defer cleanup()
+
+	// Initialize Git repository
+	execCmd(t, tmpDir, "git", "init")
+
+	//// Test case 1: Without license (should fail)
+	//t.Run("Without Enterprise License", func(t *testing.T) {
+	//	// Set environment to simulate no license
+	//	originalEnv := os.Getenv("CX_LICENSE")
+	//	os.Setenv("CX_LICENSE", "")
+	//	defer os.Setenv("CX_LICENSE", originalEnv)
+	//
+	//	// Try to install pre-commit hook
+	//	args := []string{"hooks", "pre-commit", "secrets-install-git-hook"}
+	//	err, output := executeCommand(t, args...)
+	//
+	//	// Verify error message
+	//	assert.Error(t, err)
+	//	assert.Contains(t, output.String(), "License validation failed")
+	//	assert.Contains(t, output.String(), "Please verify your CxOne License includes Enterprise Secrets")
+	//
+	//	// Verify hook was not installed
+	//	hookPath := filepath.Join(tmpDir, ".git", "hooks", "pre-commit")
+	//	_, err = os.Stat(hookPath)
+	//	assert.True(t, os.IsNotExist(err), "Hook should not be installed without license")
+	//})
+
+	// Test case 2: With license (should succeed)
+	t.Run("With Enterprise License", func(t *testing.T) {
+		// Set environment to simulate valid license
+		originalEnv := os.Getenv("CX_LICENSE")
+		os.Setenv("CX_LICENSE", "valid-license-key") // Replace with actual license
+		defer os.Setenv("CX_LICENSE", originalEnv)
+
+		// Install pre-commit hook
+		_ = executeCmdNilAssertion(t, "Installing pre-commit hook", "hooks", "pre-commit", "secrets-install-git-hook")
+
+		// Verify hook installation
+		hookPath := filepath.Join(tmpDir, ".git", "hooks", "pre-commit")
+		assert.FileExists(t, hookPath, "Hook should be installed with valid license")
+
+		// Verify hook content
+		data, err := os.ReadFile(hookPath)
+		assert.NoError(t, err)
+		assert.Contains(t, string(data), "cx-secret-detection", "Hook should contain secret detection code")
+
+		// Clean up - uninstall hook
+		_ = executeCmdNilAssertion(t, "Uninstalling pre-commit hook", "hooks", "pre-commit", "secrets-uninstall-git-hook")
+	})
+}
+
 // Helper functions
 func execCmd(t *testing.T, dir string, name string, args ...string) {
 	cmd := exec.Command(name, args...)