terraform-provider-proxmox/fwprovider/test/resource_container_test.go

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

package test

import (
	"context"
	"fmt"
	"math/rand"
	"testing"
	"time"

	"github.com/hashicorp/terraform-plugin-testing/helper/resource"
	"github.com/hashicorp/terraform-plugin-testing/terraform"
	"github.com/stretchr/testify/require"
)

const (
	accTestContainerName = "proxmox_virtual_environment_container.test_container"
)

//nolint:gochecknoglobals
var (
	accTestContainerID  = 100000 + rand.Intn(99999) //nolint:gosec
	accCloneContainerID = 200000 + rand.Intn(99999) //nolint:gosec
)

func TestAccResourceContainer(t *testing.T) { //nolint:wsl
	// download fails with 404 or "exit code 8" if run in parallel
	// t.Parallel()

	te := InitEnvironment(t)

	resource.Test(t, resource.TestCase{
		ProtoV6ProviderFactories: te.AccProviders,
		Steps: []resource.TestStep{
			{
				Config: te.RenderConfig(testAccResourceContainerCreateConfig(te, false)),
				Check:  testAccResourceContainerCreateCheck(te),
			},
			{
				Config: te.RenderConfig(testAccResourceContainerCreateConfig(te, true) + testAccResourceContainerCreateCloneConfig(te)),
				Check:  testAccResourceContainerCreateCloneCheck(te),
			},
		},
	})
}

func testAccResourceContainerCreateConfig(te *Environment, isTemplate bool) string {
	te.t.Helper()

	return fmt.Sprintf(`
resource "proxmox_virtual_environment_download_file" "ubuntu_container_template" {
	content_type = "vztmpl"
	datastore_id = "local"
	node_name = "{{.NodeName}}"
	url = "{{.ContainerImagesServer}}/images/system/ubuntu-23.04-standard_23.04-1_amd64.tar.zst"
    overwrite_unmanaged = true
}
resource "proxmox_virtual_environment_container" "test_container" {
  node_name = "{{.NodeName}}"
  vm_id     = %d
  template  = %t

  disk {
    datastore_id = "local-lvm"
    size         = 4
  }

  mount_point {
    volume = "local-lvm"
    size   = "4G"
    path   = "mnt/local"
  }

  description = <<-EOT
    my
    description
    value
  EOT

  initialization {
    hostname = "test"

    ip_config {
      ipv4 {
        address = "dhcp"
      }
    }
  }

  network_interface {
    name = "vmbr0"
  }

  operating_system {
	template_file_id = proxmox_virtual_environment_download_file.ubuntu_container_template.id
    type             = "ubuntu"
  }
}
`, accTestContainerID, isTemplate)
}

func testAccResourceContainerCreateCheck(te *Environment) resource.TestCheckFunc {
	te.t.Helper()

	return resource.ComposeTestCheckFunc(
		resource.TestCheckResourceAttr(accTestContainerName, "description", "my\ndescription\nvalue\n"),
		func(*terraform.State) error {
			ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
			defer cancel()

			err := te.NodeClient().Container(accTestContainerID).WaitForContainerStatus(ctx, "running")
			require.NoError(te.t, err, "container did not start")

			return nil
		},
	)
}

func testAccResourceContainerCreateCloneConfig(te *Environment) string {
	te.t.Helper()

	return fmt.Sprintf(`
resource "proxmox_virtual_environment_container" "test_container_clone" {
  depends_on = [proxmox_virtual_environment_container.test_container]
  node_name = "{{.NodeName}}"
  vm_id     = %d

  clone {
	vm_id = proxmox_virtual_environment_container.test_container.id
  }

  initialization {
    hostname = "test-clone"
  }
}
`, accCloneContainerID)
}

func testAccResourceContainerCreateCloneCheck(te *Environment) resource.TestCheckFunc {
	te.t.Helper()

	return resource.ComposeTestCheckFunc(
		func(*terraform.State) error {
			ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
			defer cancel()

			err := te.NodeClient().Container(accCloneContainerID).WaitForContainerStatus(ctx, "running")
			require.NoError(te.t, err, "container did not start")

			return nil
		},
	)
}