From 169f680b58542cbc37032dac03f191419dcd1507 Mon Sep 17 00:00:00 2001
From: Paul Schaub <vanitasvitae@fsfe.org>
Date: Fri, 30 Jun 2023 18:07:25 +0200
Subject: [PATCH] Test RevocationState class

---
 .../wot/dijkstra/RevocationStateTest.kt       | 43 +++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 wot-dijkstra/src/test/kotlin/org/pgpainless/wot/dijkstra/RevocationStateTest.kt

diff --git a/wot-dijkstra/src/test/kotlin/org/pgpainless/wot/dijkstra/RevocationStateTest.kt b/wot-dijkstra/src/test/kotlin/org/pgpainless/wot/dijkstra/RevocationStateTest.kt
new file mode 100644
index 00000000..4aaf96d0
--- /dev/null
+++ b/wot-dijkstra/src/test/kotlin/org/pgpainless/wot/dijkstra/RevocationStateTest.kt
@@ -0,0 +1,43 @@
+package org.pgpainless.wot.dijkstra
+
+import org.junit.jupiter.api.Test
+import org.pgpainless.wot.dijkstra.sq.ReferenceTime
+import org.pgpainless.wot.dijkstra.sq.RevocationState
+import java.util.*
+import kotlin.test.assertFalse
+import kotlin.test.assertTrue
+
+class RevocationStateTest {
+
+    @Test
+    fun testNotRevoked() {
+        val notRevoked = RevocationState.notRevoked()
+        assertTrue { notRevoked.isNotRevoked() }
+        assertFalse { notRevoked.isSoftRevocation() }
+        assertFalse { notRevoked.isHardRevocation() }
+        assertFalse { notRevoked.isEffective(ReferenceTime.timestamp(Date())) }
+    }
+
+    @Test
+    fun testSoftRevocation() {
+        val timestamp = Date()
+        val before = Date(timestamp.time - 5000)
+        val after = Date(timestamp.time + 5000)
+        val softRevoked = RevocationState.softRevoked(timestamp)
+
+        assertTrue { softRevoked.isSoftRevocation() }
+        assertFalse { softRevoked.isHardRevocation() }
+        assertFalse { softRevoked.isNotRevoked() }
+        assertTrue { softRevoked.isEffective(ReferenceTime.timestamp(after)) }
+        assertFalse { softRevoked.isEffective(ReferenceTime.timestamp(before)) }
+    }
+
+    @Test
+    fun testHardRevoked() {
+        val hardRevoked = RevocationState.hardRevoked()
+        assertTrue { hardRevoked.isHardRevocation() }
+        assertFalse { hardRevoked.isSoftRevocation() }
+        assertFalse { hardRevoked.isNotRevoked() }
+        assertTrue { hardRevoked.isEffective(ReferenceTime.now()) }
+    }
+}
\ No newline at end of file