/*
 *    Copyright 2012 Jason Miller
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package jj.script;

import static org.hamcrest.Matchers.*;
import static org.junit.Assert.*;
import static org.mockito.BDDMockito.*;
import jj.http.server.websocket.WebSocketConnectionHost;
import jj.script.module.RootScriptEnvironment;
import jj.util.Closer;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;
import org.mockito.Captor;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;
import org.mozilla.javascript.ContinuationPending;

/**
 * @author jason
 *
 */
@RunWith(MockitoJUnitRunner.class)
public class CurrentScriptEnvironmentTest {
	
	PendingKey pendingKey;
	
	Continuation continuation = new Continuation() {
		
		PendingKey pendingKey;
		
		@Override
		public PendingKey pendingKey() {
			return pendingKey;
		}
		
		@Override
		public void pendingKey(PendingKey pendingKey) {
			this.pendingKey = pendingKey;
		}
	};

	CurrentScriptEnvironment cse;
	
	RhinoContext rhinoContext;
	
	@Mock(extraInterfaces = {RootScriptEnvironment.class}) AbstractScriptEnvironment<?> ase;
	
	@Mock(extraInterfaces = {RootScriptEnvironment.class}) WebSocketConnectionHost host;
	
	@Mock ChildScriptEnvironment<?> child;
	
	@Mock ContinuationPending continuationPending;
	
	@Captor ArgumentCaptor<ContinuationState> continuationStateCaptor;
	
	@Mock Closer mockCloser;
	
	@Before
	public void before() {
		
		pendingKey = new PendingKey();
		
		MockRhinoContextProvider rcp = new MockRhinoContextProvider();
		rhinoContext = rcp.context;
		cse = new CurrentScriptEnvironment(rcp);
		
		continuation.pendingKey(null);
	}
	
	@After
	public void after() {
		assertThat(cse.current(), is(nullValue()));
	}
	
	@Test
	public void testCapturingContinuationInNormalScope() {
		
		given(rhinoContext.captureContinuation()).willReturn(continuationPending);
		given(ase.createContinuationContext(continuationPending)).willReturn(pendingKey);
		
		try (Closer closer = cse.enterScope(ase)) {
			
			cse.preparedContinuation(continuation);
			fail("should have thrown");
		} catch (ContinuationPending cp) {
			verify(cp).setApplicationState(continuationStateCaptor.capture());
			assertThat(continuationStateCaptor.getValue().pendingKey(), is(pendingKey));
		}
		
		verify(ase).createContinuationContext(continuationPending);
		
		given(ase.restoreContextForKey(pendingKey)).willReturn(mockCloser);
		
		try (Closer closer = cse.enterScope(ase, pendingKey)) {
			
		}
		
		verify(mockCloser).close();
	}

	@Test
	public void testCurrentRootScriptEnvironment() {
		
		willReturn(host).given(child).parent();
		
		try (Closer closer = cse.enterScope(child)) {
			assertThat(cse.currentRootScriptEnvironment(), is(host));
		}
		
		try (Closer closer = cse.enterScope(ase)) {
			assertThat(cse.currentRootScriptEnvironment(), is(ase));
		}
		
	}
}