Finish license tests

Emyrk · Emyrk · commit 6763fbf1da34 · 2023-02-02T19:34:40.000-06:00
diff --git a/coderd/authzquery/authz_test.go b/coderd/authzquery/authz_test.go
@@ -17,7 +17,7 @@ import (
 	"cdr.dev/slog"
 	"github.com/coder/coder/coderd/authzquery"
 	"github.com/coder/coder/coderd/coderdtest"
-	"github.com/coder/coder/coderd/database/databasefake"
+	"github.com/coder/coder/coderd/database/dbfake"
 	"github.com/coder/coder/coderd/rbac"
 )
 
@@ -43,7 +43,7 @@ func TestNotAuthorizedError(t *testing.T) {
 // as only the first db call will be made. But it is better than nothing.
 func TestAuthzQueryRecursive(t *testing.T) {
 	t.Parallel()
-	q := authzquery.NewAuthzQuerier(databasefake.New(), &coderdtest.RecordingAuthorizer{
+	q := authzquery.NewAuthzQuerier(dbfake.New(), &coderdtest.RecordingAuthorizer{
 		Wrapped: &coderdtest.FakeAuthorizer{AlwaysReturn: nil},
 	}, slog.Make())
 	actor := rbac.Subject{
diff --git a/coderd/authzquery/license_test.go b/coderd/authzquery/license_test.go
@@ -21,4 +21,56 @@ func (suite *MethodTestSuite) TestLicense() {
 			return methodCase(inputs(), asserts(l, rbac.ActionRead))
 		})
 	})
+	suite.Run("InsertLicense", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			return methodCase(inputs(database.InsertLicenseParams{}), asserts(rbac.ResourceLicense, rbac.ActionCreate))
+		})
+	})
+	suite.Run("InsertOrUpdateLogoURL", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			return methodCase(inputs("value"), asserts(rbac.ResourceDeploymentConfig, rbac.ActionUpdate))
+		})
+	})
+	suite.Run("InsertOrUpdateServiceBanner", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			return methodCase(inputs("value"), asserts(rbac.ResourceDeploymentConfig, rbac.ActionUpdate))
+		})
+	})
+	suite.Run("GetLicenseByID", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			l, err := db.InsertLicense(context.Background(), database.InsertLicenseParams{
+				Uuid: uuid.NullUUID{UUID: uuid.New(), Valid: true},
+			})
+			require.NoError(t, err)
+			return methodCase(inputs(l.ID), asserts(l, rbac.ActionRead))
+		})
+	})
+	suite.Run("DeleteLicense", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			l, err := db.InsertLicense(context.Background(), database.InsertLicenseParams{
+				Uuid: uuid.NullUUID{UUID: uuid.New(), Valid: true},
+			})
+			require.NoError(t, err)
+			return methodCase(inputs(l.ID), asserts(l, rbac.ActionDelete))
+		})
+	})
+	suite.Run("GetDeploymentID", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			return methodCase(inputs(), asserts())
+		})
+	})
+	suite.Run("GetLogoURL", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			err := db.InsertOrUpdateLogoURL(context.Background(), "value")
+			require.NoError(t, err)
+			return methodCase(inputs(), asserts())
+		})
+	})
+	suite.Run("GetServiceBanner", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			err := db.InsertOrUpdateServiceBanner(context.Background(), "value")
+			require.NoError(t, err)
+			return methodCase(inputs(), asserts())
+		})
+	})
 }
diff --git a/coderd/authzquery/methods_test.go b/coderd/authzquery/methods_test.go
@@ -16,7 +16,7 @@ import (
 	"github.com/coder/coder/coderd/authzquery"
 	"github.com/coder/coder/coderd/coderdtest"
 	"github.com/coder/coder/coderd/database"
-	"github.com/coder/coder/coderd/database/databasefake"
+	"github.com/coder/coder/coderd/database/dbfake"
 	"github.com/coder/coder/coderd/rbac"
 )
 
@@ -91,7 +91,7 @@ func (s *MethodTestSuite) RunMethodTest(testCaseF func(t *testing.T, db database
 	methodName := names[len(names)-1]
 	s.methodAccounting[methodName]++
 
-	db := databasefake.New()
+	db := dbfake.New()
 	rec := &coderdtest.RecordingAuthorizer{
 		Wrapped: &coderdtest.FakeAuthorizer{
 			AlwaysReturn: nil,
