From 4041054a3cb358632cbf7cb3803a16894932d245 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Zbigniew=20J=C4=99drzejewski-Szmek?= <zbyszek@in.waw.pl>
Date: Fri, 21 Sep 2018 10:23:39 +0200
Subject: [PATCH] Fix failing tests in test_api.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Based on commit 0722e2edc93ae818bddd18f23be5b18c3a2a1d55 by
Kale Franz <kfranz@continuum.io>.

Signed-off-by: Zbigniew Jędrzejewski-Szmek <zbyszek@in.waw.pl>
---
 tests/test_api.py | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/tests/test_api.py b/tests/test_api.py
index f9ef665d7d..c555ef0229 100644
--- a/tests/test_api.py
+++ b/tests/test_api.py
@@ -196,9 +196,10 @@ def test_PackageCacheData_contract():
 def test_PackageCacheData_return_value_contract():
     pc = PackageCacheData(context.pkgs_dirs[0])
 
-    single_pcrec = next(pc.iter_records())
-    get_result = pc.get(PackageRef.from_objects(single_pcrec))
-    assert isinstance(get_result, PackageCacheRecord)
+    single_pcrec = next(pc.iter_records(), None)
+    if single_pcrec:
+        get_result = pc.get(PackageRef.from_objects(single_pcrec))
+        assert isinstance(get_result, PackageCacheRecord)
 
     query_result = pc.query('openssl')
     assert isinstance(query_result, tuple)
@@ -258,20 +259,21 @@ def test_PrefixData_contract():
 def test_PrefixData_return_value_contract():
     pd = PrefixData(context.conda_prefix)
 
-    single_prefix_rec = next(pd.iter_records())
-    get_result = pd.get(PackageRef.from_objects(single_prefix_rec))
-    assert isinstance(get_result, PrefixRecord)
+    single_prefix_rec = next(pd.iter_records(), None)
+    if single_prefix_rec:
+        get_result = pd.get(PackageRef.from_objects(single_prefix_rec))
+        assert isinstance(get_result, PrefixRecord)
 
-    query_result = pd.query('openssl')
-    assert isinstance(query_result, tuple)
-    assert all(isinstance(prefix_rec, PrefixRecord) for prefix_rec in query_result)
+        query_result = pd.query('openssl')
+        assert isinstance(query_result, tuple)
+        assert all(isinstance(prefix_rec, PrefixRecord) for prefix_rec in query_result)
 
     iter_records_result = pd.iter_records()
     assert isiterable(iter_records_result)
     assert all(isinstance(prefix_rec, PrefixRecord) for prefix_rec in iter_records_result)
 
     is_writable_result = pd.is_writable
-    assert is_writable_result is True or is_writable_result is False
+    assert is_writable_result in {True, False, None}
 
     reload_result = pd.reload()
     assert isinstance(reload_result, PrefixData)
-- 
2.19.0.rc1