@@ -118,9 +118,9 @@ def setUpTestData(cls):
118118 DocumentRelease .objects .bulk_create (
119119 DocumentRelease (lang = lang , release = release )
120120 for lang , release in [
121- ("en" , None ),
122- ("en" , r1 ),
123- ("en" , r2 ),
121+ (settings . DEFAULT_LANGUAGE_CODE , None ),
122+ (settings . DEFAULT_LANGUAGE_CODE , r1 ),
123+ (settings . DEFAULT_LANGUAGE_CODE , r2 ),
124124 ("sv" , r1 ),
125125 ("ar" , r1 ),
126126 ]
@@ -129,23 +129,28 @@ def setUpTestData(cls):
129129 def test_by_version (self ):
130130 self .assertQuerySetEqual (
131131 DocumentRelease .objects .by_version ("1.0" ),
132- [("en" , "1.0" ), ("sv" , "1.0" ), ("ar" , "1.0" )],
132+ [(settings . DEFAULT_LANGUAGE_CODE , "1.0" ), ("sv" , "1.0" ), ("ar" , "1.0" )],
133133 transform = attrgetter ("lang" , "version" ),
134134 ordered = False ,
135135 )
136136
137137 def test_by_version_dev (self ):
138138 self .assertQuerySetEqual (
139139 DocumentRelease .objects .by_version ("dev" ),
140- [("en" , "dev" )],
140+ [(settings . DEFAULT_LANGUAGE_CODE , "dev" )],
141141 transform = attrgetter ("lang" , "version" ),
142142 ordered = False ,
143143 )
144144
145145 def test_by_versions (self ):
146146 self .assertQuerySetEqual (
147147 DocumentRelease .objects .by_versions ("1.0" , "dev" ),
148- [("en" , "dev" ), ("en" , "1.0" ), ("sv" , "1.0" ), ("ar" , "1.0" )],
148+ [
149+ (settings .DEFAULT_LANGUAGE_CODE , "dev" ),
150+ (settings .DEFAULT_LANGUAGE_CODE , "1.0" ),
151+ ("sv" , "1.0" ),
152+ ("ar" , "1.0" ),
153+ ],
149154 transform = attrgetter ("lang" , "version" ),
150155 ordered = False ,
151156 )
@@ -155,18 +160,20 @@ def test_by_versions_empty(self):
155160 DocumentRelease .objects .by_versions ()
156161
157162 def test_get_by_version_and_lang_exists (self ):
158- doc = DocumentRelease .objects .get_by_version_and_lang ("1.0" , "en" )
163+ doc = DocumentRelease .objects .get_by_version_and_lang (
164+ "1.0" , settings .DEFAULT_LANGUAGE_CODE
165+ )
159166 self .assertEqual (doc .release .version , "1.0" )
160- self .assertEqual (doc .lang , "en" )
167+ self .assertEqual (doc .lang , settings . DEFAULT_LANGUAGE_CODE )
161168
162169 def test_get_by_version_and_lang_missing (self ):
163170 with self .assertRaises (DocumentRelease .DoesNotExist ):
164171 DocumentRelease .objects .get_by_version_and_lang ("2.0" , "sv" )
165172
166173 def test_get_available_languages_by_version (self ):
167174 get = DocumentRelease .objects .get_available_languages_by_version
168- self .assertEqual (list (get ("1.0" )), ["ar" , "en" , "sv" ])
169- self .assertEqual (list (get ("2.0" )), ["en" ])
175+ self .assertEqual (list (get ("1.0" )), ["ar" , settings . DEFAULT_LANGUAGE_CODE , "sv" ])
176+ self .assertEqual (list (get ("2.0" )), [settings . DEFAULT_LANGUAGE_CODE ])
170177 self .assertEqual (list (get ("3.0" )), [])
171178
172179
@@ -440,7 +447,7 @@ def test_search_title(self):
440447 with self .assertNumQueries (1 ):
441448 self .assertQuerySetEqual (
442449 misspelled_query ,
443- [("Generic views" , "en" , "1.2.3" )],
450+ [("Generic views" , settings . DEFAULT_LANGUAGE_CODE , "1.2.3" )],
444451 transform = attrgetter (
445452 "headline" , "release.lang" , "release.release.version"
446453 ),
0 commit comments