slips/src/test/Marina.Test/Unit/DataAccess/DatabaseGatewayTest.cs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

using System.Data;
using Marina.DataAccess;
using MbUnit.Framework;
using Rhino.Mocks;

namespace Marina.Test.Unit.DataAccess {
	[TestFixture]
	public class DatabaseGatewayTest {
		private MockRepository _mockery;
		private IDatabaseConnectionFactory _mockConnectionFactory;

		[SetUp]
		public void Setup() {
			_mockery = new MockRepository( );
			_mockConnectionFactory = _mockery.DynamicMock< IDatabaseConnectionFactory >( );
		}

		public IDatabaseGateway CreateSUT() {
			return new DatabaseGateway( _mockConnectionFactory );
		}

		[Test]
		public void Should_leverage_factory_to_create_a_new_connection() {
			IDatabaseConnection mockConnection = _mockery.DynamicMock< IDatabaseConnection >( );
			using ( _mockery.Record( ) ) {
				Expect.Call( _mockConnectionFactory.Create( ) ).Return( mockConnection );
			}

			using ( _mockery.Playback( ) ) {
				CreateSUT( ).LoadTableUsing( "" );
			}
		}

		[Test]
		public void Should_create_command_using_database_connection() {
			IDatabaseConnection mockConnection = _mockery.DynamicMock< IDatabaseConnection >( );
			IDatabaseCommand mockCommand = _mockery.DynamicMock< IDatabaseCommand >( );
			string sqlQuery = "SELECT * FROM ?";

			using ( _mockery.Record( ) ) {
				SetupResult.For( _mockConnectionFactory.Create( ) ).Return( mockConnection );
				Expect.Call( mockConnection.CreateCommandFor( sqlQuery ) ).Return( mockCommand );
			}

			using ( _mockery.Playback( ) ) {
				CreateSUT( ).LoadTableUsing( sqlQuery );
			}
		}

		[RowTest]
		[Row( "SELECT * FROM Cars" )]
		[Row( "SELECT * FROM Persons" )]
		public void Should_execute_query_on_command( string sqlQuery ) {
			IDatabaseConnection mockConnection = _mockery.DynamicMock< IDatabaseConnection >( );
			IDatabaseCommand mockCommand = _mockery.DynamicMock< IDatabaseCommand >( );

			DataTable table = new DataTable( );
			using ( _mockery.Record( ) ) {
				SetupResult.For( _mockConnectionFactory.Create( ) ).Return( mockConnection );
				SetupResult.For( mockConnection.CreateCommandFor( sqlQuery ) ).Return( mockCommand );
				Expect.Call( mockCommand.ExecuteQuery( ) ).Return( table );
			}

			using ( _mockery.Playback( ) ) {
				Assert.AreEqual( table, CreateSUT( ).LoadTableUsing( sqlQuery ) );
			}
		}

		[Test]
		public void Should_close_the_connection_after_executing_the_command() {
			IDatabaseConnection mockConnection = _mockery.DynamicMock< IDatabaseConnection >( );
			IDatabaseCommand mockCommand = _mockery.DynamicMock< IDatabaseCommand >( );
			string sqlQuery = "SELECT * FROM ?";

			DataTable table = new DataTable( );
			using ( _mockery.Record( ) ) {
				SetupResult.For( _mockConnectionFactory.Create( ) ).Return( mockConnection );
				SetupResult.For( mockConnection.CreateCommandFor( sqlQuery ) ).Return( mockCommand );

				using ( _mockery.Ordered( ) ) {
					Expect.Call( mockCommand.ExecuteQuery( ) ).Return( table );
					mockConnection.Dispose( );
				}
			}

			using ( _mockery.Playback( ) ) {
				CreateSUT( ).LoadTableUsing( sqlQuery );
			}
		}
	}
}