99import java .util .ArrayList ;
1010import java .util .Iterator ;
1111import java .util .List ;
12+ import java .util .concurrent .CountDownLatch ;
13+ import java .util .concurrent .TimeUnit ;
1214
13- import static junit .framework .TestCase .assertEquals ;
15+ import static junit .framework .TestCase .* ;
1416
1517
1618public class QueryTestCase {
@@ -121,54 +123,88 @@ public void onCompletion(ResponseType responseType, QueryResult queryresult, Err
121123
122124 @ Test
123125 public void test_42_variants_single_uid_find () throws Exception {
126+ Assume .assumeFalse ("variantUID not configured" , variantUID == null || variantUID .trim ().isEmpty ());
124127 final Query q = TestCred .stack ().contentType (contentTypeUID ).query ();
125- q .variants (variantUID ).find (new QueryResultsCallBack () {
128+ q .variants (variantUID );
129+ assertEquals (variantUID .trim (), q .getHeaders ().get ("x-cs-variant-uid" ));
130+ assertNull (q .getHeaders ().get ("branch" ));
131+
132+ final CountDownLatch latch = new CountDownLatch (1 );
133+ q .find (new QueryResultsCallBack () {
126134 @ Override
127135 public void onCompletion (ResponseType responseType , QueryResult queryresult , Error error ) {
128136 if (error == null ) {
129137 Log .d (TAG , "variants single find: " + queryresult .getResultObjects ().size () + " entries" );
130138 }
139+ latch .countDown ();
131140 }
132141 });
142+ assertTrue ("find() callback timed out" , latch .await (30 , TimeUnit .SECONDS ));
133143 }
134144
135145 @ Test
136146 public void test_43_variants_array_find () throws Exception {
147+ Assume .assumeFalse ("variantsUID not configured" , variantsUID == null || variantsUID .length == 0 );
137148 final Query q = TestCred .stack ().contentType (contentTypeUID ).query ();
138- q .variants (variantsUID ).find (new QueryResultsCallBack () {
149+ q .variants (variantsUID );
150+ assertNotNull (q .getHeaders ().get ("x-cs-variant-uid" ));
151+ assertNull (q .getHeaders ().get ("branch" ));
152+
153+ final CountDownLatch latch = new CountDownLatch (1 );
154+ q .find (new QueryResultsCallBack () {
139155 @ Override
140156 public void onCompletion (ResponseType responseType , QueryResult queryresult , Error error ) {
141157 if (error == null ) {
142158 Log .d (TAG , "variants array find: " + queryresult .getResultObjects ().size () + " entries" );
143159 }
160+ latch .countDown ();
144161 }
145162 });
163+ assertTrue ("find() callback timed out" , latch .await (30 , TimeUnit .SECONDS ));
146164 }
147165
148166 @ Test
149167 public void test_44_variants_single_uid_with_branch_find () throws Exception {
168+ Assume .assumeFalse ("variantUID not configured" , variantUID == null || variantUID .trim ().isEmpty ());
169+ Assume .assumeFalse ("variantBranch not configured" , variantBranch == null || variantBranch .trim ().isEmpty ());
150170 final Query q = TestCred .stack ().contentType (contentTypeUID ).query ();
151- q .variants (variantUID , variantBranch ).find (new QueryResultsCallBack () {
171+ q .variants (variantUID , variantBranch );
172+ assertEquals (variantUID .trim (), q .getHeaders ().get ("x-cs-variant-uid" ));
173+ assertEquals (variantBranch .trim (), q .getHeaders ().get ("branch" ));
174+
175+ final CountDownLatch latch = new CountDownLatch (1 );
176+ q .find (new QueryResultsCallBack () {
152177 @ Override
153178 public void onCompletion (ResponseType responseType , QueryResult queryresult , Error error ) {
154179 if (error == null ) {
155180 Log .d (TAG , "variants single+branch find: " + queryresult .getResultObjects ().size () + " entries" );
156181 }
182+ latch .countDown ();
157183 }
158184 });
185+ assertTrue ("find() callback timed out" , latch .await (30 , TimeUnit .SECONDS ));
159186 }
160187
161188 @ Test
162189 public void test_45_variants_array_with_branch_find () throws Exception {
190+ Assume .assumeFalse ("variantsUID not configured" , variantsUID == null || variantsUID .length == 0 );
191+ Assume .assumeFalse ("variantBranch not configured" , variantBranch == null || variantBranch .trim ().isEmpty ());
163192 final Query q = TestCred .stack ().contentType (contentTypeUID ).query ();
164- q .variants (variantsUID , variantBranch ).find (new QueryResultsCallBack () {
193+ q .variants (variantsUID , variantBranch );
194+ assertNotNull (q .getHeaders ().get ("x-cs-variant-uid" ));
195+ assertEquals (variantBranch .trim (), q .getHeaders ().get ("branch" ));
196+
197+ final CountDownLatch latch = new CountDownLatch (1 );
198+ q .find (new QueryResultsCallBack () {
165199 @ Override
166200 public void onCompletion (ResponseType responseType , QueryResult queryresult , Error error ) {
167201 if (error == null ) {
168202 Log .d (TAG , "variants array+branch find: " + queryresult .getResultObjects ().size () + " entries" );
169203 }
204+ latch .countDown ();
170205 }
171206 });
207+ assertTrue ("find() callback timed out" , latch .await (30 , TimeUnit .SECONDS ));
172208 }
173209
174210}
0 commit comments