ccitemplatetests.java

spring 框架代码

		ct.execute(interactionSpec, generator, extractor);

		connectionFactoryControl.verify();
		connectionControl.verify();
		interactionControl.verify();
		generatorControl.verify();
		extractorControl.verify();
	}

	public void testTemplateExecuteInputOutputConnectionSpec() throws ResourceException {
		MockControl connectionFactoryControl = MockControl.createStrictControl(ConnectionFactory.class);
		ConnectionFactory connectionFactory = (ConnectionFactory) connectionFactoryControl.getMock();
		MockControl connectionControl = MockControl.createStrictControl(Connection.class);
		Connection connection = (Connection) connectionControl.getMock();
		MockControl interactionControl = MockControl.createStrictControl(Interaction.class);
		Interaction interaction = (Interaction) interactionControl.getMock();

		MockControl inputRecordControl = MockControl.createStrictControl(Record.class);
		Record inputRecord = (Record) inputRecordControl.getMock();
		MockControl outputRecordControl = MockControl.createStrictControl(Record.class);
		Record outputRecord = (Record) outputRecordControl.getMock();

		MockControl connectionSpecControl = MockControl.createStrictControl(ConnectionSpec.class);
		ConnectionSpec connectionSpec = (ConnectionSpec) connectionSpecControl.getMock();
		MockControl interactionSpecControl = MockControl.createStrictControl(InteractionSpec.class);
		InteractionSpec interactionSpec = (InteractionSpec) interactionSpecControl.getMock();

		connectionFactory.getConnection(connectionSpec);
		connectionFactoryControl.setReturnValue(connection, 1);

		connection.createInteraction();
		connectionControl.setReturnValue(interaction, 1);

		interaction.execute(interactionSpec, inputRecord, outputRecord);
		interactionControl.setReturnValue(true, 1);

		interaction.close();
		interactionControl.setVoidCallable(1);

		connection.close();
		connectionControl.setVoidCallable(1);

		connectionFactoryControl.replay();
		connectionControl.replay();
		interactionControl.replay();

		ConnectionSpecConnectionFactoryAdapter adapter = new ConnectionSpecConnectionFactoryAdapter();
		adapter.setTargetConnectionFactory(connectionFactory);
		adapter.setConnectionSpec(connectionSpec);
		CciTemplate ct = new CciTemplate(adapter);
		ct.execute(interactionSpec, inputRecord, outputRecord);

		connectionFactoryControl.verify();
		connectionControl.verify();
		interactionControl.verify();
	}

	public void testTemplateExecuteInputOutputResultsSetFalse() throws ResourceException, SQLException {
		MockControl connectionFactoryControl = MockControl.createStrictControl(ConnectionFactory.class);
		ConnectionFactory connectionFactory = (ConnectionFactory) connectionFactoryControl.getMock();
		MockControl recordFactoryControl = MockControl.createStrictControl(RecordFactory.class);
		RecordFactory recordFactory = (RecordFactory) recordFactoryControl.getMock();
		MockControl connectionControl = MockControl.createStrictControl(Connection.class);
		Connection connection = (Connection) connectionControl.getMock();
		MockControl interactionControl = MockControl.createStrictControl(Interaction.class);
		Interaction interaction = (Interaction) interactionControl.getMock();
		MockControl recordControl = MockControl.createStrictControl(Record.class);
		Record record = (Record) recordControl.getMock();
		MockControl resultsetControl = MockControl.createStrictControl(ResultSet.class);
		ResultSet resultset = (ResultSet) resultsetControl.getMock();
		MockControl generatorControl = MockControl.createStrictControl(RecordCreator.class);
		RecordCreator generator = (RecordCreator) generatorControl.getMock();
		MockControl extractorControl = MockControl.createStrictControl(RecordExtractor.class);
		RecordExtractor extractor = (RecordExtractor) extractorControl.getMock();

		MockControl interactionSpecControl = MockControl.createStrictControl(InteractionSpec.class);
		InteractionSpec interactionSpec = (InteractionSpec) interactionSpecControl.getMock();

		Object obj = new Object();

		connectionFactory.getRecordFactory();
		connectionFactoryControl.setReturnValue(recordFactory, 1);

		connectionFactory.getConnection();
		connectionFactoryControl.setReturnValue(connection);

		connection.createInteraction();
		connectionControl.setReturnValue(interaction);

		generator.createRecord(recordFactory);
		generatorControl.setReturnValue(record);

		interaction.execute(interactionSpec, record);
		interactionControl.setReturnValue(resultset, 1);

		extractor.extractData(resultset);
		extractorControl.setReturnValue(obj);

		resultset.close();
		resultsetControl.setVoidCallable(1);

		interaction.close();
		interactionControl.setVoidCallable(1);

		connection.close();
		connectionControl.setVoidCallable(1);

		connectionFactoryControl.replay();
		connectionControl.replay();
		interactionControl.replay();
		generatorControl.replay();
		extractorControl.replay();
		resultsetControl.replay();

		CciTemplate ct = new CciTemplate(connectionFactory);
		ct.execute(interactionSpec, generator, extractor);

		connectionFactoryControl.verify();
		connectionControl.verify();
		interactionControl.verify();
		generatorControl.verify();
		extractorControl.verify();
		resultsetControl.verify();
	}

	public void testTemplateExecuteConnectionCallback() throws ResourceException, SQLException {
		MockControl connectionFactoryControl = MockControl.createStrictControl(ConnectionFactory.class);
		ConnectionFactory connectionFactory = (ConnectionFactory) connectionFactoryControl.getMock();
		MockControl connectionControl = MockControl.createStrictControl(Connection.class);
		Connection connection = (Connection) connectionControl.getMock();
		MockControl connectionCallbackControl = MockControl.createStrictControl(ConnectionCallback.class);
		ConnectionCallback connectionCallback = (ConnectionCallback) connectionCallbackControl.getMock();

		Object obj = new Object();

		connectionFactory.getConnection();
		connectionFactoryControl.setReturnValue(connection);

		connectionCallback.doInConnection(connection, connectionFactory);
		connectionCallbackControl.setReturnValue(obj);

		connection.close();
		connectionControl.setVoidCallable(1);

		connectionFactoryControl.replay();
		connectionControl.replay();
		connectionCallbackControl.replay();

		CciTemplate ct = new CciTemplate(connectionFactory);
		ct.execute(connectionCallback);

		connectionFactoryControl.verify();
		connectionControl.verify();
		connectionCallbackControl.verify();
	}

	public void testTemplateExecuteInteractionCallback() throws ResourceException, SQLException {
		MockControl connectionFactoryControl = MockControl.createStrictControl(ConnectionFactory.class);
		ConnectionFactory connectionFactory = (ConnectionFactory) connectionFactoryControl.getMock();
		MockControl connectionControl = MockControl.createStrictControl(Connection.class);
		Connection connection = (Connection) connectionControl.getMock();
		MockControl interactionControl = MockControl.createStrictControl(Interaction.class);
		Interaction interaction = (Interaction) interactionControl.getMock();
		MockControl interactionCallbackControl = MockControl.createStrictControl(InteractionCallback.class);
		InteractionCallback interactionCallback = (InteractionCallback) interactionCallbackControl.getMock();

		Object obj = new Object();

		connectionFactory.getConnection();
		connectionFactoryControl.setReturnValue(connection);

		connection.createInteraction();
		connectionControl.setReturnValue(interaction);

		interactionCallback.doInInteraction(interaction, connectionFactory);
		interactionCallbackControl.setReturnValue(obj);

		interaction.close();
		interactionControl.setVoidCallable(1);

		connection.close();
		connectionControl.setVoidCallable(1);

		connectionFactoryControl.replay();
		connectionControl.replay();
		interactionControl.replay();
		interactionCallbackControl.replay();

		CciTemplate ct = new CciTemplate(connectionFactory);
		ct.execute(interactionCallback);

		connectionFactoryControl.verify();
		connectionControl.verify();
		interactionControl.verify();
		interactionCallbackControl.verify();
	}

	public void testTemplateExecuteInputTrueTrueWithCreator() throws ResourceException {
		MockControl connectionFactoryControl = MockControl.createStrictControl(ConnectionFactory.class);
		ConnectionFactory connectionFactory = (ConnectionFactory) connectionFactoryControl.getMock();
		MockControl connectionControl = MockControl.createStrictControl(Connection.class);
		Connection connection = (Connection) connectionControl.getMock();
		MockControl interactionControl = MockControl.createStrictControl(Interaction.class);
		Interaction interaction = (Interaction) interactionControl.getMock();
		MockControl creatorControl = MockControl.createStrictControl(RecordCreator.class);
		RecordCreator creator = (RecordCreator) creatorControl.getMock();

		MockControl inputOutputRecordControl = MockControl.createStrictControl(Record.class);
		Record inputOutputRecord = (Record) inputOutputRecordControl.getMock();

		MockControl interactionSpecControl = MockControl.createStrictControl(InteractionSpec.class);
		InteractionSpec interactionSpec = (InteractionSpec) interactionSpecControl.getMock();

		connectionFactory.getConnection();
		connectionFactoryControl.setReturnValue(connection);

		connection.createInteraction();
		connectionControl.setReturnValue(interaction);

		interaction.execute(interactionSpec, inputOutputRecord, inputOutputRecord);
		interactionControl.setReturnValue(true, 1);

		interaction.close();
		interactionControl.setVoidCallable(1);

		connection.close();
		connectionControl.setVoidCallable(1);

		connectionFactoryControl.replay();
		connectionControl.replay();
		interactionControl.replay();
		creatorControl.replay();

		CciTemplate ct = new CciTemplate(connectionFactory);
		ct.setOutputRecordCreator(creator);
		ct.execute(interactionSpec, inputOutputRecord, inputOutputRecord);

		connectionFactoryControl.verify();
		connectionControl.verify();
		interactionControl.verify();
		creatorControl.verify();
	}

	public void testTemplateExecuteInputTrueTrue() throws ResourceException {
		MockControl connectionFactoryControl = MockControl.createStrictControl(ConnectionFactory.class);
		ConnectionFactory connectionFactory = (ConnectionFactory) connectionFactoryControl.getMock();
		MockControl connectionControl = MockControl.createStrictControl(Connection.class);
		Connection connection = (Connection) connectionControl.getMock();
		MockControl interactionControl = MockControl.createStrictControl(Interaction.class);
		Interaction interaction = (Interaction) interactionControl.getMock();

		MockControl inputOutputRecordControl = MockControl.createStrictControl(Record.class);
		Record inputOutputRecord = (Record) inputOutputRecordControl.getMock();

		MockControl interactionSpecControl = MockControl.createStrictControl(InteractionSpec.class);
		InteractionSpec interactionSpec = (InteractionSpec) interactionSpecControl.getMock();

		connectionFactory.getConnection();
		connectionFactoryControl.setReturnValue(connection);

		connection.createInteraction();
		connectionControl.setReturnValue(interaction);

		interaction.execute(interactionSpec, inputOutputRecord, inputOutputRecord);
		interactionControl.setReturnValue(true, 1);

		interaction.close();
		interactionControl.setVoidCallable(1);

		connection.close();
		connectionControl.setVoidCallable(1);

		connectionFactoryControl.replay();
		connectionControl.replay();
		interactionControl.replay();

		CciTemplate ct = new CciTemplate(connectionFactory);
		ct.execute(interactionSpec, inputOutputRecord, inputOutputRecord);

		connectionFactoryControl.verify();
		connectionControl.verify();
		interactionControl.verify();
	}

	public void testTemplateExecuteInputFalseTrue() throws ResourceException {
		MockControl connectionFactoryControl = MockControl.createStrictControl(ConnectionFactory.class);
		ConnectionFactory connectionFactory = (ConnectionFactory) connectionFactoryControl.getMock();
		MockControl connectionControl = MockControl.createStrictControl(Connection.class);
		Connection connection = (Connection) connectionControl.getMock();
		MockControl interactionControl = MockControl.createStrictControl(Interaction.class);
		Interaction interaction = (Interaction) interactionControl.getMock();

		MockControl inputOutputRecordControl = MockControl.createStrictControl(Record.class);
		Record inputOutputRecord = (Record) inputOutputRecordControl.getMock();

		MockControl interactionSpecControl = MockControl.createStrictControl(InteractionSpec.class);
		InteractionSpec interactionSpec = (InteractionSpec) interactionSpecControl.getMock();

		connectionFactory.getConnection();
		connectionFactoryControl.setReturnValue(connection);

		connection.createInteraction();
		connectionControl.setReturnValue(interaction);

		interaction.execute(interactionSpec, inputOutputRecord);
		interactionControl.setReturnValue(null, 1);

		interaction.close();
		interactionControl.setVoidCallable(1);

		connection.close();
		connectionControl.setVoidCallable(1);

		connectionFactoryControl.replay();
		connectionControl.replay();
		interactionControl.replay();

		CciTemplate ct = new CciTemplate(connectionFactory);
		Record tmpOutputRecord = (Record) ct.execute(interactionSpec, inputOutputRecord);
		assertNull(tmpOutputRecord);

		connectionFactoryControl.verify();
		connectionControl.verify();
		interactionControl.verify();
	}

}