Skip to content

Commit 3d19861

Browse files
committed
util/quota/gpu: extract gpu on-prem upgrade similar to patches, but only one of them counts + testing + fixing tests
1 parent dc6684b commit 3d19861

File tree

4 files changed

+238
-78
lines changed

4 files changed

+238
-78
lines changed

src/packages/database/postgres/site-license/hook.test.ts

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -70,13 +70,14 @@ test("allow for much larger max_upgrades", () => {
7070
{},
7171
{ userX: {} },
7272
{ a: site_licenses.a },
73-
site_settings
73+
site_settings,
7474
);
7575
expect(q1).toEqual({
7676
quota: {
7777
network: true,
7878
member_host: false,
7979
privileged: false,
80+
gpu: false,
8081
memory_request: 1000,
8182
cpu_request: 0.1,
8283
disk_quota: 3000,
@@ -94,7 +95,7 @@ test("allow for much larger max_upgrades", () => {
9495
{},
9596
{ userX: {} },
9697
site_licenses,
97-
site_settings
98+
site_settings,
9899
);
99100
expect(q2).toEqual({
100101
quota: {
@@ -110,6 +111,7 @@ test("allow for much larger max_upgrades", () => {
110111
memory_request: 1000,
111112
network: true,
112113
privileged: false,
114+
gpu: false,
113115
},
114116
reasons: {},
115117
});
@@ -155,13 +157,14 @@ test("two licenses", () => {
155157
{},
156158
{ userX: {} },
157159
{ a: site_licenses.a },
158-
site_settings
160+
site_settings,
159161
);
160162
expect(q1).toEqual({
161163
quota: {
162164
network: true,
163165
member_host: false,
164166
privileged: false,
167+
gpu: false,
165168
memory_request: 1000,
166169
cpu_request: 0.05,
167170
disk_quota: 3000,
@@ -179,13 +182,14 @@ test("two licenses", () => {
179182
{},
180183
{ userX: {} },
181184
site_licenses,
182-
site_settings
185+
site_settings,
183186
);
184187
expect(q1).toEqual({
185188
quota: {
186189
network: true,
187190
member_host: false,
188191
privileged: false,
192+
gpu: false,
189193
memory_request: 1000,
190194
cpu_request: 0.05,
191195
disk_quota: 3000,
@@ -213,6 +217,7 @@ test("two licenses", () => {
213217
memory_request: 1000,
214218
network: true,
215219
privileged: false,
220+
gpu: false,
216221
},
217222
reasons: {},
218223
});
@@ -311,7 +316,7 @@ test("two licenses", () => {
311316
// memory_limit: 1000,
312317
// memory_request: 200,
313318
// network: false,
314-
// privileged: false,
319+
// privileged: false,gpu:false,
315320
// },
316321
// reasons: {},
317322
// });

0 commit comments

Comments
 (0)