Fix failing tests from permission migration

packet-guardian · Mar 12, 2018 · f598a41 · f598a41
1 parent 11febbd
commit f598a41
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/common/schema.go b/src/common/schema.go
@@ -68,6 +68,9 @@ var (
 		"valid_start",
 		"valid_end",
 		"valid_forever",
+		"ui_group",
+		"api_group",
+		"allow_status_api",
 	}
 
 	SessionTableCols = []string{

diff --git a/src/server/routes_test.go b/src/server/routes_test.go
@@ -23,7 +23,7 @@ func TestRootHandlerLoggedInNormal(t *testing.T) {
 	defer db.Close()
 
 	userRow := sqlmock.NewRows(common.UserTableCols).AddRow(
-		1, "testUser", "", 0, 0, 0, 1, 1, 1, 0, 0,
+		1, "testUser", "", 0, 0, 0, 1, 1, 1, 0, 0, "default", "disable", false,
 	)
 
 	mock.ExpectQuery("SELECT .*? FROM \"user\"").WithArgs("testuser").WillReturnRows(userRow)
@@ -68,7 +68,7 @@ func TestRootHandlerLoggedInAdmin(t *testing.T) {
 	defer db.Close()
 
 	userRow := sqlmock.NewRows(common.UserTableCols).AddRow(
-		1, "testUser", "", 0, 0, 0, 1, 1, 1, 0, 0,
+		1, "testUser", "", 0, 0, 0, 1, 1, 1, 0, 0, "admin", "disable", false,
 	)
 
 	mock.ExpectQuery("SELECT .*? FROM \"user\"").WithArgs("testuser").WillReturnRows(userRow)