add dbcrypt methods

johnstcn · Emyrk · commit 193d3b3c7027 · 2023-12-15T11:27:02.000-06:00
diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go
@@ -1557,6 +1557,28 @@ func (s *MethodTestSuite) TestTailnetFunctions() {
 	}))
 }
 
+func (s *MethodTestSuite) TestDBCrypt() {
+	s.Run("GetDBCryptKeys", s.Subtest(func(db database.Store, check *expects) {
+		check.Args().
+			Asserts(rbac.ResourceSystem, rbac.ActionRead).
+			Returns([]database.DBCryptKey{})
+	}))
+	s.Run("InsertDBCryptKey", s.Subtest(func(db database.Store, check *expects) {
+		check.Args(database.InsertDBCryptKeyParams{}).
+			Asserts(rbac.ResourceSystem, rbac.ActionCreate).
+			Returns()
+	}))
+	s.Run("RevokeDBCryptKey", s.Subtest(func(db database.Store, check *expects) {
+		err := db.InsertDBCryptKey(context.Background(), database.InsertDBCryptKeyParams{
+			ActiveKeyDigest: "revoke me",
+		})
+		s.NoError(err)
+		check.Args("revoke me").
+			Asserts(rbac.ResourceSystem, rbac.ActionUpdate).
+			Returns()
+	}))
+}
+
 func (s *MethodTestSuite) TestSystemFunctions() {
 	s.Run("UpdateUserLinkedID", s.Subtest(func(db database.Store, check *expects) {
 		u := dbgen.User(s.T(), db, database.User{})
