From 87789676c0d9a2d7bfe3718ff4a9c002b6816bdc Mon Sep 17 00:00:00 2001 From: Shivaram Lingamneni Date: Thu, 5 May 2022 22:43:33 -0400 Subject: [PATCH] add tests covering (*AcceptManager).Unaccept --- irc/accept_test.go | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/irc/accept_test.go b/irc/accept_test.go index bf2e1022..fad39151 100644 --- a/irc/accept_test.go +++ b/irc/accept_test.go @@ -12,6 +12,9 @@ func TestAccept(t *testing.T) { bob := new(Client) eve := new(Client) + // must not panic: + am.Unaccept(eve, bob) + assertEqual(am.MaySendTo(alice, bob), false) assertEqual(am.MaySendTo(bob, alice), false) assertEqual(am.MaySendTo(alice, eve), false) @@ -46,6 +49,24 @@ func TestAccept(t *testing.T) { assertEqual(am.MaySendTo(bob, eve), false) assertEqual(am.MaySendTo(eve, bob), true) + am.Accept(eve, bob) + + assertEqual(am.MaySendTo(alice, bob), true) + assertEqual(am.MaySendTo(bob, alice), true) + assertEqual(am.MaySendTo(alice, eve), false) + assertEqual(am.MaySendTo(eve, alice), false) + assertEqual(am.MaySendTo(bob, eve), true) + assertEqual(am.MaySendTo(eve, bob), true) + + am.Unaccept(eve, bob) + + assertEqual(am.MaySendTo(alice, bob), true) + assertEqual(am.MaySendTo(bob, alice), true) + assertEqual(am.MaySendTo(alice, eve), false) + assertEqual(am.MaySendTo(eve, alice), false) + assertEqual(am.MaySendTo(bob, eve), false) + assertEqual(am.MaySendTo(eve, bob), true) + am.Remove(alice) assertEqual(am.MaySendTo(alice, bob), false)