|
|
@@ -12,6 +12,7 @@ import (
|
|
|
apiv1 "github.com/OliveTin/OliveTin/gen/olivetin/api/v1"
|
|
|
apiv1connect "github.com/OliveTin/OliveTin/gen/olivetin/api/v1/apiv1connect"
|
|
|
config "github.com/OliveTin/OliveTin/internal/config"
|
|
|
+ "github.com/OliveTin/OliveTin/internal/entities"
|
|
|
"github.com/OliveTin/OliveTin/internal/executor"
|
|
|
|
|
|
"net/http"
|
|
|
@@ -93,3 +94,82 @@ func TestGetActionsAndStart(t *testing.T) {
|
|
|
|
|
|
defer conn.Close()
|
|
|
}
|
|
|
+
|
|
|
+func TestGetEntities(t *testing.T) {
|
|
|
+ cfg := config.DefaultConfig()
|
|
|
+
|
|
|
+ ts, client := getNewTestServerAndClient(t, cfg)
|
|
|
+ defer ts.Close()
|
|
|
+
|
|
|
+ setupTestEntities()
|
|
|
+
|
|
|
+ resp, err := client.GetEntities(context.Background(), connect.NewRequest(&apiv1.GetEntitiesRequest{}))
|
|
|
+
|
|
|
+ assert.NoError(t, err, "GetEntities should not return an error")
|
|
|
+ assert.NotNil(t, resp, "GetEntities response should not be nil")
|
|
|
+ assert.NotNil(t, resp.Msg, "GetEntities response message should not be nil")
|
|
|
+
|
|
|
+ entityDefinitions := resp.Msg.EntityDefinitions
|
|
|
+ assert.Equal(t, 3, len(entityDefinitions), "Should return 3 entity definitions")
|
|
|
+
|
|
|
+ validateEntityOrderAndStructure(t, entityDefinitions)
|
|
|
+ validateNoDuplicates(t, entityDefinitions)
|
|
|
+ validateConsistency(t, client, entityDefinitions)
|
|
|
+}
|
|
|
+
|
|
|
+func setupTestEntities() {
|
|
|
+ entities.ClearEntities("server")
|
|
|
+ entities.ClearEntities("database")
|
|
|
+ entities.ClearEntities("application")
|
|
|
+
|
|
|
+ entities.AddEntity("server", "zebra", map[string]any{"title": "Server Zebra", "hostname": "zebra.example.com"})
|
|
|
+ entities.AddEntity("server", "alpha", map[string]any{"title": "Server Alpha", "hostname": "alpha.example.com"})
|
|
|
+ entities.AddEntity("server", "beta", map[string]any{"title": "Server Beta", "hostname": "beta.example.com"})
|
|
|
+
|
|
|
+ entities.AddEntity("database", "mysql", map[string]any{"title": "MySQL Database", "type": "mysql"})
|
|
|
+ entities.AddEntity("database", "postgres", map[string]any{"title": "PostgreSQL Database", "type": "postgres"})
|
|
|
+
|
|
|
+ entities.AddEntity("application", "webapp", map[string]any{"title": "Web Application", "port": 8080})
|
|
|
+}
|
|
|
+
|
|
|
+func validateEntityOrderAndStructure(t *testing.T, entityDefinitions []*apiv1.EntityDefinition) {
|
|
|
+ assert.Equal(t, "application", entityDefinitions[0].Title, "First entity should be 'application' (alphabetically first)")
|
|
|
+ assert.Equal(t, 1, len(entityDefinitions[0].Instances), "Application should have 1 instance")
|
|
|
+ assert.Equal(t, "webapp", entityDefinitions[0].Instances[0].UniqueKey, "Application instance should be 'webapp'")
|
|
|
+
|
|
|
+ assert.Equal(t, "database", entityDefinitions[1].Title, "Second entity should be 'database' (alphabetically second)")
|
|
|
+ assert.Equal(t, 2, len(entityDefinitions[1].Instances), "Database should have 2 instances")
|
|
|
+ assert.Equal(t, "mysql", entityDefinitions[1].Instances[0].UniqueKey, "First database instance should be 'mysql' (alphabetically first)")
|
|
|
+ assert.Equal(t, "postgres", entityDefinitions[1].Instances[1].UniqueKey, "Second database instance should be 'postgres' (alphabetically second)")
|
|
|
+
|
|
|
+ assert.Equal(t, "server", entityDefinitions[2].Title, "Third entity should be 'server' (alphabetically third)")
|
|
|
+ assert.Equal(t, 3, len(entityDefinitions[2].Instances), "Server should have 3 instances")
|
|
|
+ assert.Equal(t, "alpha", entityDefinitions[2].Instances[0].UniqueKey, "First server instance should be 'alpha' (alphabetically first)")
|
|
|
+ assert.Equal(t, "beta", entityDefinitions[2].Instances[1].UniqueKey, "Second server instance should be 'beta' (alphabetically second)")
|
|
|
+ assert.Equal(t, "zebra", entityDefinitions[2].Instances[2].UniqueKey, "Third server instance should be 'zebra' (alphabetically third)")
|
|
|
+}
|
|
|
+
|
|
|
+func validateNoDuplicates(t *testing.T, entityDefinitions []*apiv1.EntityDefinition) {
|
|
|
+ instanceKeys := make(map[string]map[string]bool)
|
|
|
+ for _, def := range entityDefinitions {
|
|
|
+ instanceKeys[def.Title] = make(map[string]bool)
|
|
|
+ for _, inst := range def.Instances {
|
|
|
+ assert.False(t, instanceKeys[def.Title][inst.UniqueKey], "Instance key %s should not be duplicated in entity %s", inst.UniqueKey, def.Title)
|
|
|
+ instanceKeys[def.Title][inst.UniqueKey] = true
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func validateConsistency(t *testing.T, client apiv1connect.OliveTinApiServiceClient, entityDefinitions []*apiv1.EntityDefinition) {
|
|
|
+ resp2, err2 := client.GetEntities(context.Background(), connect.NewRequest(&apiv1.GetEntitiesRequest{}))
|
|
|
+ assert.NoError(t, err2, "Second GetEntities call should not return an error")
|
|
|
+ assert.Equal(t, len(entityDefinitions), len(resp2.Msg.EntityDefinitions), "Second call should return same number of entity definitions")
|
|
|
+
|
|
|
+ for i, def := range entityDefinitions {
|
|
|
+ assert.Equal(t, def.Title, resp2.Msg.EntityDefinitions[i].Title, "Entity order should be consistent across calls")
|
|
|
+ assert.Equal(t, len(def.Instances), len(resp2.Msg.EntityDefinitions[i].Instances), "Instance count should be consistent")
|
|
|
+ for j, inst := range def.Instances {
|
|
|
+ assert.Equal(t, inst.UniqueKey, resp2.Msg.EntityDefinitions[i].Instances[j].UniqueKey, "Instance order should be consistent across calls")
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|