Skip Redis tests if no connection

engelke · engelke · commit a06a8be5d385 · 2020-08-03T14:51:31.000-07:00
diff --git a/appengine/standard/migration/memorystore/main_test.py b/appengine/standard/migration/memorystore/main_test.py
@@ -28,6 +28,11 @@
 
 @patch('main.query_for_data', return_value='data')
 def test_get_data_not_present(query_fn, testbed):
+    try:
+        main.client.set(KEY_PREFIX + 'counter', '0', 9000)
+    except Exception:
+        pytest.skip('Redis is unavailable')
+
     data = main.get_data(KEY_PREFIX + 'key')
     query_fn.assert_called_once_with()
     assert data == 'data'
@@ -37,14 +42,23 @@ def test_get_data_not_present(query_fn, testbed):
 
 @patch('main.query_for_data', return_value='data')
 def test_get_data_present(query_fn, testbed):
-    main.client.set(KEY_PREFIX + 'key', 'data', 9000)
+    try:
+        main.client.set(KEY_PREFIX + 'key', 'data', 9000)
+    except Exception:
+        pytest.skip('Redis is unavailable')
+
     data = main.get_data()
     query_fn.assert_not_called()
     assert data == 'data'
     main.client.delete(KEY_PREFIX + 'key')
 
 
 def test_add_values(testbed):
+    try:
+        main.client.set(KEY_PREFIX + 'counter', '0', 9000)
+    except Exception:
+        pytest.skip('Redis is unavailable')
+
     main.add_values(TEST_VALUES)
     for key, value in TEST_VALUES.iteritems():
         assert main.client.get(key) == value
