Merge pull request #129 from rhmdnd/create-controls-table

Add a table for controls
rhmdnd · Aug 4, 2022 · f78eb9d · f78eb9d
2 parents f3d84f2 + a52579c
commit f78eb9d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/migrations/000008_create_controls_table.down.sql b/migrations/000008_create_controls_table.down.sql
@@ -0,0 +1 @@
+DROP TABLE IF EXISTS controls;
diff --git a/migrations/000008_create_controls_table.up.sql b/migrations/000008_create_controls_table.up.sql
@@ -0,0 +1,9 @@
+CREATE TABLE IF NOT EXISTS controls (
+  id uuid PRIMARY KEY,
+  name VARCHAR(255),
+  severity VARCHAR(50),
+  profile_id UUID,
+  metadata_id UUID,
+  CONSTRAINT fk_controls_metadata_id FOREIGN KEY (metadata_id) REFERENCES metadata (id),
+  CONSTRAINT fk_controls_profile_id FOREIGN KEY (profile_id) REFERENCES profiles (id)
+);
diff --git a/tests/integration_test.go b/tests/integration_test.go
@@ -303,7 +303,7 @@ func TestMigration(t *testing.T) { // nolint:paralleltest // database tests shou
 	// Upgrade the database and make sure all upgrades apply cleanly.
 	err = m.Up()
 	version, dirty, _ = m.Version()
-	expectedVersion = uint(7)
+	expectedVersion = uint(8)
 	assert.Equal(t, expectedVersion, version, "Database version mismatch: want %d but got %d", expectedVersion, version)
 	assert.Equal(t, false, dirty, "Database state mismatch: want %t but got %t", false, dirty)
 	assert.Equal(t, err, nil, "Error upgrading the database: %s", err)