Get IPAddress if resource already has IP assigned

internal/graphapi/ipaddress_test.go

@@ -69,11 +69,12 @@ func Test_IPAddress_Lifecycle(t *testing.T) {
 
 	ipbt := (&IPBlockTypeBuilder{}).MustNew(ctx)
 	ipb := (&IPBlockBuilder{IPBlockTypeID: ipbt.ID}).MustNew(ctx)
+	node := gidx.MustNewID(nodePrefix)
 
 	t.Run("Create", func(t *testing.T) {
 		ipa, err := client.CreateIPAddress(ctx, testclient.CreateIPAddressInput{
 			IP:          gofakeit.IPv4Address(),
-			NodeID:      gidx.MustNewID(nodePrefix),
+			NodeID:      node,
 			NodeOwnerID: gidx.MustNewID(ownerPrefix),
 			Reserved:    newBool(true),
 			IPBlockID:   ipb.ID,
@@ -90,6 +91,12 @@ func Test_IPAddress_Lifecycle(t *testing.T) {
 		require.NotNil(t, ipaUpdate)
 		require.Equal(t, false, ipaUpdate.UpdateIPAddress.IPAddress.Reserved)
 
+		addr, err := client.GetIPAddressByNode(ctx, node)
+
+		require.NoError(t, err)
+		require.NotNil(t, addr)
+		require.Equal(t, addr.IPAddressByNode.ID, ipa.CreateIPAddress.IPAddress.ID)
+
 		ipaDelete, err := client.DeleteIPAddress(ctx, ipa.CreateIPAddress.IPAddress.ID)
 
 		require.NoError(t, err)

internal/testclient/ipaddress.gql

@@ -12,6 +12,20 @@ query getIPAddress($id: ID!) {
   }
 }
 
+query getIPAddressByNode($id: ID!) {
+  ipAddressByNode(id: $id) {
+    id
+    ip
+    reserved
+    ipBlock {
+      id
+      prefix
+      allowAutoAllocate
+      allowAutoSubnet
+    }
+  }
+}
+
 mutation CreateIPAddress($input: CreateIPAddressInput!) {
   createIPAddress(input: $input) {
     ipAddress {

internal/testclient/schema/schema.graphql

@@ -539,6 +539,11 @@ type Query {
 		"""ID of the ip address"""
 		id: ID!
 	): IPAddress!
+	"""Look up ip address by Node ID"""
+	ipAddressByNode(
+		"""ID of the Node"""
+		id: ID!
+	): IPAddress!
 	"""Look up ip block by ID"""
 	ipBlock(
 		"""ID of the ip block"""

pkg/ipamclient/client.go

@@ -204,3 +204,22 @@ func (c *Client) GetIPAddresses(ctx context.Context, nodeID string) ([]IPAddress
 
 	return nodeIPs.NodeIPAddress.LoadBalancerFragment.IPAddresses, nil
 }
+
+// GetIPAddressByNode returns an IP Address by node id
+func (c *Client) GetIPAddressByNode(ctx context.Context, nodeID string) (*GetIPAddressByNode, error) {
+	_, err := gidx.Parse(nodeID)
+	if err != nil {
+		return nil, err
+	}
+
+	vars := map[string]interface{}{
+		"id": graphql.ID(nodeID),
+	}
+
+	var nodeIP GetIPAddressByNode
+	if err := c.gqlCli.Query(ctx, &nodeIP, vars); err != nil {
+		return nil, err
+	}
+
+	return &nodeIP, nil
+}

pkg/ipamclient/client_test.go

@@ -236,6 +236,33 @@ func TestGetIPAddressesByNodeID(t *testing.T) {
 	})
 }
 
+func TestGetIPAddressByNode(t *testing.T) {
+	cli := Client{
+		gqlCli: mustNewGQLTestClient(
+			`{
+	  "data": {
+		"ipAddressByNode": {
+		  "id": "ipamipa-12345",
+		  "ip": "192.168.10.1",
+		  "reserved": false
+		}
+	  }
+	}`),
+	}
+
+	ipResult, err := cli.GetIPAddressByNode(context.Background(), "nodeids-test")
+	require.NoError(t, err)
+	require.NotNil(t, ipResult)
+
+	assert.Equal(t, ipResult.IPAddress.ID, "ipamipa-12345")
+	assert.Equal(t, ipResult.IPAddress.IP, "192.168.10.1")
+	assert.False(t, ipResult.IPAddress.Reserved)
+
+	baddress, err := cli.GetIPAddressByNode(context.TODO(), "badprefix-test")
+	require.Error(t, err)
+	require.Nil(t, baddress)
+}
+
 func mustNewGQLTestClient(respJSON string) *graphql.Client {
 	mux := http.NewServeMux()
 	mux.HandleFunc("/query", func(w http.ResponseWriter, req *http.Request) {

pkg/ipamclient/types.go

@@ -109,3 +109,8 @@ type GetIPAddressesByNode struct {
 		} `graphql:"... on LoadBalancer"`
 	} `graphql:"node(id: $id)"`
 }
+
+// GetIPAddressByNode is the query used for getting an IP Address
+type GetIPAddressByNode struct {
+	IPAddress GetIPAddressResult `graphql:"ipAddressByNode(id: $id)"`
+}