remove node info data source impl from dmacvicar#1059

libvirt/data_source_libvirt_node_info.test.go

@@ -0,0 +1,107 @@
+package libvirt
+
+import (
+	"encoding/xml"
+	"reflect"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+// MockClient represents a mock client for testing.
+type MockClient struct {
+	caps string // XML capabilities data
+}
+
+// ConnectGetCapabilities returns mock capabilities data.
+func (c *MockClient) ConnectGetCapabilities() (string, error) {
+	return c.caps, nil
+}
+
+// mockCapabilitiesXML provides a sample XML for libvirt capabilities.
+const mockCapabilitiesXML = `
+<capabilities>
+	<host>
+		<uuid>test-uuid</uuid>
+		<cpu>
+			<arch>x86_64</arch>
+			<model>model-name</model>
+			<vendor>vendor-name</vendor>
+			<topology sockets="2" dies="1" cores="4" threads="2"/>
+			<feature name="feature1"/>
+			<feature name="feature2"/>
+			<feature name="feature3"/>
+		</cpu>
+		<migration_features>
+			<live />
+			<uri_transports>
+				<uri_transport>transport#1</uri_transport>
+				<uri_transport>transport#2</uri_transport>
+				<uri_transport>transport#3</uri_transport>
+			</uri_transports>
+		</migration_features>
+	</host>
+</capabilities>
+`
+
+func Test_resourceLibvirtNodeInfoRead(t *testing.T) {
+	// Set up mock client with predefined capabilities XML
+	mockClient := &MockClient{caps: mockCapabilitiesXML}
+
+	// Initialize a Terraform resource data schema
+	d := schema.TestResourceDataRaw(t, datasourceLibvirtNodeInfo().Schema, map[string]interface{}{})
+
+	// Call the function under test
+	err := resourceLibvirtNodeInfoRead(d, mockClient)
+	if err != nil {
+		t.Fatalf("resourceLibvirtNodeInfoRead() error = %v", err)
+	}
+
+	// Define expected values based on mockCapabilitiesXML
+	expectedUUID := "test-uuid"
+	expectedArch := "x86_64"
+	expectedModel := "model-name"
+	expectedVendor := "vendor-name"
+	expectedSockets := 2
+	expectedDies := 1
+	expectedCores := 4
+	expectedThreads := 2
+	expectedLiveMigrationSupport := true
+	expectedLiveMigrationTransports := []interface{}{"transport#1", "transport#2", "transport#3"}
+	expectedFeatures := []interface{}{"feature1", "feature2", "feature3"}
+
+	// Assert that the resource data is set correctly
+	if got := d.Get("arch"); got != expectedArch {
+		t.Errorf("resourceLibvirtNodeInfoRead() arch = %v, want %v", got, expectedArch)
+	}
+	if got := d.Get("model"); got != expectedModel {
+		t.Errorf("resourceLibvirtNodeInfoRead() model = %v, want %v", got, expectedModel)
+	}
+	if got := d.Get("vendor"); got != expectedVendor {
+		t.Errorf("resourceLibvirtNodeInfoRead() vendor = %v, want %v", got, expectedVendor)
+	}
+	if got := d.Get("sockets"); got != expectedSockets {
+		t.Errorf("resourceLibvirtNodeInfoRead() sockets = %v, want %v", got, expectedSockets)
+	}
+	if got := d.Get("dies"); got != expectedDies {
+		t.Errorf("resourceLibvirtNodeInfoRead() dies = %v, want %v", got, expectedDies)
+	}
+	if got := d.Get("cores"); got != expectedCores {
+		t.Errorf("resourceLibvirtNodeInfoRead() cores = %v, want %v", got, expectedCores)
+	}
+	if got := d.Get("threads"); got != expectedThreads {
+		t.Errorf("resourceLibvirtNodeInfoRead() threads = %v, want %v", got, expectedThreads)
+	}
+	if got := d.Get("live_migration_support"); got != expectedLiveMigrationSupport {
+		t.Errorf("resourceLibvirtNodeInfoRead() live_migration_support = %v, want %v", got, expectedLiveMigrationSupport)
+	}
+	if got := d.Get("live_migration_transports"); !reflect.DeepEqual(got, expectedLiveMigrationTransports) {
+		t.Errorf("resourceLibvirtNodeInfoRead() live_migration_transports = %v, want %v", got, expectedLiveMigrationTransports)
+	}
+	if got := d.Get("features"); !reflect.DeepEqual(got, expectedFeatures) {
+		t.Errorf("resourceLibvirtNodeInfoRead() features = %v, want %v", got, expectedFeatures)
+	}
+	if got := d.Id(); got != expectedUUID {
+		t.Errorf("resourceLibvirtNodeInfoRead() UUID = %v, want %v", got, expectedUUID)
+	}
+}