Added test file and changes to deployment flow

Author: cavinkavi

azurefunctions_setup/shared_azure_workload/azureworkload.py

@@ -11,7 +11,7 @@
 # Global variable for hostname
 hostname = socket.gethostname()
 
-# Placeholder for `execute_function`
+# Placeholder for `execute_function`, will be updated according to server/trace-func-py/trace_func.py
 def execute_function(input, runTime, totalMem):
     startTime = process_time_ns()
 

pkg/driver/deployment/azure_functions.go

@@ -34,14 +34,14 @@ func newAzureFunctionsDeployer() *azureFunctionsDeployer {
 
 func (afd *azureFunctionsDeployer) Deploy(cfg *config.Configuration) {
 	afd.functions = cfg.Functions
-	deployAzureFunctions(afd.functions)
+	DeployAzureFunctions(afd.functions)
 }
 
 func (afd *azureFunctionsDeployer) Clean() {
-	cleanAzureFunctions()
+	CleanAzureFunctions()
 }
 
-func deployAzureFunctions(functions []*common.Function) {
+func DeployAzureFunctions(functions []*common.Function) {
 	// 1. Run script to extract workload from trace_func.py
 	// 2. Initialize resources required for Azure Functions deployment
 	// 3. Create function folders
@@ -83,7 +83,7 @@ func deployAzureFunctions(functions []*common.Function) {
 	}
 }
 
-func cleanAzureFunctions() {
+func CleanAzureFunctions() {
 	// Load azurefunctionsconfig yaml file
 	config, err := LoadConfig("azurefunctions_setup/azurefunctionsconfig.yaml")
 	if err != nil {

pkg/driver/deployment/azure_functions_test.go

@@ -0,0 +1,119 @@
+package deployment_test
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"os"
+	"os/exec"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/vhive-serverless/loader/pkg/common"
+	"github.com/vhive-serverless/loader/pkg/driver/deployment"
+)
+
+func TestDeploymentWithAzureFunctions(t *testing.T) {
+
+	/* Test Setup */
+
+	// Mock functions for deployment
+	functions := []*common.Function{
+		{Name: "function0"},
+		{Name: "function1"},
+	}
+
+	// Change the working directory to the project root
+	err := os.Chdir("../../../")
+	if err != nil {
+		t.Fatalf("Failed to change working directory: %s", err)
+	}
+
+	// Log current working directory for debugging
+	wd, err := os.Getwd()
+	assert.NoError(t, err, "Failed to get current working directory")
+	t.Logf("Current working directory: %s", wd)
+
+	/* Deployment Tests */
+
+	// Deploy Azure Functions
+	t.Log("Testing `DeployAzureFunctions`...")
+	deployment.DeployAzureFunctions(functions)
+
+	// Verify the Function App exists (name and resource group from azurefunctionsconfig.yaml)
+	t.Log("Verifying Function App deployment...")
+	cmd := exec.Command("az", "functionapp", "show", "--name", "invitrofunctionapp", "--resource-group", "ExperimentResourceGroup")
+	err = cmd.Run()
+	assert.NoError(t, err, "Function App does not exist. Deployment failed.")
+
+	// Verify deployed functions and POST requests
+	t.Log("Verifying deployed functions...")
+	for i, function := range functions {
+		// Construct function URL
+		functionURL := fmt.Sprintf("https://invitrofunctionapp.azurewebsites.net/api/function%d", i)
+
+		// Prepare JSON payload
+		payload := map[string]interface{}{
+			"RuntimeInMilliSec": 2000,
+			"MemoryInMebiBytes": 256,
+		}
+		payloadBytes, _ := json.Marshal(payload)
+
+		// Create POST request
+		req, err := http.NewRequest("POST", functionURL, bytes.NewBuffer(payloadBytes))
+		assert.NoError(t, err, "Failed to create POST request for function: "+function.Name)
+		req.Header.Set("Content-Type", "application/json")
+
+		// Send POST request
+		client := &http.Client{}
+		resp, err := client.Do(req)
+
+		// Retry if not immediately accessible
+		if err != nil || resp.StatusCode != http.StatusOK {
+			t.Logf("Function %s is not immediately accessible. Retrying...", function.Name)
+			time.Sleep(5 * time.Second)
+			resp, err = client.Do(req)
+		}
+
+		// Validate response
+		assert.NoError(t, err, "Failed to access deployed function: "+function.Name)
+		assert.Equal(t, http.StatusOK, resp.StatusCode, "Deployed function is not accessible: "+function.Name)
+
+		// Parse and validate response body
+		var response map[string]interface{}
+		err = json.NewDecoder(resp.Body).Decode(&response)
+		assert.NoError(t, err, "Failed to decode JSON response from function: "+function.Name)
+		t.Logf("Response from function %s: %v", function.Name, response)
+
+		// Check response fields
+		assert.Equal(t, "Success", response["Status"])
+		assert.Contains(t, response, "DurationInMicroSec")
+		assert.Contains(t, response, "ExecutionTime")
+
+		resp.Body.Close()
+	}
+
+	/* Cleanup Tests */
+
+	// Cleanup resources
+	t.Log("Testing `CleanAzureFunctions`...")
+	deployment.CleanAzureFunctions()
+
+	// Verify Cleanup
+	t.Log("Verifying cleanup...")
+	// Check that temporary files and folders are removed
+	_, err = os.Stat("azure_functions_for_zip")
+	assert.True(t, os.IsNotExist(err), "Temp directory should be deleted")
+	_, err = os.Stat("azurefunctions.zip")
+	assert.True(t, os.IsNotExist(err), "Temp zip file should be deleted")
+
+	// Verify that the Azure resource group no longer exists
+	cmd = exec.Command("az", "group", "exists", "--name", "ExperimentResourceGroup")
+	output, err := cmd.Output()
+	assert.NoError(t, err, "Failed to check Azure resource group existence")
+	assert.Equal(t, "false\n", string(output), "Resource group should be deleted")
+
+	t.Log("End-to-End test and cleanup completed successfully.")
+}