From 085d97fb2dd4f9c14e2658a6835d5c9a0324bd1e Mon Sep 17 00:00:00 2001
From: Mike McLean <mikem@redhat.com>
Date: Aug 15 2024 09:50:50 +0000
Subject: one more test


---

diff --git a/tests/test_hub/test_auto_arch_refuse.py b/tests/test_hub/test_auto_arch_refuse.py
index 96c6f3a..6e320f5 100644
--- a/tests/test_hub/test_auto_arch_refuse.py
+++ b/tests/test_hub/test_auto_arch_refuse.py
@@ -70,6 +70,22 @@ class AutoRefuseTest(unittest.TestCase):
         self.get_tag.assert_called_once_with('TAG_ID')
         self.set_refusal.assert_called_once()
 
+    def test_mixed_hosts(self):
+        good1 = [{'id': n, 'arches': 'x86_64 i686'} for n in range(0,5)]
+        bad1 = [{'id': n, 'arches': 'ia64'} for n in range(5,10)]
+        good2 = [{'id': n, 'arches': 'aarch64'} for n in range(10,15)]
+        bad2 = [{'id': n, 'arches': 'sparc64'} for n in range(15,20)]
+        hosts = good1 + bad1 + good2 + bad2
+        self.context.handlers.call.return_value = hosts
+        scheduler._auto_arch_refuse(100)
+
+        self.Task.assert_called_once_with(100)
+        self.get_tag.assert_called_once_with('TAG_ID')
+
+        # should only refuse the bad ones
+        expect = [mock.call(h['id'], 100, soft=False, msg='automatic arch refusal') for h in bad1 + bad2]
+        self.assertListEqual(self.set_refusal.mock_calls, expect)
+
     def test_not_noarch(self):
         self.taskinfo['arch'] = 'x86_64'