Merge pull request #817 from pyathena-dev/fix/816-cache-test-work-group

Assert cache hits in the test work group instead of primary

Author: laughingman7743

docs/usage.md

@@ -221,6 +221,8 @@ See the [Athena documentation](https://docs.aws.amazon.com/athena/latest/ug/reus
 
 You can attempt to re-use the results from a previously executed query to help save time and money in the cases where your underlying data isn't changing.
 Set the `cache_size` or `cache_expiration_time` parameter of `cursor.execute()` to a number larger than 0 to enable caching.
+`cache_size` is the number of the most recent query executions in the work group to search, including executions by other clients of the same work group.
+In a busy work group, a previous execution may no longer be among them.
 
 ```python
 from pyathena import connect

tests/pyathena/test_cursor.py

@@ -93,6 +93,9 @@ def test_iterator(self, cursor):
         assert list(cursor) == [(1,)]
         pytest.raises(StopIteration, cursor.__next__)
 
+    # Cache hits are asserted in ENV.work_group: the default work group runs most test
+    # queries, which can push earlier executions out of the cache_size window.
+    @pytest.mark.parametrize("cursor", [{"work_group": ENV.work_group}], indirect=["cursor"])
     def test_cache_size(self, cursor):
         # To test caching, we need to make sure the query is unique, otherwise
         # we might accidentally pick up the cache results from another CI run.
@@ -143,6 +146,7 @@ def test_cache_expiration_time(self, cursor):
         assert query_id_1 != query_id_2
         assert query_id_3 in [query_id_1, query_id_2]
 
+    @pytest.mark.parametrize("cursor", [{"work_group": ENV.work_group}], indirect=["cursor"])
     def test_cache_expiration_time_with_cache_size(self, cursor):
         # Cache miss
         query = f"SELECT * FROM one_row -- {datetime.now(timezone.utc)!s}"