From 94411bf6864b736e489b5685ff684cccb9f6326e Mon Sep 17 00:00:00 2001
From: KCarretto <Kcarretto@gmail.com>
Date: Tue, 23 Jan 2024 23:12:38 +0000
Subject: [PATCH] Added some tests for the proto enums

---
 .../c2/c2pb/enum_host_platform_test.go        | 24 +++++++++++++++++++
 .../c2/c2pb/enum_process_status_test.go       | 24 +++++++++++++++++++
 2 files changed, 48 insertions(+)

diff --git a/tavern/internal/c2/c2pb/enum_host_platform_test.go b/tavern/internal/c2/c2pb/enum_host_platform_test.go
index 9896195c5..673c68acc 100644
--- a/tavern/internal/c2/c2pb/enum_host_platform_test.go
+++ b/tavern/internal/c2/c2pb/enum_host_platform_test.go
@@ -1,12 +1,36 @@
 package c2pb_test
 
 import (
+	"bytes"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 	"realm.pub/tavern/internal/c2/c2pb"
 )
 
+func TestHostPlatformValues(t *testing.T) {
+	assert.NotEmpty(t, c2pb.Host_Platform(0).Values())
+}
+
+func TestHostPlatformValue(t *testing.T) {
+	val, err := c2pb.Host_Platform(0).Value()
+	require.NoError(t, err)
+	require.NotNil(t, val)
+}
+
+func TestHostPlatformMarshalGraphQL(t *testing.T) {
+	var buf bytes.Buffer
+	c2pb.Host_Platform(0).MarshalGQL(&buf)
+	assert.Equal(t, `"PLATFORM_UNSPECIFIED"`, buf.String())
+}
+
+func TestHostPlatformUnmarshalGraphQL(t *testing.T) {
+	var platform c2pb.Host_Platform
+	assert.NoError(t, (*c2pb.Host_Platform).UnmarshalGQL(&platform, `PLATFORM_LINUX`))
+	assert.Equal(t, c2pb.Host_PLATFORM_LINUX, platform)
+}
+
 func TestHostPlatformScan(t *testing.T) {
 	tests := []struct {
 		name         string
diff --git a/tavern/internal/c2/c2pb/enum_process_status_test.go b/tavern/internal/c2/c2pb/enum_process_status_test.go
index c1febd695..b4c44a00b 100644
--- a/tavern/internal/c2/c2pb/enum_process_status_test.go
+++ b/tavern/internal/c2/c2pb/enum_process_status_test.go
@@ -1,12 +1,36 @@
 package c2pb_test
 
 import (
+	"bytes"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 	"realm.pub/tavern/internal/c2/c2pb"
 )
 
+func TestProcessStatusValues(t *testing.T) {
+	assert.NotEmpty(t, c2pb.Process_Status(0).Values())
+}
+
+func TestProcessStatusValue(t *testing.T) {
+	val, err := c2pb.Process_Status(0).Value()
+	require.NoError(t, err)
+	require.NotNil(t, val)
+}
+
+func TestProcessStatusMarshalGraphQL(t *testing.T) {
+	var buf bytes.Buffer
+	c2pb.Process_Status(0).MarshalGQL(&buf)
+	assert.Equal(t, `"STATUS_UNSPECIFIED"`, buf.String())
+}
+
+func TestProcessStatusUnmarshalGraphQL(t *testing.T) {
+	var status c2pb.Process_Status
+	assert.NoError(t, (*c2pb.Process_Status).UnmarshalGQL(&status, `STATUS_IDLE`))
+	assert.Equal(t, c2pb.Process_STATUS_IDLE, status)
+}
+
 func TestProcessStatusScan(t *testing.T) {
 	tests := []struct {
 		name       string