WIP: Implement database tests for subject table

Fixes #81

go.mod

@@ -5,6 +5,7 @@ go 1.18
 require (
 	github.com/aws/aws-sdk-go v1.44.22
 	github.com/golang-migrate/migrate/v4 v4.15.2
+	github.com/google/uuid v1.3.0
 	github.com/lib/pq v1.10.6
 	github.com/spf13/viper v1.12.0
 	github.com/stretchr/testify v1.7.4

go.sum

@@ -597,6 +597,7 @@ github.com/google/uuid v1.0.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+
 github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/google/uuid v1.2.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
 github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/googleapis/gax-go/v2 v2.0.4/go.mod h1:0Wqv26UfaUD9n4G6kQubkQ+KchISgw+vpHVxEJEs9eg=
 github.com/googleapis/gax-go/v2 v2.0.5/go.mod h1:DWXyrwAJ9X0FpwwEdw+IPEYBICEFu5mhpdKc/us6bOk=

migrations/000001_create_subject_table.up.sql

@@ -1,5 +1,5 @@
 CREATE TABLE IF NOT EXISTS subjects (
-   id serial PRIMARY KEY,
+   id uuid PRIMARY KEY,
    name VARCHAR(255),
    type VARCHAR(50)
 );

tests/integration/helpers.go

@@ -0,0 +1,7 @@
+package integration_test
+
+type Subject struct {
+	ID   string
+	Name string
+	Type string
+}

tests/integration/integration_test.go

@@ -5,14 +5,18 @@ import (
 	"errors"
 	"fmt"
 	"net"
+	"strings"
 	"testing"
 	"time"
 
 	migrate "github.com/golang-migrate/migrate/v4"
 	"github.com/golang-migrate/migrate/v4/database/postgres"
 	_ "github.com/golang-migrate/migrate/v4/source/file"
+	"github.com/google/uuid"
 	_ "github.com/lib/pq"
 	"github.com/stretchr/testify/assert"
+	gorm_postgres "gorm.io/driver/postgres"
+	"gorm.io/gorm"
 )
 
 func getDatabaseConnection(t *testing.T) *sql.DB {
@@ -66,15 +70,71 @@ func getMigrationHelper(t *testing.T) *migrate.Migrate {
 	return m
 }
 
+func getGormHelper() *gorm.DB {
+	connStr := "host=localhost user=dbadmin dbname=compliance password=secret port=5432 sslmode=disable"
+	gormDB, _ := gorm.Open(gorm_postgres.Open(connStr), &gorm.Config{})
+	return gormDB
+}
+
+func getUUIDString() string {
+	value, _ := uuid.NewRandom()
+	return value.String()
+}
+
 // Each test assumes the database is unmanaged. The test is responsible for
 // setting up the state it requires for its test logic. This keeps the
 // getMigrationHelper() method clean of any assumptions about what the tests
 // expect from it.
 func TestSubject(t *testing.T) { // nolint:paralleltest // database tests should not run in parallel
-	getMigrationHelper(t)
-	// m.Up()
-	// Test logic
-	// m.Down()
+	m := getMigrationHelper(t)
+	if err := m.Up(); err != nil {
+		t.Fatalf("Unable to upgrade database: %s", err)
+	}
+
+	id := getUUIDString()
+	name := "cluster.example.com"
+	subjectTypeStr := getUUIDString()
+
+	s := Subject{ID: id, Name: name, Type: subjectTypeStr}
+	gormDB := getGormHelper()
+	gormDB.Create(&s)
+
+	subject := Subject{}
+	gormDB.First(&subject, "id = ?", id)
+
+	expectedSubject := Subject{ID: id, Name: name, Type: subjectTypeStr}
+	assert.Equal(t, expectedSubject.ID, subject.ID, "expected %s got %s", expectedSubject.ID, subject.ID)
+	assert.Equal(t, expectedSubject.Name, subject.Name, "expected %s got %s", expectedSubject.Name, subject.Name)
+	assert.Equal(t, expectedSubject.Type, subject.Type, "expected %s got %s", expectedSubject.Type, subject.Type)
+
+	if err := m.Drop(); err != nil {
+		t.Fatalf("Unable to drop database: %s", err)
+	}
+	if err := m.Down(); err != nil {
+		t.Fatalf("Unable to downgrade database: %s", err)
+	}
+}
+
+func TestSubjectFailsWithLongName(t *testing.T) { // nolint:paralleltest // database tests should not run in parallel
+	m := getMigrationHelper(t)
+	if err := m.Up(); err != nil {
+		t.Fatalf("Unable to upgrade database: %s", err)
+	}
+
+	id := getUUIDString()
+	name := strings.Repeat("a", 256)
+	subjectTypeStr := getUUIDString()
+
+	s := Subject{ID: id, Name: name, Type: subjectTypeStr}
+	gormDB := getGormHelper()
+	err := gormDB.Create(&s).Error
+	assert.NotEmpty(t, err, "Shouldn't be able to insert values longer than 255 characters")
+	if err := m.Drop(); err != nil {
+		t.Fatalf("Unable to drop database: %s", err)
+	}
+	if err := m.Down(); err != nil {
+		t.Fatalf("Unable to downgrade database: %s", err)
+	}
 }
 
 func TestMigration(t *testing.T) { // nolint:paralleltest // database tests should not run in parallel