fix:add tests

microsoft · Dec 18, 2024 · 44528fa · 44528fa
1 parent 0c07bc7
commit 44528fa
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/internal/bcrypt/ntstatus_test.go b/internal/bcrypt/ntstatus_test.go
@@ -5,6 +5,13 @@ import "testing"
 func TestNTStatusErrMessage(t *testing.T) {
 
 	StatusWait1 := NTStatus(0x00000001)
+	StatusWait2 := NTStatus(0x00000002)
 
-	t.Error(StatusWait1.Error())
+	if StatusWait1.Error() != "STATUS_WAIT_1" {
+		t.Errorf("Expected STATUS_WAIT_2, got %s", StatusWait2.Error())
+	}
+
+	if StatusWait2.Error() != "STATUS_WAIT_2" {
+		t.Errorf("Expected STATUS_WAIT_2, got %s", StatusWait2.Error())
+	}
 }