From 73e9ce20f2d45f14975933d5772ef1a4ec18c2fd Mon Sep 17 00:00:00 2001
From: Carsten Otto <git@c-otto.de>
Date: Sun, 27 Mar 2022 00:10:35 +0100
Subject: [PATCH] add convenience method

---
 model/src/main/java/de/cotto/lndmanagej/model/Policy.java     | 4 ++++
 model/src/test/java/de/cotto/lndmanagej/model/PolicyTest.java | 2 ++
 .../cotto/lndmanagej/pickhardtpayments/FlowComputation.java   | 2 +-
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/model/src/main/java/de/cotto/lndmanagej/model/Policy.java b/model/src/main/java/de/cotto/lndmanagej/model/Policy.java
index 6d9be845..dde1ce76 100644
--- a/model/src/main/java/de/cotto/lndmanagej/model/Policy.java
+++ b/model/src/main/java/de/cotto/lndmanagej/model/Policy.java
@@ -2,4 +2,8 @@ package de.cotto.lndmanagej.model;
 
 public record Policy(long feeRate, Coins baseFee, boolean enabled) {
     public static final Policy UNKNOWN = new Policy(0, Coins.NONE, false);
+
+    public boolean disabled() {
+        return !enabled;
+    }
 }
diff --git a/model/src/test/java/de/cotto/lndmanagej/model/PolicyTest.java b/model/src/test/java/de/cotto/lndmanagej/model/PolicyTest.java
index 23da20c1..f79247aa 100644
--- a/model/src/test/java/de/cotto/lndmanagej/model/PolicyTest.java
+++ b/model/src/test/java/de/cotto/lndmanagej/model/PolicyTest.java
@@ -20,11 +20,13 @@ class PolicyTest {
     @Test
     void enabled() {
         assertThat(POLICY_1.enabled()).isTrue();
+        assertThat(POLICY_1.disabled()).isFalse();
     }
 
     @Test
     void disabled() {
         assertThat(POLICY_DISABLED.enabled()).isFalse();
+        assertThat(POLICY_DISABLED.disabled()).isTrue();
     }
 
     @Test
diff --git a/pickhardt-payments/src/main/java/de/cotto/lndmanagej/pickhardtpayments/FlowComputation.java b/pickhardt-payments/src/main/java/de/cotto/lndmanagej/pickhardtpayments/FlowComputation.java
index 13f3b03e..4cb1c08f 100644
--- a/pickhardt-payments/src/main/java/de/cotto/lndmanagej/pickhardtpayments/FlowComputation.java
+++ b/pickhardt-payments/src/main/java/de/cotto/lndmanagej/pickhardtpayments/FlowComputation.java
@@ -71,7 +71,7 @@ public class FlowComputation {
         Set<EdgeWithLiquidityInformation> edgesWithLiquidityInformation = new LinkedHashSet<>();
         Pubkey ownPubkey = grpcGetInfo.getPubkey();
         for (DirectedChannelEdge channelEdge : channelEdges) {
-            if (!channelEdge.policy().enabled()) {
+            if (channelEdge.policy().disabled()) {
                 continue;
             }
             ChannelId channelId = channelEdge.channelId();