8000 update db mock · coder/coder@55a7c4d · GitHub
[go: up one dir, main page]
Skip to content

Commit 55a7c4d

Browse files
EmyrkEmyrk
committed
update db mock
1 parent 69d0323 commit 55a7c4d

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

‎coderd/wsbuilder/wsbuilder_test.go

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ func TestBuilder_NoOptions(t *testing.T) {
5757
mDB := expectDB(t,
5858
// Inputs
5959
withTemplate,
60+
withTerraformValuesErrNoRows,
6061
withInactiveVersion(nil),
6162
withLastBuildFound,
6263
withTemplateVersionVariables(inactiveVersionID, nil),
@@ -113,6 +114,7 @@ func TestBuilder_Initiator(t *testing.T) {
113114
mDB := expectDB(t,
114115
// Inputs
115116
withTemplate,
117+
withTerraformValuesErrNoRows,
116118
withInactiveVersion(nil),
117119
withLastBuildFound,
118120
withTemplateVersionVariables(inactiveVersionID, nil),
@@ -159,6 +161,7 @@ func TestBuilder_Baggage(t *testing.T) {
159161
mDB := expectDB(t,
160162
// Inputs
161163
withTemplate,
164+
withTerraformValuesErrNoRows,
162165
withInactiveVersion(nil),
163166
withLastBuildFound,
164167
withTemplateVersionVariables(inactiveVersionID, nil),
@@ -197,6 +200,7 @@ func TestBuilder_Reason(t *testing.T) {
197200
mDB := expectDB(t,
198201
// Inputs
199202
withTemplate,
203+
withTerraformValuesErrNoRows,
200204
withInactiveVersion(nil),
201205
withLastBuildFound,
202206
withTemplateVersionVariables(inactiveVersionID, nil),
@@ -235,6 +239,7 @@ func TestBuilder_ActiveVersion(t *testing.T) {
235239
mDB := expectDB(t,
236240
// Inputs
237241
withTemplate,
242+
withTerraformValuesErrNoRows,
238243
withActiveVersion(nil),
239244
withLastBuildNotFound,
240245
withTemplateVersionVariables(activeVersionID, nil),
@@ -338,6 +343,7 @@ func TestWorkspaceBuildWithTags(t *testing.T) {
338343
mDB := expectDB(t,
339344
// Inputs
340345
withTemplate,
346+
withTerraformValuesErrNoRows,
341347
withInactiveVersion(richParameters),
342348
withLastBuildFound,
343349
withTemplateVersionVariables(inactiveVersionID, templateVersionVariables),
@@ -433,6 +439,7 @@ func TestWorkspaceBuildWithRichParameters(t *testing.T) {
433439
mDB := expectDB(t,
434440
// Inputs
435441
withTemplate,
442+
withTerraformValuesErrNoRows,
436443
withInactiveVersion(richParameters),
437444
withLastBuildFound,
438445
withTemplateVersionVariables(inactiveVersionID, nil),
@@ -480,6 +487,7 @@ func TestWorkspaceBuildWithRichParameters(t *testing.T) {
480487
mDB := expectDB(t,
481488
// Inputs
482489
withTemplate,
490+
withTerraformValuesErrNoRows,
483491
withInactiveVersion(richParameters),
484492
withLastBuildFound,
485493
withTemplateVersionVariables(inactiveVersionID, nil),
@@ -533,6 +541,7 @@ func TestWorkspaceBuildWithRichParameters(t *testing.T) {
533541
mDB := expectDB(t,
534542
// Inputs
535543
withTemplate,
544+
withTerraformValuesErrNoRows,
536545
withInactiveVersion(richParameters),
537546
withLastBuildFound,
538547
withTemplateVersionVariables(inactiveVersionID, nil),
@@ -565,6 +574,7 @@ func TestWorkspaceBuildWithRichParameters(t *testing.T) {
565574
mDB := expectDB(t,
566575
// Inputs
567576
withTemplate,
577+
withTerraformValuesErrNoRows,
568578
withInactiveVersion(richParameters),
569579
withLastBuildFound,
570580
withTemplateVersionVariables(inactiveVersionID, nil),
@@ -617,6 +627,7 @@ func TestWorkspaceBuildWithRichParameters(t *testing.T) {
617627
mDB := expectDB(t,
618628
// Inputs
619629
withTemplate,
630+
withTerraformValuesErrNoRows,
620631
withActiveVersion(version2params),
621632
withLastBuildFound,
622633
withTemplateVersionVariables(activeVersionID, nil),
@@ -680,6 +691,7 @@ func TestWorkspaceBuildWithRichParameters(t *testing.T) {
680691
mDB := expectDB(t,
681692
// Inputs
682693
withTemplate,
694+
withTerraformValuesErrNoRows,
683695
withActiveVersion(version2params),
684696
withLastBuildFound,
685697
withTemplateVersionVariables(activeVersionID, nil),
@@ -741,6 +753,7 @@ func TestWorkspaceBuildWithRichParameters(t *testing.T) {
741753
mDB := expectDB(t,
742754
// Inputs
743755
withTemplate,
756+
withTerraformValuesErrNoRows,
744757
withActiveVersion(version2params),
745758
withLastBuildFound,
746759
withTemplateVersionVariables(activeVersionID, nil),
@@ -788,6 +801,7 @@ func TestWorkspaceBuildWithPreset(t *testing.T) {
788801
mDB := expectDB(t,
789802
// Inputs
790803
withTemplate,
804+
withTerraformValuesErrNoRows,
791805
withActiveVersion(nil),
792806
// building workspaces using presets with different combinations of parameters
793807
// is tested at the API layer, in TestWorkspace. Here, it is sufficient to
@@ -902,6 +916,12 @@ func withTemplate(mTx *dbmock.MockStore) {
902916
}, nil)
903917
}
904918

919+
func withTerraformValuesErrNoRows(mTx *dbmock.MockStore) {
920+
mTx.EXPECT().GetTemplateVersionTerraformValues(gomock.Any(), gomock.Any()).
921+
Times(1).
922+
Return(database.TemplateVersionTerraformValue{}, sql.ErrNoRows)
923+
}
924+
905925
// withInTx runs the given functions on the same db mock.
906926
func withInTx(mTx *dbmock.MockStore) {
907927
mTx.EXPECT().InTx(gomock.Any(), gomock.Any()).Times(1).DoAndReturn(

0 commit comments

Comments
 (0)
0