4
0

config_test.go 33 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469
  1. // Copyright 2019 Frédéric Guillot. All rights reserved.
  2. // Use of this source code is governed by the Apache 2.0
  3. // license that can be found in the LICENSE file.
  4. package config // import "miniflux.app/config"
  5. import (
  6. "io/ioutil"
  7. "os"
  8. "testing"
  9. )
  10. func TestDebugModeOn(t *testing.T) {
  11. os.Clearenv()
  12. os.Setenv("DEBUG", "1")
  13. parser := NewParser()
  14. opts, err := parser.ParseEnvironmentVariables()
  15. if err != nil {
  16. t.Fatalf(`Parsing failure: %v`, err)
  17. }
  18. if !opts.HasDebugMode() {
  19. t.Fatalf(`Unexpected debug mode value, got "%v"`, opts.HasDebugMode())
  20. }
  21. }
  22. func TestDebugModeOff(t *testing.T) {
  23. os.Clearenv()
  24. parser := NewParser()
  25. opts, err := parser.ParseEnvironmentVariables()
  26. if err != nil {
  27. t.Fatalf(`Parsing failure: %v`, err)
  28. }
  29. if opts.HasDebugMode() {
  30. t.Fatalf(`Unexpected debug mode value, got "%v"`, opts.HasDebugMode())
  31. }
  32. }
  33. func TestCustomBaseURL(t *testing.T) {
  34. os.Clearenv()
  35. os.Setenv("BASE_URL", "http://example.org")
  36. parser := NewParser()
  37. opts, err := parser.ParseEnvironmentVariables()
  38. if err != nil {
  39. t.Fatalf(`Parsing failure: %v`, err)
  40. }
  41. if opts.BaseURL() != "http://example.org" {
  42. t.Fatalf(`Unexpected base URL, got "%s"`, opts.BaseURL())
  43. }
  44. if opts.RootURL() != "http://example.org" {
  45. t.Fatalf(`Unexpected root URL, got "%s"`, opts.RootURL())
  46. }
  47. if opts.BasePath() != "" {
  48. t.Fatalf(`Unexpected base path, got "%s"`, opts.BasePath())
  49. }
  50. }
  51. func TestCustomBaseURLWithTrailingSlash(t *testing.T) {
  52. os.Clearenv()
  53. os.Setenv("BASE_URL", "http://example.org/folder/")
  54. parser := NewParser()
  55. opts, err := parser.ParseEnvironmentVariables()
  56. if err != nil {
  57. t.Fatalf(`Parsing failure: %v`, err)
  58. }
  59. if opts.BaseURL() != "http://example.org/folder" {
  60. t.Fatalf(`Unexpected base URL, got "%s"`, opts.BaseURL())
  61. }
  62. if opts.RootURL() != "http://example.org" {
  63. t.Fatalf(`Unexpected root URL, got "%s"`, opts.RootURL())
  64. }
  65. if opts.BasePath() != "/folder" {
  66. t.Fatalf(`Unexpected base path, got "%s"`, opts.BasePath())
  67. }
  68. }
  69. func TestBaseURLWithoutScheme(t *testing.T) {
  70. os.Clearenv()
  71. os.Setenv("BASE_URL", "example.org/folder/")
  72. _, err := NewParser().ParseEnvironmentVariables()
  73. if err == nil {
  74. t.Fatalf(`Parsing must fail`)
  75. }
  76. }
  77. func TestBaseURLWithInvalidScheme(t *testing.T) {
  78. os.Clearenv()
  79. os.Setenv("BASE_URL", "ftp://example.org/folder/")
  80. _, err := NewParser().ParseEnvironmentVariables()
  81. if err == nil {
  82. t.Fatalf(`Parsing must fail`)
  83. }
  84. }
  85. func TestInvalidBaseURL(t *testing.T) {
  86. os.Clearenv()
  87. os.Setenv("BASE_URL", "http://example|org")
  88. _, err := NewParser().ParseEnvironmentVariables()
  89. if err == nil {
  90. t.Fatalf(`Parsing must fail`)
  91. }
  92. }
  93. func TestDefaultBaseURL(t *testing.T) {
  94. os.Clearenv()
  95. parser := NewParser()
  96. opts, err := parser.ParseEnvironmentVariables()
  97. if err != nil {
  98. t.Fatalf(`Parsing failure: %v`, err)
  99. }
  100. if opts.BaseURL() != defaultBaseURL {
  101. t.Fatalf(`Unexpected base URL, got "%s"`, opts.BaseURL())
  102. }
  103. if opts.RootURL() != defaultBaseURL {
  104. t.Fatalf(`Unexpected root URL, got "%s"`, opts.RootURL())
  105. }
  106. if opts.BasePath() != "" {
  107. t.Fatalf(`Unexpected base path, got "%s"`, opts.BasePath())
  108. }
  109. }
  110. func TestDatabaseURL(t *testing.T) {
  111. os.Clearenv()
  112. os.Setenv("DATABASE_URL", "foobar")
  113. parser := NewParser()
  114. opts, err := parser.ParseEnvironmentVariables()
  115. if err != nil {
  116. t.Fatalf(`Parsing failure: %v`, err)
  117. }
  118. expected := "foobar"
  119. result := opts.DatabaseURL()
  120. if result != expected {
  121. t.Errorf(`Unexpected DATABASE_URL value, got %q instead of %q`, result, expected)
  122. }
  123. if opts.IsDefaultDatabaseURL() {
  124. t.Errorf(`This is not the default database URL and it should returns false`)
  125. }
  126. }
  127. func TestDefaultDatabaseURLValue(t *testing.T) {
  128. os.Clearenv()
  129. parser := NewParser()
  130. opts, err := parser.ParseEnvironmentVariables()
  131. if err != nil {
  132. t.Fatalf(`Parsing failure: %v`, err)
  133. }
  134. expected := defaultDatabaseURL
  135. result := opts.DatabaseURL()
  136. if result != expected {
  137. t.Errorf(`Unexpected DATABASE_URL value, got %q instead of %q`, result, expected)
  138. }
  139. if !opts.IsDefaultDatabaseURL() {
  140. t.Errorf(`This is the default database URL and it should returns true`)
  141. }
  142. }
  143. func TestDefaultDatabaseMaxConnsValue(t *testing.T) {
  144. os.Clearenv()
  145. parser := NewParser()
  146. opts, err := parser.ParseEnvironmentVariables()
  147. if err != nil {
  148. t.Fatalf(`Parsing failure: %v`, err)
  149. }
  150. expected := defaultDatabaseMaxConns
  151. result := opts.DatabaseMaxConns()
  152. if result != expected {
  153. t.Fatalf(`Unexpected DATABASE_MAX_CONNS value, got %v instead of %v`, result, expected)
  154. }
  155. }
  156. func TestDatabaseMaxConns(t *testing.T) {
  157. os.Clearenv()
  158. os.Setenv("DATABASE_MAX_CONNS", "42")
  159. parser := NewParser()
  160. opts, err := parser.ParseEnvironmentVariables()
  161. if err != nil {
  162. t.Fatalf(`Parsing failure: %v`, err)
  163. }
  164. expected := 42
  165. result := opts.DatabaseMaxConns()
  166. if result != expected {
  167. t.Fatalf(`Unexpected DATABASE_MAX_CONNS value, got %v instead of %v`, result, expected)
  168. }
  169. }
  170. func TestDefaultDatabaseMinConnsValue(t *testing.T) {
  171. os.Clearenv()
  172. parser := NewParser()
  173. opts, err := parser.ParseEnvironmentVariables()
  174. if err != nil {
  175. t.Fatalf(`Parsing failure: %v`, err)
  176. }
  177. expected := defaultDatabaseMinConns
  178. result := opts.DatabaseMinConns()
  179. if result != expected {
  180. t.Fatalf(`Unexpected DATABASE_MIN_CONNS value, got %v instead of %v`, result, expected)
  181. }
  182. }
  183. func TestDatabaseMinConns(t *testing.T) {
  184. os.Clearenv()
  185. os.Setenv("DATABASE_MIN_CONNS", "42")
  186. parser := NewParser()
  187. opts, err := parser.ParseEnvironmentVariables()
  188. if err != nil {
  189. t.Fatalf(`Parsing failure: %v`, err)
  190. }
  191. expected := 42
  192. result := opts.DatabaseMinConns()
  193. if result != expected {
  194. t.Fatalf(`Unexpected DATABASE_MIN_CONNS value, got %v instead of %v`, result, expected)
  195. }
  196. }
  197. func TestListenAddr(t *testing.T) {
  198. os.Clearenv()
  199. os.Setenv("LISTEN_ADDR", "foobar")
  200. parser := NewParser()
  201. opts, err := parser.ParseEnvironmentVariables()
  202. if err != nil {
  203. t.Fatalf(`Parsing failure: %v`, err)
  204. }
  205. expected := "foobar"
  206. result := opts.ListenAddr()
  207. if result != expected {
  208. t.Fatalf(`Unexpected LISTEN_ADDR value, got %q instead of %q`, result, expected)
  209. }
  210. }
  211. func TestListenAddrWithPortDefined(t *testing.T) {
  212. os.Clearenv()
  213. os.Setenv("PORT", "3000")
  214. os.Setenv("LISTEN_ADDR", "foobar")
  215. parser := NewParser()
  216. opts, err := parser.ParseEnvironmentVariables()
  217. if err != nil {
  218. t.Fatalf(`Parsing failure: %v`, err)
  219. }
  220. expected := ":3000"
  221. result := opts.ListenAddr()
  222. if result != expected {
  223. t.Fatalf(`Unexpected LISTEN_ADDR value, got %q instead of %q`, result, expected)
  224. }
  225. }
  226. func TestDefaultListenAddrValue(t *testing.T) {
  227. os.Clearenv()
  228. parser := NewParser()
  229. opts, err := parser.ParseEnvironmentVariables()
  230. if err != nil {
  231. t.Fatalf(`Parsing failure: %v`, err)
  232. }
  233. expected := defaultListenAddr
  234. result := opts.ListenAddr()
  235. if result != expected {
  236. t.Fatalf(`Unexpected LISTEN_ADDR value, got %q instead of %q`, result, expected)
  237. }
  238. }
  239. func TestCertFile(t *testing.T) {
  240. os.Clearenv()
  241. os.Setenv("CERT_FILE", "foobar")
  242. parser := NewParser()
  243. opts, err := parser.ParseEnvironmentVariables()
  244. if err != nil {
  245. t.Fatalf(`Parsing failure: %v`, err)
  246. }
  247. expected := "foobar"
  248. result := opts.CertFile()
  249. if result != expected {
  250. t.Fatalf(`Unexpected CERT_FILE value, got %q instead of %q`, result, expected)
  251. }
  252. }
  253. func TestDefaultCertFileValue(t *testing.T) {
  254. os.Clearenv()
  255. parser := NewParser()
  256. opts, err := parser.ParseEnvironmentVariables()
  257. if err != nil {
  258. t.Fatalf(`Parsing failure: %v`, err)
  259. }
  260. expected := defaultCertFile
  261. result := opts.CertFile()
  262. if result != expected {
  263. t.Fatalf(`Unexpected CERT_FILE value, got %q instead of %q`, result, expected)
  264. }
  265. }
  266. func TestKeyFile(t *testing.T) {
  267. os.Clearenv()
  268. os.Setenv("KEY_FILE", "foobar")
  269. parser := NewParser()
  270. opts, err := parser.ParseEnvironmentVariables()
  271. if err != nil {
  272. t.Fatalf(`Parsing failure: %v`, err)
  273. }
  274. expected := "foobar"
  275. result := opts.CertKeyFile()
  276. if result != expected {
  277. t.Fatalf(`Unexpected KEY_FILE value, got %q instead of %q`, result, expected)
  278. }
  279. }
  280. func TestDefaultKeyFileValue(t *testing.T) {
  281. os.Clearenv()
  282. parser := NewParser()
  283. opts, err := parser.ParseEnvironmentVariables()
  284. if err != nil {
  285. t.Fatalf(`Parsing failure: %v`, err)
  286. }
  287. expected := defaultKeyFile
  288. result := opts.CertKeyFile()
  289. if result != expected {
  290. t.Fatalf(`Unexpected KEY_FILE value, got %q instead of %q`, result, expected)
  291. }
  292. }
  293. func TestCertDomain(t *testing.T) {
  294. os.Clearenv()
  295. os.Setenv("CERT_DOMAIN", "example.org")
  296. parser := NewParser()
  297. opts, err := parser.ParseEnvironmentVariables()
  298. if err != nil {
  299. t.Fatalf(`Parsing failure: %v`, err)
  300. }
  301. expected := "example.org"
  302. result := opts.CertDomain()
  303. if result != expected {
  304. t.Fatalf(`Unexpected CERT_DOMAIN value, got %q instead of %q`, result, expected)
  305. }
  306. }
  307. func TestDefaultCertDomainValue(t *testing.T) {
  308. os.Clearenv()
  309. parser := NewParser()
  310. opts, err := parser.ParseEnvironmentVariables()
  311. if err != nil {
  312. t.Fatalf(`Parsing failure: %v`, err)
  313. }
  314. expected := defaultCertDomain
  315. result := opts.CertDomain()
  316. if result != expected {
  317. t.Fatalf(`Unexpected CERT_DOMAIN value, got %q instead of %q`, result, expected)
  318. }
  319. }
  320. func TestCertCache(t *testing.T) {
  321. os.Clearenv()
  322. os.Setenv("CERT_CACHE", "foobar")
  323. parser := NewParser()
  324. opts, err := parser.ParseEnvironmentVariables()
  325. if err != nil {
  326. t.Fatalf(`Parsing failure: %v`, err)
  327. }
  328. expected := "foobar"
  329. result := opts.CertCache()
  330. if result != expected {
  331. t.Fatalf(`Unexpected CERT_CACHE value, got %q instead of %q`, result, expected)
  332. }
  333. }
  334. func TestDefaultCertCacheValue(t *testing.T) {
  335. os.Clearenv()
  336. parser := NewParser()
  337. opts, err := parser.ParseEnvironmentVariables()
  338. if err != nil {
  339. t.Fatalf(`Parsing failure: %v`, err)
  340. }
  341. expected := defaultCertCache
  342. result := opts.CertCache()
  343. if result != expected {
  344. t.Fatalf(`Unexpected CERT_CACHE value, got %q instead of %q`, result, expected)
  345. }
  346. }
  347. func TestDefaultCleanupFrequencyHoursValue(t *testing.T) {
  348. os.Clearenv()
  349. parser := NewParser()
  350. opts, err := parser.ParseEnvironmentVariables()
  351. if err != nil {
  352. t.Fatalf(`Parsing failure: %v`, err)
  353. }
  354. expected := defaultCleanupFrequencyHours
  355. result := opts.CleanupFrequencyHours()
  356. if result != expected {
  357. t.Fatalf(`Unexpected CLEANUP_FREQUENCY_HOURS value, got %v instead of %v`, result, expected)
  358. }
  359. }
  360. func TestCleanupFrequencyHours(t *testing.T) {
  361. os.Clearenv()
  362. os.Setenv("CLEANUP_FREQUENCY_HOURS", "42")
  363. os.Setenv("CLEANUP_FREQUENCY", "19")
  364. parser := NewParser()
  365. opts, err := parser.ParseEnvironmentVariables()
  366. if err != nil {
  367. t.Fatalf(`Parsing failure: %v`, err)
  368. }
  369. expected := 42
  370. result := opts.CleanupFrequencyHours()
  371. if result != expected {
  372. t.Fatalf(`Unexpected CLEANUP_FREQUENCY_HOURS value, got %v instead of %v`, result, expected)
  373. }
  374. }
  375. func TestDeprecatedCleanupFrequencyHoursVar(t *testing.T) {
  376. os.Clearenv()
  377. os.Setenv("CLEANUP_FREQUENCY", "42")
  378. parser := NewParser()
  379. opts, err := parser.ParseEnvironmentVariables()
  380. if err != nil {
  381. t.Fatalf(`Parsing failure: %v`, err)
  382. }
  383. expected := 42
  384. result := opts.CleanupFrequencyHours()
  385. if result != expected {
  386. t.Fatalf(`Unexpected CLEANUP_FREQUENCY value, got %v instead of %v`, result, expected)
  387. }
  388. }
  389. func TestDefaultCleanupArchiveReadDaysValue(t *testing.T) {
  390. os.Clearenv()
  391. parser := NewParser()
  392. opts, err := parser.ParseEnvironmentVariables()
  393. if err != nil {
  394. t.Fatalf(`Parsing failure: %v`, err)
  395. }
  396. expected := 60
  397. result := opts.CleanupArchiveReadDays()
  398. if result != expected {
  399. t.Fatalf(`Unexpected CLEANUP_ARCHIVE_READ_DAYS value, got %v instead of %v`, result, expected)
  400. }
  401. }
  402. func TestCleanupArchiveReadDays(t *testing.T) {
  403. os.Clearenv()
  404. os.Setenv("CLEANUP_ARCHIVE_READ_DAYS", "7")
  405. os.Setenv("ARCHIVE_READ_DAYS", "19")
  406. parser := NewParser()
  407. opts, err := parser.ParseEnvironmentVariables()
  408. if err != nil {
  409. t.Fatalf(`Parsing failure: %v`, err)
  410. }
  411. expected := 7
  412. result := opts.CleanupArchiveReadDays()
  413. if result != expected {
  414. t.Fatalf(`Unexpected CLEANUP_ARCHIVE_READ_DAYS value, got %v instead of %v`, result, expected)
  415. }
  416. }
  417. func TestDeprecatedCleanupArchiveReadDaysVar(t *testing.T) {
  418. os.Clearenv()
  419. os.Setenv("ARCHIVE_READ_DAYS", "7")
  420. parser := NewParser()
  421. opts, err := parser.ParseEnvironmentVariables()
  422. if err != nil {
  423. t.Fatalf(`Parsing failure: %v`, err)
  424. }
  425. expected := 7
  426. result := opts.CleanupArchiveReadDays()
  427. if result != expected {
  428. t.Fatalf(`Unexpected ARCHIVE_READ_DAYS value, got %v instead of %v`, result, expected)
  429. }
  430. }
  431. func TestDefaultCleanupRemoveSessionsDaysValue(t *testing.T) {
  432. os.Clearenv()
  433. parser := NewParser()
  434. opts, err := parser.ParseEnvironmentVariables()
  435. if err != nil {
  436. t.Fatalf(`Parsing failure: %v`, err)
  437. }
  438. expected := 30
  439. result := opts.CleanupRemoveSessionsDays()
  440. if result != expected {
  441. t.Fatalf(`Unexpected CLEANUP_REMOVE_SESSIONS_DAYS value, got %v instead of %v`, result, expected)
  442. }
  443. }
  444. func TestCleanupRemoveSessionsDays(t *testing.T) {
  445. os.Clearenv()
  446. os.Setenv("CLEANUP_REMOVE_SESSIONS_DAYS", "7")
  447. parser := NewParser()
  448. opts, err := parser.ParseEnvironmentVariables()
  449. if err != nil {
  450. t.Fatalf(`Parsing failure: %v`, err)
  451. }
  452. expected := 7
  453. result := opts.CleanupRemoveSessionsDays()
  454. if result != expected {
  455. t.Fatalf(`Unexpected CLEANUP_REMOVE_SESSIONS_DAYS value, got %v instead of %v`, result, expected)
  456. }
  457. }
  458. func TestDefaultWorkerPoolSizeValue(t *testing.T) {
  459. os.Clearenv()
  460. parser := NewParser()
  461. opts, err := parser.ParseEnvironmentVariables()
  462. if err != nil {
  463. t.Fatalf(`Parsing failure: %v`, err)
  464. }
  465. expected := defaultWorkerPoolSize
  466. result := opts.WorkerPoolSize()
  467. if result != expected {
  468. t.Fatalf(`Unexpected WORKER_POOL_SIZE value, got %v instead of %v`, result, expected)
  469. }
  470. }
  471. func TestWorkerPoolSize(t *testing.T) {
  472. os.Clearenv()
  473. os.Setenv("WORKER_POOL_SIZE", "42")
  474. parser := NewParser()
  475. opts, err := parser.ParseEnvironmentVariables()
  476. if err != nil {
  477. t.Fatalf(`Parsing failure: %v`, err)
  478. }
  479. expected := 42
  480. result := opts.WorkerPoolSize()
  481. if result != expected {
  482. t.Fatalf(`Unexpected WORKER_POOL_SIZE value, got %v instead of %v`, result, expected)
  483. }
  484. }
  485. func TestDefautPollingFrequencyValue(t *testing.T) {
  486. os.Clearenv()
  487. parser := NewParser()
  488. opts, err := parser.ParseEnvironmentVariables()
  489. if err != nil {
  490. t.Fatalf(`Parsing failure: %v`, err)
  491. }
  492. expected := defaultPollingFrequency
  493. result := opts.PollingFrequency()
  494. if result != expected {
  495. t.Fatalf(`Unexpected POLLING_FREQUENCY value, got %v instead of %v`, result, expected)
  496. }
  497. }
  498. func TestPollingFrequency(t *testing.T) {
  499. os.Clearenv()
  500. os.Setenv("POLLING_FREQUENCY", "42")
  501. parser := NewParser()
  502. opts, err := parser.ParseEnvironmentVariables()
  503. if err != nil {
  504. t.Fatalf(`Parsing failure: %v`, err)
  505. }
  506. expected := 42
  507. result := opts.PollingFrequency()
  508. if result != expected {
  509. t.Fatalf(`Unexpected POLLING_FREQUENCY value, got %v instead of %v`, result, expected)
  510. }
  511. }
  512. func TestDefaultBatchSizeValue(t *testing.T) {
  513. os.Clearenv()
  514. parser := NewParser()
  515. opts, err := parser.ParseEnvironmentVariables()
  516. if err != nil {
  517. t.Fatalf(`Parsing failure: %v`, err)
  518. }
  519. expected := defaultBatchSize
  520. result := opts.BatchSize()
  521. if result != expected {
  522. t.Fatalf(`Unexpected BATCH_SIZE value, got %v instead of %v`, result, expected)
  523. }
  524. }
  525. func TestBatchSize(t *testing.T) {
  526. os.Clearenv()
  527. os.Setenv("BATCH_SIZE", "42")
  528. parser := NewParser()
  529. opts, err := parser.ParseEnvironmentVariables()
  530. if err != nil {
  531. t.Fatalf(`Parsing failure: %v`, err)
  532. }
  533. expected := 42
  534. result := opts.BatchSize()
  535. if result != expected {
  536. t.Fatalf(`Unexpected BATCH_SIZE value, got %v instead of %v`, result, expected)
  537. }
  538. }
  539. func TestDefautPollingSchedulerValue(t *testing.T) {
  540. os.Clearenv()
  541. parser := NewParser()
  542. opts, err := parser.ParseEnvironmentVariables()
  543. if err != nil {
  544. t.Fatalf(`Parsing failure: %v`, err)
  545. }
  546. expected := defaultPollingScheduler
  547. result := opts.PollingScheduler()
  548. if result != expected {
  549. t.Fatalf(`Unexpected POLLING_SCHEDULER value, got %v instead of %v`, result, expected)
  550. }
  551. }
  552. func TestPollingScheduler(t *testing.T) {
  553. os.Clearenv()
  554. os.Setenv("POLLING_SCHEDULER", "entry_count_based")
  555. parser := NewParser()
  556. opts, err := parser.ParseEnvironmentVariables()
  557. if err != nil {
  558. t.Fatalf(`Parsing failure: %v`, err)
  559. }
  560. expected := "entry_count_based"
  561. result := opts.PollingScheduler()
  562. if result != expected {
  563. t.Fatalf(`Unexpected POLLING_SCHEDULER value, got %v instead of %v`, result, expected)
  564. }
  565. }
  566. func TestDefautSchedulerCountBasedMaxIntervalValue(t *testing.T) {
  567. os.Clearenv()
  568. parser := NewParser()
  569. opts, err := parser.ParseEnvironmentVariables()
  570. if err != nil {
  571. t.Fatalf(`Parsing failure: %v`, err)
  572. }
  573. expected := defaultSchedulerCountBasedMaxInterval
  574. result := opts.SchedulerCountBasedMaxInterval()
  575. if result != expected {
  576. t.Fatalf(`Unexpected SCHEDULER_ENTRY_COUNT_BASED_MAX_INTERVAL value, got %v instead of %v`, result, expected)
  577. }
  578. }
  579. func TestDefautSchedulerCountBasedMaxInterval(t *testing.T) {
  580. os.Clearenv()
  581. os.Setenv("SCHEDULER_ENTRY_COUNT_BASED_MAX_INTERVAL", "30")
  582. parser := NewParser()
  583. opts, err := parser.ParseEnvironmentVariables()
  584. if err != nil {
  585. t.Fatalf(`Parsing failure: %v`, err)
  586. }
  587. expected := 30
  588. result := opts.SchedulerCountBasedMaxInterval()
  589. if result != expected {
  590. t.Fatalf(`Unexpected SCHEDULER_ENTRY_COUNT_BASED_MAX_INTERVAL value, got %v instead of %v`, result, expected)
  591. }
  592. }
  593. func TestDefautSchedulerCountBasedMinIntervalValue(t *testing.T) {
  594. os.Clearenv()
  595. parser := NewParser()
  596. opts, err := parser.ParseEnvironmentVariables()
  597. if err != nil {
  598. t.Fatalf(`Parsing failure: %v`, err)
  599. }
  600. expected := defaultSchedulerCountBasedMinInterval
  601. result := opts.SchedulerCountBasedMinInterval()
  602. if result != expected {
  603. t.Fatalf(`Unexpected SCHEDULER_ENTRY_COUNT_BASED_MIN_INTERVAL value, got %v instead of %v`, result, expected)
  604. }
  605. }
  606. func TestDefautSchedulerCountBasedMinInterval(t *testing.T) {
  607. os.Clearenv()
  608. os.Setenv("SCHEDULER_ENTRY_COUNT_BASED_MIN_INTERVAL", "30")
  609. parser := NewParser()
  610. opts, err := parser.ParseEnvironmentVariables()
  611. if err != nil {
  612. t.Fatalf(`Parsing failure: %v`, err)
  613. }
  614. expected := 30
  615. result := opts.SchedulerCountBasedMinInterval()
  616. if result != expected {
  617. t.Fatalf(`Unexpected SCHEDULER_ENTRY_COUNT_BASED_MIN_INTERVAL value, got %v instead of %v`, result, expected)
  618. }
  619. }
  620. func TestOAuth2UserCreationWhenUnset(t *testing.T) {
  621. os.Clearenv()
  622. parser := NewParser()
  623. opts, err := parser.ParseEnvironmentVariables()
  624. if err != nil {
  625. t.Fatalf(`Parsing failure: %v`, err)
  626. }
  627. expected := false
  628. result := opts.IsOAuth2UserCreationAllowed()
  629. if result != expected {
  630. t.Fatalf(`Unexpected OAUTH2_USER_CREATION value, got %v instead of %v`, result, expected)
  631. }
  632. }
  633. func TestOAuth2UserCreationAdmin(t *testing.T) {
  634. os.Clearenv()
  635. os.Setenv("OAUTH2_USER_CREATION", "1")
  636. parser := NewParser()
  637. opts, err := parser.ParseEnvironmentVariables()
  638. if err != nil {
  639. t.Fatalf(`Parsing failure: %v`, err)
  640. }
  641. expected := true
  642. result := opts.IsOAuth2UserCreationAllowed()
  643. if result != expected {
  644. t.Fatalf(`Unexpected OAUTH2_USER_CREATION value, got %v instead of %v`, result, expected)
  645. }
  646. }
  647. func TestOAuth2ClientID(t *testing.T) {
  648. os.Clearenv()
  649. os.Setenv("OAUTH2_CLIENT_ID", "foobar")
  650. parser := NewParser()
  651. opts, err := parser.ParseEnvironmentVariables()
  652. if err != nil {
  653. t.Fatalf(`Parsing failure: %v`, err)
  654. }
  655. expected := "foobar"
  656. result := opts.OAuth2ClientID()
  657. if result != expected {
  658. t.Fatalf(`Unexpected OAUTH2_CLIENT_ID value, got %q instead of %q`, result, expected)
  659. }
  660. }
  661. func TestDefaultOAuth2ClientIDValue(t *testing.T) {
  662. os.Clearenv()
  663. parser := NewParser()
  664. opts, err := parser.ParseEnvironmentVariables()
  665. if err != nil {
  666. t.Fatalf(`Parsing failure: %v`, err)
  667. }
  668. expected := defaultOAuth2ClientID
  669. result := opts.OAuth2ClientID()
  670. if result != expected {
  671. t.Fatalf(`Unexpected OAUTH2_CLIENT_ID value, got %q instead of %q`, result, expected)
  672. }
  673. }
  674. func TestOAuth2ClientSecret(t *testing.T) {
  675. os.Clearenv()
  676. os.Setenv("OAUTH2_CLIENT_SECRET", "secret")
  677. parser := NewParser()
  678. opts, err := parser.ParseEnvironmentVariables()
  679. if err != nil {
  680. t.Fatalf(`Parsing failure: %v`, err)
  681. }
  682. expected := "secret"
  683. result := opts.OAuth2ClientSecret()
  684. if result != expected {
  685. t.Fatalf(`Unexpected OAUTH2_CLIENT_SECRET value, got %q instead of %q`, result, expected)
  686. }
  687. }
  688. func TestDefaultOAuth2ClientSecretValue(t *testing.T) {
  689. os.Clearenv()
  690. parser := NewParser()
  691. opts, err := parser.ParseEnvironmentVariables()
  692. if err != nil {
  693. t.Fatalf(`Parsing failure: %v`, err)
  694. }
  695. expected := defaultOAuth2ClientSecret
  696. result := opts.OAuth2ClientSecret()
  697. if result != expected {
  698. t.Fatalf(`Unexpected OAUTH2_CLIENT_SECRET value, got %q instead of %q`, result, expected)
  699. }
  700. }
  701. func TestOAuth2RedirectURL(t *testing.T) {
  702. os.Clearenv()
  703. os.Setenv("OAUTH2_REDIRECT_URL", "http://example.org")
  704. parser := NewParser()
  705. opts, err := parser.ParseEnvironmentVariables()
  706. if err != nil {
  707. t.Fatalf(`Parsing failure: %v`, err)
  708. }
  709. expected := "http://example.org"
  710. result := opts.OAuth2RedirectURL()
  711. if result != expected {
  712. t.Fatalf(`Unexpected OAUTH2_REDIRECT_URL value, got %q instead of %q`, result, expected)
  713. }
  714. }
  715. func TestDefaultOAuth2RedirectURLValue(t *testing.T) {
  716. os.Clearenv()
  717. parser := NewParser()
  718. opts, err := parser.ParseEnvironmentVariables()
  719. if err != nil {
  720. t.Fatalf(`Parsing failure: %v`, err)
  721. }
  722. expected := defaultOAuth2RedirectURL
  723. result := opts.OAuth2RedirectURL()
  724. if result != expected {
  725. t.Fatalf(`Unexpected OAUTH2_REDIRECT_URL value, got %q instead of %q`, result, expected)
  726. }
  727. }
  728. func TestOAuth2OidcDiscoveryEndpoint(t *testing.T) {
  729. os.Clearenv()
  730. os.Setenv("OAUTH2_OIDC_DISCOVERY_ENDPOINT", "http://example.org")
  731. parser := NewParser()
  732. opts, err := parser.ParseEnvironmentVariables()
  733. if err != nil {
  734. t.Fatalf(`Parsing failure: %v`, err)
  735. }
  736. expected := "http://example.org"
  737. result := opts.OAuth2OidcDiscoveryEndpoint()
  738. if result != expected {
  739. t.Fatalf(`Unexpected OAUTH2_OIDC_DISCOVERY_ENDPOINT value, got %q instead of %q`, result, expected)
  740. }
  741. }
  742. func TestDefaultOAuth2OidcDiscoveryEndpointValue(t *testing.T) {
  743. os.Clearenv()
  744. parser := NewParser()
  745. opts, err := parser.ParseEnvironmentVariables()
  746. if err != nil {
  747. t.Fatalf(`Parsing failure: %v`, err)
  748. }
  749. expected := defaultOAuth2OidcDiscoveryEndpoint
  750. result := opts.OAuth2OidcDiscoveryEndpoint()
  751. if result != expected {
  752. t.Fatalf(`Unexpected OAUTH2_REDIRECT_URL value, got %q instead of %q`, result, expected)
  753. }
  754. }
  755. func TestOAuth2Provider(t *testing.T) {
  756. os.Clearenv()
  757. os.Setenv("OAUTH2_PROVIDER", "google")
  758. parser := NewParser()
  759. opts, err := parser.ParseEnvironmentVariables()
  760. if err != nil {
  761. t.Fatalf(`Parsing failure: %v`, err)
  762. }
  763. expected := "google"
  764. result := opts.OAuth2Provider()
  765. if result != expected {
  766. t.Fatalf(`Unexpected OAUTH2_PROVIDER value, got %q instead of %q`, result, expected)
  767. }
  768. }
  769. func TestDefaultOAuth2ProviderValue(t *testing.T) {
  770. os.Clearenv()
  771. parser := NewParser()
  772. opts, err := parser.ParseEnvironmentVariables()
  773. if err != nil {
  774. t.Fatalf(`Parsing failure: %v`, err)
  775. }
  776. expected := defaultOAuth2Provider
  777. result := opts.OAuth2Provider()
  778. if result != expected {
  779. t.Fatalf(`Unexpected OAUTH2_PROVIDER value, got %q instead of %q`, result, expected)
  780. }
  781. }
  782. func TestHSTSWhenUnset(t *testing.T) {
  783. os.Clearenv()
  784. parser := NewParser()
  785. opts, err := parser.ParseEnvironmentVariables()
  786. if err != nil {
  787. t.Fatalf(`Parsing failure: %v`, err)
  788. }
  789. expected := true
  790. result := opts.HasHSTS()
  791. if result != expected {
  792. t.Fatalf(`Unexpected DISABLE_HSTS value, got %v instead of %v`, result, expected)
  793. }
  794. }
  795. func TestHSTS(t *testing.T) {
  796. os.Clearenv()
  797. os.Setenv("DISABLE_HSTS", "1")
  798. parser := NewParser()
  799. opts, err := parser.ParseEnvironmentVariables()
  800. if err != nil {
  801. t.Fatalf(`Parsing failure: %v`, err)
  802. }
  803. expected := false
  804. result := opts.HasHSTS()
  805. if result != expected {
  806. t.Fatalf(`Unexpected DISABLE_HSTS value, got %v instead of %v`, result, expected)
  807. }
  808. }
  809. func TestDisableHTTPServiceWhenUnset(t *testing.T) {
  810. os.Clearenv()
  811. parser := NewParser()
  812. opts, err := parser.ParseEnvironmentVariables()
  813. if err != nil {
  814. t.Fatalf(`Parsing failure: %v`, err)
  815. }
  816. expected := true
  817. result := opts.HasHTTPService()
  818. if result != expected {
  819. t.Fatalf(`Unexpected DISABLE_HTTP_SERVICE value, got %v instead of %v`, result, expected)
  820. }
  821. }
  822. func TestDisableHTTPService(t *testing.T) {
  823. os.Clearenv()
  824. os.Setenv("DISABLE_HTTP_SERVICE", "1")
  825. parser := NewParser()
  826. opts, err := parser.ParseEnvironmentVariables()
  827. if err != nil {
  828. t.Fatalf(`Parsing failure: %v`, err)
  829. }
  830. expected := false
  831. result := opts.HasHTTPService()
  832. if result != expected {
  833. t.Fatalf(`Unexpected DISABLE_HTTP_SERVICE value, got %v instead of %v`, result, expected)
  834. }
  835. }
  836. func TestDisableSchedulerServiceWhenUnset(t *testing.T) {
  837. os.Clearenv()
  838. parser := NewParser()
  839. opts, err := parser.ParseEnvironmentVariables()
  840. if err != nil {
  841. t.Fatalf(`Parsing failure: %v`, err)
  842. }
  843. expected := true
  844. result := opts.HasSchedulerService()
  845. if result != expected {
  846. t.Fatalf(`Unexpected DISABLE_SCHEDULER_SERVICE value, got %v instead of %v`, result, expected)
  847. }
  848. }
  849. func TestDisableSchedulerService(t *testing.T) {
  850. os.Clearenv()
  851. os.Setenv("DISABLE_SCHEDULER_SERVICE", "1")
  852. parser := NewParser()
  853. opts, err := parser.ParseEnvironmentVariables()
  854. if err != nil {
  855. t.Fatalf(`Parsing failure: %v`, err)
  856. }
  857. expected := false
  858. result := opts.HasSchedulerService()
  859. if result != expected {
  860. t.Fatalf(`Unexpected DISABLE_SCHEDULER_SERVICE value, got %v instead of %v`, result, expected)
  861. }
  862. }
  863. func TestRunMigrationsWhenUnset(t *testing.T) {
  864. os.Clearenv()
  865. parser := NewParser()
  866. opts, err := parser.ParseEnvironmentVariables()
  867. if err != nil {
  868. t.Fatalf(`Parsing failure: %v`, err)
  869. }
  870. expected := false
  871. result := opts.RunMigrations()
  872. if result != expected {
  873. t.Fatalf(`Unexpected RUN_MIGRATIONS value, got %v instead of %v`, result, expected)
  874. }
  875. }
  876. func TestRunMigrations(t *testing.T) {
  877. os.Clearenv()
  878. os.Setenv("RUN_MIGRATIONS", "yes")
  879. parser := NewParser()
  880. opts, err := parser.ParseEnvironmentVariables()
  881. if err != nil {
  882. t.Fatalf(`Parsing failure: %v`, err)
  883. }
  884. expected := true
  885. result := opts.RunMigrations()
  886. if result != expected {
  887. t.Fatalf(`Unexpected RUN_MIGRATIONS value, got %v instead of %v`, result, expected)
  888. }
  889. }
  890. func TestCreateAdminWhenUnset(t *testing.T) {
  891. os.Clearenv()
  892. parser := NewParser()
  893. opts, err := parser.ParseEnvironmentVariables()
  894. if err != nil {
  895. t.Fatalf(`Parsing failure: %v`, err)
  896. }
  897. expected := false
  898. result := opts.CreateAdmin()
  899. if result != expected {
  900. t.Fatalf(`Unexpected CREATE_ADMIN value, got %v instead of %v`, result, expected)
  901. }
  902. }
  903. func TestCreateAdmin(t *testing.T) {
  904. os.Clearenv()
  905. os.Setenv("CREATE_ADMIN", "true")
  906. parser := NewParser()
  907. opts, err := parser.ParseEnvironmentVariables()
  908. if err != nil {
  909. t.Fatalf(`Parsing failure: %v`, err)
  910. }
  911. expected := true
  912. result := opts.CreateAdmin()
  913. if result != expected {
  914. t.Fatalf(`Unexpected CREATE_ADMIN value, got %v instead of %v`, result, expected)
  915. }
  916. }
  917. func TestPocketConsumerKeyFromEnvVariable(t *testing.T) {
  918. os.Clearenv()
  919. os.Setenv("POCKET_CONSUMER_KEY", "something")
  920. parser := NewParser()
  921. opts, err := parser.ParseEnvironmentVariables()
  922. if err != nil {
  923. t.Fatalf(`Parsing failure: %v`, err)
  924. }
  925. expected := "something"
  926. result := opts.PocketConsumerKey("default")
  927. if result != expected {
  928. t.Fatalf(`Unexpected POCKET_CONSUMER_KEY value, got %q instead of %q`, result, expected)
  929. }
  930. }
  931. func TestPocketConsumerKeyFromUserPrefs(t *testing.T) {
  932. os.Clearenv()
  933. parser := NewParser()
  934. opts, err := parser.ParseEnvironmentVariables()
  935. if err != nil {
  936. t.Fatalf(`Parsing failure: %v`, err)
  937. }
  938. expected := "default"
  939. result := opts.PocketConsumerKey("default")
  940. if result != expected {
  941. t.Fatalf(`Unexpected POCKET_CONSUMER_KEY value, got %q instead of %q`, result, expected)
  942. }
  943. }
  944. func TestProxyImages(t *testing.T) {
  945. os.Clearenv()
  946. os.Setenv("PROXY_IMAGES", "all")
  947. parser := NewParser()
  948. opts, err := parser.ParseEnvironmentVariables()
  949. if err != nil {
  950. t.Fatalf(`Parsing failure: %v`, err)
  951. }
  952. expected := "all"
  953. result := opts.ProxyImages()
  954. if result != expected {
  955. t.Fatalf(`Unexpected PROXY_IMAGES value, got %q instead of %q`, result, expected)
  956. }
  957. }
  958. func TestDefaultProxyImagesValue(t *testing.T) {
  959. os.Clearenv()
  960. parser := NewParser()
  961. opts, err := parser.ParseEnvironmentVariables()
  962. if err != nil {
  963. t.Fatalf(`Parsing failure: %v`, err)
  964. }
  965. expected := defaultProxyImages
  966. result := opts.ProxyImages()
  967. if result != expected {
  968. t.Fatalf(`Unexpected PROXY_IMAGES value, got %q instead of %q`, result, expected)
  969. }
  970. }
  971. func TestHTTPSOff(t *testing.T) {
  972. os.Clearenv()
  973. parser := NewParser()
  974. opts, err := parser.ParseEnvironmentVariables()
  975. if err != nil {
  976. t.Fatalf(`Parsing failure: %v`, err)
  977. }
  978. if opts.HTTPS {
  979. t.Fatalf(`Unexpected HTTPS value, got "%v"`, opts.HTTPS)
  980. }
  981. }
  982. func TestHTTPSOn(t *testing.T) {
  983. os.Clearenv()
  984. os.Setenv("HTTPS", "on")
  985. parser := NewParser()
  986. opts, err := parser.ParseEnvironmentVariables()
  987. if err != nil {
  988. t.Fatalf(`Parsing failure: %v`, err)
  989. }
  990. if !opts.HTTPS {
  991. t.Fatalf(`Unexpected HTTPS value, got "%v"`, opts.HTTPS)
  992. }
  993. }
  994. func TestHTTPClientTimeout(t *testing.T) {
  995. os.Clearenv()
  996. os.Setenv("HTTP_CLIENT_TIMEOUT", "42")
  997. parser := NewParser()
  998. opts, err := parser.ParseEnvironmentVariables()
  999. if err != nil {
  1000. t.Fatalf(`Parsing failure: %v`, err)
  1001. }
  1002. expected := 42
  1003. result := opts.HTTPClientTimeout()
  1004. if result != expected {
  1005. t.Fatalf(`Unexpected HTTP_CLIENT_TIMEOUT value, got %d instead of %d`, result, expected)
  1006. }
  1007. }
  1008. func TestDefaultHTTPClientTimeoutValue(t *testing.T) {
  1009. os.Clearenv()
  1010. parser := NewParser()
  1011. opts, err := parser.ParseEnvironmentVariables()
  1012. if err != nil {
  1013. t.Fatalf(`Parsing failure: %v`, err)
  1014. }
  1015. expected := defaultHTTPClientTimeout
  1016. result := opts.HTTPClientTimeout()
  1017. if result != expected {
  1018. t.Fatalf(`Unexpected HTTP_CLIENT_TIMEOUT value, got %d instead of %d`, result, expected)
  1019. }
  1020. }
  1021. func TestHTTPClientMaxBodySize(t *testing.T) {
  1022. os.Clearenv()
  1023. os.Setenv("HTTP_CLIENT_MAX_BODY_SIZE", "42")
  1024. parser := NewParser()
  1025. opts, err := parser.ParseEnvironmentVariables()
  1026. if err != nil {
  1027. t.Fatalf(`Parsing failure: %v`, err)
  1028. }
  1029. expected := int64(42 * 1024 * 1024)
  1030. result := opts.HTTPClientMaxBodySize()
  1031. if result != expected {
  1032. t.Fatalf(`Unexpected HTTP_CLIENT_MAX_BODY_SIZE value, got %d instead of %d`, result, expected)
  1033. }
  1034. }
  1035. func TestDefaultHTTPClientMaxBodySizeValue(t *testing.T) {
  1036. os.Clearenv()
  1037. parser := NewParser()
  1038. opts, err := parser.ParseEnvironmentVariables()
  1039. if err != nil {
  1040. t.Fatalf(`Parsing failure: %v`, err)
  1041. }
  1042. expected := int64(defaultHTTPClientMaxBodySize * 1024 * 1024)
  1043. result := opts.HTTPClientMaxBodySize()
  1044. if result != expected {
  1045. t.Fatalf(`Unexpected HTTP_CLIENT_MAX_BODY_SIZE value, got %d instead of %d`, result, expected)
  1046. }
  1047. }
  1048. func TestParseConfigFile(t *testing.T) {
  1049. content := []byte(`
  1050. # This is a comment
  1051. DEBUG = yes
  1052. POCKET_CONSUMER_KEY= >#1234
  1053. Invalid text
  1054. `)
  1055. tmpfile, err := ioutil.TempFile(".", "miniflux.*.unit_test.conf")
  1056. if err != nil {
  1057. t.Fatal(err)
  1058. }
  1059. if _, err := tmpfile.Write(content); err != nil {
  1060. t.Fatal(err)
  1061. }
  1062. os.Clearenv()
  1063. parser := NewParser()
  1064. opts, err := parser.ParseFile(tmpfile.Name())
  1065. if err != nil {
  1066. t.Errorf(`Parsing failure: %v`, err)
  1067. }
  1068. if opts.HasDebugMode() != true {
  1069. t.Errorf(`Unexpected debug mode value, got "%v"`, opts.HasDebugMode())
  1070. }
  1071. expected := ">#1234"
  1072. result := opts.PocketConsumerKey("default")
  1073. if result != expected {
  1074. t.Errorf(`Unexpected POCKET_CONSUMER_KEY value, got %q instead of %q`, result, expected)
  1075. }
  1076. if err := tmpfile.Close(); err != nil {
  1077. t.Fatal(err)
  1078. }
  1079. if err := os.Remove(tmpfile.Name()); err != nil {
  1080. t.Fatal(err)
  1081. }
  1082. }
  1083. func TestAuthProxyHeader(t *testing.T) {
  1084. os.Clearenv()
  1085. os.Setenv("AUTH_PROXY_HEADER", "X-Forwarded-User")
  1086. parser := NewParser()
  1087. opts, err := parser.ParseEnvironmentVariables()
  1088. if err != nil {
  1089. t.Fatalf(`Parsing failure: %v`, err)
  1090. }
  1091. expected := "X-Forwarded-User"
  1092. result := opts.AuthProxyHeader()
  1093. if result != expected {
  1094. t.Fatalf(`Unexpected AUTH_PROXY_HEADER value, got %q instead of %q`, result, expected)
  1095. }
  1096. }
  1097. func TestDefaultAuthProxyHeaderValue(t *testing.T) {
  1098. os.Clearenv()
  1099. parser := NewParser()
  1100. opts, err := parser.ParseEnvironmentVariables()
  1101. if err != nil {
  1102. t.Fatalf(`Parsing failure: %v`, err)
  1103. }
  1104. expected := defaultAuthProxyHeader
  1105. result := opts.AuthProxyHeader()
  1106. if result != expected {
  1107. t.Fatalf(`Unexpected AUTH_PROXY_HEADER value, got %q instead of %q`, result, expected)
  1108. }
  1109. }
  1110. func TestAuthProxyUserCreationWhenUnset(t *testing.T) {
  1111. os.Clearenv()
  1112. parser := NewParser()
  1113. opts, err := parser.ParseEnvironmentVariables()
  1114. if err != nil {
  1115. t.Fatalf(`Parsing failure: %v`, err)
  1116. }
  1117. expected := false
  1118. result := opts.IsAuthProxyUserCreationAllowed()
  1119. if result != expected {
  1120. t.Fatalf(`Unexpected AUTH_PROXY_USER_CREATION value, got %v instead of %v`, result, expected)
  1121. }
  1122. }
  1123. func TestAuthProxyUserCreationAdmin(t *testing.T) {
  1124. os.Clearenv()
  1125. os.Setenv("AUTH_PROXY_USER_CREATION", "1")
  1126. parser := NewParser()
  1127. opts, err := parser.ParseEnvironmentVariables()
  1128. if err != nil {
  1129. t.Fatalf(`Parsing failure: %v`, err)
  1130. }
  1131. expected := true
  1132. result := opts.IsAuthProxyUserCreationAllowed()
  1133. if result != expected {
  1134. t.Fatalf(`Unexpected AUTH_PROXY_USER_CREATION value, got %v instead of %v`, result, expected)
  1135. }
  1136. }