✅ [client] Add virtual channel dispute test

Signed-off-by: Matthias Geihs <[email protected]>

Author: matthiasgeihs

client/virtual_channel_test.go

@@ -68,6 +68,56 @@ func TestVirtualChannelsOptimistic(t *testing.T) {
 	assert.NoError(t, err, "Bob: invalid final balances")
 }
 
+func TestVirtualChannelsDispute(t *testing.T) {
+	ctx, cancel := context.WithTimeout(context.Background(), testDuration)
+	defer cancel()
+
+	vct := setupVirtualChannelTest(t, ctx)
+	var err error
+	assert := assert.New(t)
+
+	// Alice registers and settles channel with Ingrid.
+	err = vct.registerAndSettle(ctx, vct.chAliceIngrid)
+	assert.NoError(err, "alice-ingrid: register and settle")
+
+	// Ingrid registers and settles channel with Alice.
+	err = vct.registerAndSettle(ctx, vct.chIngridAlice)
+	assert.NoError(err, "ingrid-alice: register and settle")
+
+	// Bob registers and settles channel with Ingrid.
+	err = vct.registerAndSettle(ctx, vct.chBobIngrid)
+	assert.NoError(err, "bob-ingrid: register and settle")
+
+	// Ingrid registers and settles channel with Bob.
+	err = vct.registerAndSettle(ctx, vct.chIngridBob)
+	assert.NoError(err, "ingrid-bob: register and settle")
+}
+
+func TestVirtualChannelsDisputeHubFirst(t *testing.T) {
+	ctx, cancel := context.WithTimeout(context.Background(), testDuration)
+	defer cancel()
+
+	vct := setupVirtualChannelTest(t, ctx)
+	var err error
+	assert := assert.New(t)
+
+	// Ingrid registers and settles channel with Alice.
+	err = vct.registerAndSettle(ctx, vct.chIngridAlice)
+	assert.NoError(err, "ingrid-alice: register and settle")
+
+	// Alice registers and settles channel with Ingrid.
+	err = vct.registerAndSettle(ctx, vct.chAliceIngrid)
+	assert.NoError(err, "alice-ingrid: register and settle")
+
+	// Ingrid registers and settles channel with Bob.
+	err = vct.registerAndSettle(ctx, vct.chIngridBob)
+	assert.NoError(err, "ingrid-bob: register and settle")
+
+	// Bob registers and settles channel with Ingrid.
+	err = vct.registerAndSettle(ctx, vct.chBobIngrid)
+	assert.NoError(err, "bob-ingrid: register and settle")
+}
+
 type virtualChannelTest struct {
 	chAliceIngrid      *client.Channel
 	chIngridAlice      *client.Channel
@@ -228,3 +278,13 @@ func setupVirtualChannelTest(t *testing.T, ctx context.Context) (vct virtualChan
 	require.NoError(err, "updating virtual channel")
 	return
 }
+
+func (vct virtualChannelTest) registerAndSettle(ctx context.Context, ch *client.Channel) (err error) {
+	err = ch.Register(ctx)
+	if err != nil {
+		return
+	}
+
+	err = ch.Settle(ctx, false)
+	return
+}