Example #1
0
    def test_api_keys_are_used_cyclically(self):
        queue = ApiQueue(["first key", "second key"])

        self.assertEqual("first key", queue.get())
        self.assertEqual("second key", queue.get())
        self.assertEqual("first key", queue.get())
        self.assertEqual("second key", queue.get())
Example #2
0
    def test_api_keys_can_be_added(self):
        queue = ApiQueue()
        queue.add("my api key")

        self.assertEqual("my api key", queue.get())
Example #3
0
    def test_can_be_initialized_with_api_keys(self):
        queue = ApiQueue(["first key", "second key"])

        self.assertEqual("first key", queue.get())
        self.assertEqual("second key", queue.get())