@@ -26,35 +26,30 @@ async def asyncTearDown(self) -> None:
26
26
await Tortoise ._drop_databases ()
27
27
await super ().asyncTearDown ()
28
28
29
+ def build_select_sql (self ) -> str :
30
+ if isinstance (self .db , OracleClient ):
31
+ return 'SELECT * FROM "eventtwo"'
32
+ return "SELECT * FROM eventtwo"
33
+
29
34
async def test_two_databases (self ):
30
35
tournament = await Tournament .create (name = "Tournament" )
31
36
await EventTwo .create (name = "Event" , tournament_id = tournament .id )
32
37
38
+ select_sql = self .build_select_sql ()
33
39
with self .assertRaises (OperationalError ):
34
- if isinstance (self .db , OracleClient ):
35
- await self .db .execute_query ('SELECT * FROM "eventtwo"' )
36
- else :
37
- await self .db .execute_query ("SELECT * FROM eventtwo" )
38
- if isinstance (self .db , OracleClient ):
39
- _ , results = await self .second_db .execute_query ('SELECT * FROM "eventtwo"' )
40
- else :
41
- _ , results = await self .second_db .execute_query ("SELECT * FROM eventtwo" )
40
+ await self .db .execute_query (select_sql )
41
+ _ , results = await self .second_db .execute_query (select_sql )
42
42
self .assertEqual (dict (results [0 ]), {"id" : 1 , "name" : "Event" , "tournament_id" : 1 })
43
43
44
44
async def test_two_databases_relation (self ):
45
45
tournament = await Tournament .create (name = "Tournament" )
46
46
event = await EventTwo .create (name = "Event" , tournament_id = tournament .id )
47
47
48
+ select_sql = self .build_select_sql ()
48
49
with self .assertRaises (OperationalError ):
49
- if isinstance (self .db , OracleClient ):
50
- await self .db .execute_query ('SELECT * FROM "eventtwo"' )
51
- else :
52
- await self .db .execute_query ("SELECT * FROM eventtwo" )
50
+ await self .db .execute_query (select_sql )
53
51
54
- if isinstance (self .db , OracleClient ):
55
- _ , results = await self .second_db .execute_query ('SELECT * FROM "eventtwo"' )
56
- else :
57
- _ , results = await self .second_db .execute_query ("SELECT * FROM eventtwo" )
52
+ _ , results = await self .second_db .execute_query (select_sql )
58
53
self .assertEqual (dict (results [0 ]), {"id" : 1 , "name" : "Event" , "tournament_id" : 1 })
59
54
60
55
teams = []
0 commit comments