test: pre set data for under attack

teknologi-umum · Dec 25, 2023 · c0b1546 · c0b1546
1 parent 86e0561
commit c0b1546
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/underattack/are_we_test.go b/underattack/are_we_test.go
@@ -10,6 +10,11 @@ func TestAreWe(t *testing.T) {
 	ctx, cancel := context.WithTimeout(context.Background(), time.Second*30)
 	defer cancel()
 
+	err := dependency.Datastore.SetUnderAttackStatus(ctx, 1, true, time.Now().Add(time.Hour), 0)
+	if err != nil {
+		t.Fatalf("setting under attack status: %s", err.Error())
+	}
+
 	attacked, err := dependency.AreWe(ctx, 1)
 	if err != nil {
 		t.Errorf("unexpected error: %v", err)