✨ Add assurances

Author: siguici

spec/vunit_test.v

@@ -1,17 +1,13 @@
-import vunit { assert_bool, assert_false, assert_nil, assert_true, expect }
+import vunit { ensure, expect }
 
-fn test_assert_nil() {
-	assert_nil(unsafe { nil })
+fn test_nil_unit() {
+	ensure(unsafe { nil }).is_nil()
+	expect(unsafe { nil }).to_be_nil()
 }
 
-fn test_assert_bool() {
-	assert_bool(true)
-	assert_bool(false)
-	assert_true(true)
-	assert_false(false)
-}
-
-fn test_expect_bool() {
+fn test_bool_unit() {
+	ensure(true).is_bool().is_true()
+	ensure(false).is_bool().is_false()
 	expect(true).to_be_bool().to_be_true()
 	expect(false).to_be_bool().to_be_false()
 }

vunit/assertions.v

@@ -14,7 +14,7 @@ fn assert_type[T](assertion bool, val T, typ string) {
 	assert_that(assertion, typ, typeof(val).name)
 }
 
-pub fn assert_nil(val voidptr) {
+pub fn assert_nil[T](val T) {
 	assert_type(filter(val).is_nil(), val, 'nil')
 }
 

vunit/assurances.v

@@ -0,0 +1,39 @@
+module vunit
+
+import vest { filter }
+
+struct Assurance[T] {
+	val T
+}
+
+pub fn ensure[T](val T) Assurance[T] {
+	return Assurance[T]{val}
+}
+
+pub fn (this Assurance[T]) is_nil() Assurance[T] {
+	if !filter(this.val).is_nil() {
+		panic('${this.val} is not nil')
+	}
+	return this
+}
+
+pub fn (this Assurance[T]) is_bool() Assurance[T] {
+	if !filter(this.val).is_bool() {
+		panic('${this.val} is not bool')
+	}
+	return this
+}
+
+pub fn (this Assurance[bool]) is_true() Assurance[bool] {
+	if !filter(this.val).is_true() {
+		panic('${this.val} is not true')
+	}
+	return this
+}
+
+pub fn (this Assurance[bool]) is_false() Assurance[bool] {
+	if !filter(this.val).is_false() {
+		panic('${this.val} is not false')
+	}
+	return this
+}

vunit/expectations.v

@@ -1,6 +1,5 @@
 module vunit
 
-@[params]
 struct Expectation[T] {
 	val T
 }
@@ -9,7 +8,7 @@ pub fn expect[T](val T) Expectation[T] {
 	return Expectation[T]{val}
 }
 
-pub fn (this Expectation[voidptr]) to_be_nil() Expectation[voidptr] {
+pub fn (this Expectation[T]) to_be_nil() Expectation[T] {
 	assert_nil(this.val)
 	return this
 }