@@ -23,6 +23,7 @@ import (
23
23
"net/http"
24
24
"os"
25
25
"strconv"
26
+ "time"
26
27
27
28
"github.com/google/exposure-notifications-server/pkg/logging"
28
29
"github.com/google/exposure-notifications-server/pkg/observability"
@@ -110,7 +111,7 @@ func realMain(ctx context.Context) error {
110
111
}
111
112
realm = database .NewRealmWithDefaults (realmName )
112
113
realm .RegionCode = realmRegionCode
113
- if err := db .SaveRealm (realm ); err != nil {
114
+ if err := db .SaveRealm (realm , database . System ); err != nil {
114
115
return fmt .Errorf ("failed to create realm %+v: %w: %v" , realm , err , realm .ErrorMessages ())
115
116
}
116
117
}
@@ -124,7 +125,7 @@ func realMain(ctx context.Context) error {
124
125
adminKey , err := realm .CreateAuthorizedApp (db , & database.AuthorizedApp {
125
126
Name : adminKeyName + suffix ,
126
127
APIKeyType : database .APIUserTypeAdmin ,
127
- })
128
+ }, database . System )
128
129
if err != nil {
129
130
return fmt .Errorf ("error trying to create a new Admin API Key: %w" , err )
130
131
}
@@ -134,7 +135,9 @@ func realMain(ctx context.Context) error {
134
135
if err != nil {
135
136
logger .Errorf ("admin API key cleanup failed: %w" , err )
136
137
}
137
- if err := app .Disable (db ); err != nil {
138
+ now := time .Now ().UTC ()
139
+ app .DeletedAt = & now
140
+ if err := db .SaveAuthorizedApp (app , database .System ); err != nil {
138
141
logger .Errorf ("admin API key disable failed: %w" , err )
139
142
}
140
143
logger .Info ("successfully cleaned up e2e test admin key" )
@@ -143,7 +146,7 @@ func realMain(ctx context.Context) error {
143
146
deviceKey , err := realm .CreateAuthorizedApp (db , & database.AuthorizedApp {
144
147
Name : deviceKeyName + suffix ,
145
148
APIKeyType : database .APIUserTypeDevice ,
146
- })
149
+ }, database . System )
147
150
if err != nil {
148
151
return fmt .Errorf ("error trying to create a new Device API Key: %w" , err )
149
152
}
@@ -153,7 +156,9 @@ func realMain(ctx context.Context) error {
153
156
if err != nil {
154
157
logger .Errorf ("device API key cleanup failed: %w" , err )
155
158
}
156
- if err := app .Disable (db ); err != nil {
159
+ now := time .Now ().UTC ()
160
+ app .DeletedAt = & now
161
+ if err := db .SaveAuthorizedApp (app , database .System ); err != nil {
157
162
logger .Errorf ("device API key disable failed: %w" , err )
158
163
}
159
164
logger .Info ("successfully cleaned up e2e test device key" )
0 commit comments