processing_options_test.go 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608
  1. package main
  2. import (
  3. "context"
  4. "encoding/base64"
  5. "fmt"
  6. "net/http"
  7. "net/url"
  8. "testing"
  9. "github.com/stretchr/testify/assert"
  10. "github.com/stretchr/testify/require"
  11. "github.com/stretchr/testify/suite"
  12. )
  13. type ProcessingOptionsTestSuite struct{ MainTestSuite }
  14. func (s *ProcessingOptionsTestSuite) getRequest(uri string) *http.Request {
  15. return &http.Request{Method: "GET", RequestURI: uri, Header: make(http.Header)}
  16. }
  17. func (s *ProcessingOptionsTestSuite) TestParseBase64URL() {
  18. imageURL := "http://images.dev/lorem/ipsum.jpg?param=value"
  19. req := s.getRequest(fmt.Sprintf("/unsafe/size:100:100/%s.png", base64.RawURLEncoding.EncodeToString([]byte(imageURL))))
  20. ctx, err := parsePath(context.Background(), req)
  21. require.Nil(s.T(), err)
  22. assert.Equal(s.T(), imageURL, getImageURL(ctx))
  23. assert.Equal(s.T(), imageTypePNG, getProcessingOptions(ctx).Format)
  24. }
  25. func (s *ProcessingOptionsTestSuite) TestParseBase64URLWithoutExtension() {
  26. imageURL := "http://images.dev/lorem/ipsum.jpg?param=value"
  27. req := s.getRequest(fmt.Sprintf("/unsafe/size:100:100/%s", base64.RawURLEncoding.EncodeToString([]byte(imageURL))))
  28. ctx, err := parsePath(context.Background(), req)
  29. require.Nil(s.T(), err)
  30. assert.Equal(s.T(), imageURL, getImageURL(ctx))
  31. assert.Equal(s.T(), imageTypeUnknown, getProcessingOptions(ctx).Format)
  32. }
  33. func (s *ProcessingOptionsTestSuite) TestParseBase64URLWithBase() {
  34. conf.BaseURL = "http://images.dev/"
  35. imageURL := "lorem/ipsum.jpg?param=value"
  36. req := s.getRequest(fmt.Sprintf("/unsafe/size:100:100/%s.png", base64.RawURLEncoding.EncodeToString([]byte(imageURL))))
  37. ctx, err := parsePath(context.Background(), req)
  38. require.Nil(s.T(), err)
  39. assert.Equal(s.T(), fmt.Sprintf("%s%s", conf.BaseURL, imageURL), getImageURL(ctx))
  40. assert.Equal(s.T(), imageTypePNG, getProcessingOptions(ctx).Format)
  41. }
  42. func (s *ProcessingOptionsTestSuite) TestParsePlainURL() {
  43. imageURL := "http://images.dev/lorem/ipsum.jpg"
  44. req := s.getRequest(fmt.Sprintf("/unsafe/size:100:100/plain/%s@png", imageURL))
  45. ctx, err := parsePath(context.Background(), req)
  46. require.Nil(s.T(), err)
  47. assert.Equal(s.T(), imageURL, getImageURL(ctx))
  48. assert.Equal(s.T(), imageTypePNG, getProcessingOptions(ctx).Format)
  49. }
  50. func (s *ProcessingOptionsTestSuite) TestParsePlainURLWithoutExtension() {
  51. imageURL := "http://images.dev/lorem/ipsum.jpg"
  52. req := s.getRequest(fmt.Sprintf("/unsafe/size:100:100/plain/%s", imageURL))
  53. ctx, err := parsePath(context.Background(), req)
  54. require.Nil(s.T(), err)
  55. assert.Equal(s.T(), imageURL, getImageURL(ctx))
  56. assert.Equal(s.T(), imageTypeUnknown, getProcessingOptions(ctx).Format)
  57. }
  58. func (s *ProcessingOptionsTestSuite) TestParsePlainURLEscaped() {
  59. imageURL := "http://images.dev/lorem/ipsum.jpg?param=value"
  60. req := s.getRequest(fmt.Sprintf("/unsafe/size:100:100/plain/%s@png", url.PathEscape(imageURL)))
  61. ctx, err := parsePath(context.Background(), req)
  62. require.Nil(s.T(), err)
  63. assert.Equal(s.T(), imageURL, getImageURL(ctx))
  64. assert.Equal(s.T(), imageTypePNG, getProcessingOptions(ctx).Format)
  65. }
  66. func (s *ProcessingOptionsTestSuite) TestParsePlainURLWithBase() {
  67. conf.BaseURL = "http://images.dev/"
  68. imageURL := "lorem/ipsum.jpg"
  69. req := s.getRequest(fmt.Sprintf("/unsafe/size:100:100/plain/%s@png", imageURL))
  70. ctx, err := parsePath(context.Background(), req)
  71. require.Nil(s.T(), err)
  72. assert.Equal(s.T(), fmt.Sprintf("%s%s", conf.BaseURL, imageURL), getImageURL(ctx))
  73. assert.Equal(s.T(), imageTypePNG, getProcessingOptions(ctx).Format)
  74. }
  75. func (s *ProcessingOptionsTestSuite) TestParsePlainURLEscapedWithBase() {
  76. conf.BaseURL = "http://images.dev/"
  77. imageURL := "lorem/ipsum.jpg?param=value"
  78. req := s.getRequest(fmt.Sprintf("/unsafe/size:100:100/plain/%s@png", url.PathEscape(imageURL)))
  79. ctx, err := parsePath(context.Background(), req)
  80. require.Nil(s.T(), err)
  81. assert.Equal(s.T(), fmt.Sprintf("%s%s", conf.BaseURL, imageURL), getImageURL(ctx))
  82. assert.Equal(s.T(), imageTypePNG, getProcessingOptions(ctx).Format)
  83. }
  84. func (s *ProcessingOptionsTestSuite) TestParseURLAllowedSource() {
  85. conf.AllowedSources = []string{"local://", "http://images.dev/"}
  86. req := s.getRequest("/unsafe/plain/http://images.dev/lorem/ipsum.jpg")
  87. _, err := parsePath(context.Background(), req)
  88. require.Nil(s.T(), err)
  89. }
  90. func (s *ProcessingOptionsTestSuite) TestParseURLNotAllowedSource() {
  91. conf.AllowedSources = []string{"local://", "http://images.dev/"}
  92. req := s.getRequest("/unsafe/plain/s3://images/lorem/ipsum.jpg")
  93. _, err := parsePath(context.Background(), req)
  94. require.Error(s.T(), err)
  95. }
  96. func (s *ProcessingOptionsTestSuite) TestParsePathBasic() {
  97. req := s.getRequest("/unsafe/fill/100/200/noea/1/plain/http://images.dev/lorem/ipsum.jpg@png")
  98. ctx, err := parsePath(context.Background(), req)
  99. require.Nil(s.T(), err)
  100. po := getProcessingOptions(ctx)
  101. assert.Equal(s.T(), resizeFill, po.ResizingType)
  102. assert.Equal(s.T(), 100, po.Width)
  103. assert.Equal(s.T(), 200, po.Height)
  104. assert.Equal(s.T(), gravityNorthEast, po.Gravity.Type)
  105. assert.True(s.T(), po.Enlarge)
  106. assert.Equal(s.T(), imageTypePNG, po.Format)
  107. }
  108. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedFormat() {
  109. req := s.getRequest("/unsafe/format:webp/plain/http://images.dev/lorem/ipsum.jpg")
  110. ctx, err := parsePath(context.Background(), req)
  111. require.Nil(s.T(), err)
  112. po := getProcessingOptions(ctx)
  113. assert.Equal(s.T(), imageTypeWEBP, po.Format)
  114. }
  115. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedResize() {
  116. req := s.getRequest("/unsafe/resize:fill:100:200:1/plain/http://images.dev/lorem/ipsum.jpg")
  117. ctx, err := parsePath(context.Background(), req)
  118. require.Nil(s.T(), err)
  119. po := getProcessingOptions(ctx)
  120. assert.Equal(s.T(), resizeFill, po.ResizingType)
  121. assert.Equal(s.T(), 100, po.Width)
  122. assert.Equal(s.T(), 200, po.Height)
  123. assert.True(s.T(), po.Enlarge)
  124. }
  125. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedResizingType() {
  126. req := s.getRequest("/unsafe/resizing_type:fill/plain/http://images.dev/lorem/ipsum.jpg")
  127. ctx, err := parsePath(context.Background(), req)
  128. require.Nil(s.T(), err)
  129. po := getProcessingOptions(ctx)
  130. assert.Equal(s.T(), resizeFill, po.ResizingType)
  131. }
  132. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedSize() {
  133. req := s.getRequest("/unsafe/size:100:200:1/plain/http://images.dev/lorem/ipsum.jpg")
  134. ctx, err := parsePath(context.Background(), req)
  135. require.Nil(s.T(), err)
  136. po := getProcessingOptions(ctx)
  137. assert.Equal(s.T(), 100, po.Width)
  138. assert.Equal(s.T(), 200, po.Height)
  139. assert.True(s.T(), po.Enlarge)
  140. }
  141. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedWidth() {
  142. req := s.getRequest("/unsafe/width:100/plain/http://images.dev/lorem/ipsum.jpg")
  143. ctx, err := parsePath(context.Background(), req)
  144. require.Nil(s.T(), err)
  145. po := getProcessingOptions(ctx)
  146. assert.Equal(s.T(), 100, po.Width)
  147. }
  148. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedHeight() {
  149. req := s.getRequest("/unsafe/height:100/plain/http://images.dev/lorem/ipsum.jpg")
  150. ctx, err := parsePath(context.Background(), req)
  151. require.Nil(s.T(), err)
  152. po := getProcessingOptions(ctx)
  153. assert.Equal(s.T(), 100, po.Height)
  154. }
  155. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedEnlarge() {
  156. req := s.getRequest("/unsafe/enlarge:1/plain/http://images.dev/lorem/ipsum.jpg")
  157. ctx, err := parsePath(context.Background(), req)
  158. require.Nil(s.T(), err)
  159. po := getProcessingOptions(ctx)
  160. assert.True(s.T(), po.Enlarge)
  161. }
  162. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedExtend() {
  163. req := s.getRequest("/unsafe/extend:1:so:10:20/plain/http://images.dev/lorem/ipsum.jpg")
  164. ctx, err := parsePath(context.Background(), req)
  165. require.Nil(s.T(), err)
  166. po := getProcessingOptions(ctx)
  167. assert.Equal(s.T(), true, po.Extend.Enabled)
  168. assert.Equal(s.T(), gravitySouth, po.Extend.Gravity.Type)
  169. assert.Equal(s.T(), 10.0, po.Extend.Gravity.X)
  170. assert.Equal(s.T(), 20.0, po.Extend.Gravity.Y)
  171. }
  172. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedGravity() {
  173. req := s.getRequest("/unsafe/gravity:soea/plain/http://images.dev/lorem/ipsum.jpg")
  174. ctx, err := parsePath(context.Background(), req)
  175. require.Nil(s.T(), err)
  176. po := getProcessingOptions(ctx)
  177. assert.Equal(s.T(), gravitySouthEast, po.Gravity.Type)
  178. }
  179. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedGravityFocuspoint() {
  180. req := s.getRequest("/unsafe/gravity:fp:0.5:0.75/plain/http://images.dev/lorem/ipsum.jpg")
  181. ctx, err := parsePath(context.Background(), req)
  182. require.Nil(s.T(), err)
  183. po := getProcessingOptions(ctx)
  184. assert.Equal(s.T(), gravityFocusPoint, po.Gravity.Type)
  185. assert.Equal(s.T(), 0.5, po.Gravity.X)
  186. assert.Equal(s.T(), 0.75, po.Gravity.Y)
  187. }
  188. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedQuality() {
  189. req := s.getRequest("/unsafe/quality:55/plain/http://images.dev/lorem/ipsum.jpg")
  190. ctx, err := parsePath(context.Background(), req)
  191. require.Nil(s.T(), err)
  192. po := getProcessingOptions(ctx)
  193. assert.Equal(s.T(), 55, po.Quality)
  194. }
  195. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedBackground() {
  196. req := s.getRequest("/unsafe/background:128:129:130/plain/http://images.dev/lorem/ipsum.jpg")
  197. ctx, err := parsePath(context.Background(), req)
  198. require.Nil(s.T(), err)
  199. po := getProcessingOptions(ctx)
  200. assert.True(s.T(), po.Flatten)
  201. assert.Equal(s.T(), uint8(128), po.Background.R)
  202. assert.Equal(s.T(), uint8(129), po.Background.G)
  203. assert.Equal(s.T(), uint8(130), po.Background.B)
  204. }
  205. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedBackgroundHex() {
  206. req := s.getRequest("/unsafe/background:ffddee/plain/http://images.dev/lorem/ipsum.jpg")
  207. ctx, err := parsePath(context.Background(), req)
  208. require.Nil(s.T(), err)
  209. po := getProcessingOptions(ctx)
  210. assert.True(s.T(), po.Flatten)
  211. assert.Equal(s.T(), uint8(0xff), po.Background.R)
  212. assert.Equal(s.T(), uint8(0xdd), po.Background.G)
  213. assert.Equal(s.T(), uint8(0xee), po.Background.B)
  214. }
  215. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedBackgroundDisable() {
  216. req := s.getRequest("/unsafe/background:fff/background:/plain/http://images.dev/lorem/ipsum.jpg")
  217. ctx, err := parsePath(context.Background(), req)
  218. require.Nil(s.T(), err)
  219. po := getProcessingOptions(ctx)
  220. assert.False(s.T(), po.Flatten)
  221. }
  222. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedBlur() {
  223. req := s.getRequest("/unsafe/blur:0.2/plain/http://images.dev/lorem/ipsum.jpg")
  224. ctx, err := parsePath(context.Background(), req)
  225. require.Nil(s.T(), err)
  226. po := getProcessingOptions(ctx)
  227. assert.Equal(s.T(), float32(0.2), po.Blur)
  228. }
  229. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedSharpen() {
  230. req := s.getRequest("/unsafe/sharpen:0.2/plain/http://images.dev/lorem/ipsum.jpg")
  231. ctx, err := parsePath(context.Background(), req)
  232. require.Nil(s.T(), err)
  233. po := getProcessingOptions(ctx)
  234. assert.Equal(s.T(), float32(0.2), po.Sharpen)
  235. }
  236. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedDpr() {
  237. req := s.getRequest("/unsafe/dpr:2/plain/http://images.dev/lorem/ipsum.jpg")
  238. ctx, err := parsePath(context.Background(), req)
  239. require.Nil(s.T(), err)
  240. po := getProcessingOptions(ctx)
  241. assert.Equal(s.T(), 2.0, po.Dpr)
  242. }
  243. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedWatermark() {
  244. req := s.getRequest("/unsafe/watermark:0.5:soea:10:20:0.6/plain/http://images.dev/lorem/ipsum.jpg")
  245. ctx, err := parsePath(context.Background(), req)
  246. require.Nil(s.T(), err)
  247. po := getProcessingOptions(ctx)
  248. assert.True(s.T(), po.Watermark.Enabled)
  249. assert.Equal(s.T(), gravitySouthEast, po.Watermark.Gravity.Type)
  250. assert.Equal(s.T(), 10.0, po.Watermark.Gravity.X)
  251. assert.Equal(s.T(), 20.0, po.Watermark.Gravity.Y)
  252. assert.Equal(s.T(), 0.6, po.Watermark.Scale)
  253. }
  254. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedPreset() {
  255. conf.Presets["test1"] = urlOptions{
  256. urlOption{Name: "resizing_type", Args: []string{"fill"}},
  257. }
  258. conf.Presets["test2"] = urlOptions{
  259. urlOption{Name: "blur", Args: []string{"0.2"}},
  260. urlOption{Name: "quality", Args: []string{"50"}},
  261. }
  262. req := s.getRequest("/unsafe/preset:test1:test2/plain/http://images.dev/lorem/ipsum.jpg")
  263. ctx, err := parsePath(context.Background(), req)
  264. require.Nil(s.T(), err)
  265. po := getProcessingOptions(ctx)
  266. assert.Equal(s.T(), resizeFill, po.ResizingType)
  267. assert.Equal(s.T(), float32(0.2), po.Blur)
  268. assert.Equal(s.T(), 50, po.Quality)
  269. }
  270. func (s *ProcessingOptionsTestSuite) TestParsePathPresetDefault() {
  271. conf.Presets["default"] = urlOptions{
  272. urlOption{Name: "resizing_type", Args: []string{"fill"}},
  273. urlOption{Name: "blur", Args: []string{"0.2"}},
  274. urlOption{Name: "quality", Args: []string{"50"}},
  275. }
  276. req := s.getRequest("/unsafe/quality:70/plain/http://images.dev/lorem/ipsum.jpg")
  277. ctx, err := parsePath(context.Background(), req)
  278. require.Nil(s.T(), err)
  279. po := getProcessingOptions(ctx)
  280. assert.Equal(s.T(), resizeFill, po.ResizingType)
  281. assert.Equal(s.T(), float32(0.2), po.Blur)
  282. assert.Equal(s.T(), 70, po.Quality)
  283. }
  284. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedPresetLoopDetection() {
  285. conf.Presets["test1"] = urlOptions{
  286. urlOption{Name: "resizing_type", Args: []string{"fill"}},
  287. }
  288. conf.Presets["test2"] = urlOptions{
  289. urlOption{Name: "blur", Args: []string{"0.2"}},
  290. urlOption{Name: "quality", Args: []string{"50"}},
  291. }
  292. req := s.getRequest("/unsafe/preset:test1:test2:test1/plain/http://images.dev/lorem/ipsum.jpg")
  293. ctx, err := parsePath(context.Background(), req)
  294. require.Nil(s.T(), err)
  295. po := getProcessingOptions(ctx)
  296. require.ElementsMatch(s.T(), po.UsedPresets, []string{"test1", "test2"})
  297. }
  298. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedCachebuster() {
  299. req := s.getRequest("/unsafe/cachebuster:123/plain/http://images.dev/lorem/ipsum.jpg")
  300. ctx, err := parsePath(context.Background(), req)
  301. require.Nil(s.T(), err)
  302. po := getProcessingOptions(ctx)
  303. assert.Equal(s.T(), "123", po.CacheBuster)
  304. }
  305. func (s *ProcessingOptionsTestSuite) TestParsePathAdvancedStripMetadata() {
  306. req := s.getRequest("/unsafe/strip_metadata:true/plain/http://images.dev/lorem/ipsum.jpg")
  307. ctx, err := parsePath(context.Background(), req)
  308. require.Nil(s.T(), err)
  309. po := getProcessingOptions(ctx)
  310. assert.True(s.T(), po.StripMetadata)
  311. }
  312. func (s *ProcessingOptionsTestSuite) TestParsePathWebpDetection() {
  313. conf.EnableWebpDetection = true
  314. req := s.getRequest("/unsafe/plain/http://images.dev/lorem/ipsum.jpg")
  315. req.Header.Set("Accept", "image/webp")
  316. ctx, err := parsePath(context.Background(), req)
  317. require.Nil(s.T(), err)
  318. po := getProcessingOptions(ctx)
  319. assert.Equal(s.T(), true, po.PreferWebP)
  320. assert.Equal(s.T(), false, po.EnforceWebP)
  321. }
  322. func (s *ProcessingOptionsTestSuite) TestParsePathWebpEnforce() {
  323. conf.EnforceWebp = true
  324. req := s.getRequest("/unsafe/plain/http://images.dev/lorem/ipsum.jpg@png")
  325. req.Header.Set("Accept", "image/webp")
  326. ctx, err := parsePath(context.Background(), req)
  327. require.Nil(s.T(), err)
  328. po := getProcessingOptions(ctx)
  329. assert.Equal(s.T(), true, po.PreferWebP)
  330. assert.Equal(s.T(), true, po.EnforceWebP)
  331. }
  332. func (s *ProcessingOptionsTestSuite) TestParsePathWidthHeader() {
  333. conf.EnableClientHints = true
  334. req := s.getRequest("/unsafe/plain/http://images.dev/lorem/ipsum.jpg@png")
  335. req.Header.Set("Width", "100")
  336. ctx, err := parsePath(context.Background(), req)
  337. require.Nil(s.T(), err)
  338. po := getProcessingOptions(ctx)
  339. assert.Equal(s.T(), 100, po.Width)
  340. }
  341. func (s *ProcessingOptionsTestSuite) TestParsePathWidthHeaderDisabled() {
  342. req := s.getRequest("/unsafe/plain/http://images.dev/lorem/ipsum.jpg@png")
  343. req.Header.Set("Width", "100")
  344. ctx, err := parsePath(context.Background(), req)
  345. require.Nil(s.T(), err)
  346. po := getProcessingOptions(ctx)
  347. assert.Equal(s.T(), 0, po.Width)
  348. }
  349. func (s *ProcessingOptionsTestSuite) TestParsePathWidthHeaderRedefine() {
  350. conf.EnableClientHints = true
  351. req := s.getRequest("/unsafe/width:150/plain/http://images.dev/lorem/ipsum.jpg@png")
  352. req.Header.Set("Width", "100")
  353. ctx, err := parsePath(context.Background(), req)
  354. require.Nil(s.T(), err)
  355. po := getProcessingOptions(ctx)
  356. assert.Equal(s.T(), 150, po.Width)
  357. }
  358. func (s *ProcessingOptionsTestSuite) TestParsePathViewportWidthHeader() {
  359. conf.EnableClientHints = true
  360. req := s.getRequest("/unsafe/plain/http://images.dev/lorem/ipsum.jpg@png")
  361. req.Header.Set("Viewport-Width", "100")
  362. ctx, err := parsePath(context.Background(), req)
  363. require.Nil(s.T(), err)
  364. po := getProcessingOptions(ctx)
  365. assert.Equal(s.T(), 100, po.Width)
  366. }
  367. func (s *ProcessingOptionsTestSuite) TestParsePathViewportWidthHeaderDisabled() {
  368. req := s.getRequest("/unsafe/plain/http://images.dev/lorem/ipsum.jpg@png")
  369. req.Header.Set("Viewport-Width", "100")
  370. ctx, err := parsePath(context.Background(), req)
  371. require.Nil(s.T(), err)
  372. po := getProcessingOptions(ctx)
  373. assert.Equal(s.T(), 0, po.Width)
  374. }
  375. func (s *ProcessingOptionsTestSuite) TestParsePathViewportWidthHeaderRedefine() {
  376. conf.EnableClientHints = true
  377. req := s.getRequest("/unsafe/width:150/plain/http://images.dev/lorem/ipsum.jpg@png")
  378. req.Header.Set("Viewport-Width", "100")
  379. ctx, err := parsePath(context.Background(), req)
  380. require.Nil(s.T(), err)
  381. po := getProcessingOptions(ctx)
  382. assert.Equal(s.T(), 150, po.Width)
  383. }
  384. func (s *ProcessingOptionsTestSuite) TestParsePathDprHeader() {
  385. conf.EnableClientHints = true
  386. req := s.getRequest("/unsafe/plain/http://images.dev/lorem/ipsum.jpg@png")
  387. req.Header.Set("DPR", "2")
  388. ctx, err := parsePath(context.Background(), req)
  389. require.Nil(s.T(), err)
  390. po := getProcessingOptions(ctx)
  391. assert.Equal(s.T(), 2.0, po.Dpr)
  392. }
  393. func (s *ProcessingOptionsTestSuite) TestParsePathDprHeaderDisabled() {
  394. req := s.getRequest("/unsafe/plain/http://images.dev/lorem/ipsum.jpg@png")
  395. req.Header.Set("DPR", "2")
  396. ctx, err := parsePath(context.Background(), req)
  397. require.Nil(s.T(), err)
  398. po := getProcessingOptions(ctx)
  399. assert.Equal(s.T(), 1.0, po.Dpr)
  400. }
  401. func (s *ProcessingOptionsTestSuite) TestParsePathSigned() {
  402. conf.Keys = []securityKey{securityKey("test-key")}
  403. conf.Salts = []securityKey{securityKey("test-salt")}
  404. conf.AllowInsecure = false
  405. req := s.getRequest("/HcvNognEV1bW6f8zRqxNYuOkV0IUf1xloRb57CzbT4g/width:150/plain/http://images.dev/lorem/ipsum.jpg@png")
  406. _, err := parsePath(context.Background(), req)
  407. require.Nil(s.T(), err)
  408. }
  409. func (s *ProcessingOptionsTestSuite) TestParsePathSignedInvalid() {
  410. conf.Keys = []securityKey{securityKey("test-key")}
  411. conf.Salts = []securityKey{securityKey("test-salt")}
  412. conf.AllowInsecure = false
  413. req := s.getRequest("/unsafe/width:150/plain/http://images.dev/lorem/ipsum.jpg@png")
  414. _, err := parsePath(context.Background(), req)
  415. require.Error(s.T(), err)
  416. assert.Equal(s.T(), errInvalidSignature.Error(), err.Error())
  417. }
  418. func (s *ProcessingOptionsTestSuite) TestParsePathOnlyPresets() {
  419. conf.OnlyPresets = true
  420. conf.Presets["test1"] = urlOptions{
  421. urlOption{Name: "blur", Args: []string{"0.2"}},
  422. }
  423. conf.Presets["test2"] = urlOptions{
  424. urlOption{Name: "quality", Args: []string{"50"}},
  425. }
  426. req := s.getRequest("/unsafe/test1:test2/plain/http://images.dev/lorem/ipsum.jpg")
  427. ctx, err := parsePath(context.Background(), req)
  428. require.Nil(s.T(), err)
  429. po := getProcessingOptions(ctx)
  430. assert.Equal(s.T(), float32(0.2), po.Blur)
  431. assert.Equal(s.T(), 50, po.Quality)
  432. }
  433. func (s *ProcessingOptionsTestSuite) TestParseBase64URLOnlyPresets() {
  434. conf.OnlyPresets = true
  435. conf.Presets["test1"] = urlOptions{
  436. urlOption{Name: "blur", Args: []string{"0.2"}},
  437. }
  438. conf.Presets["test2"] = urlOptions{
  439. urlOption{Name: "quality", Args: []string{"50"}},
  440. }
  441. imageURL := "http://images.dev/lorem/ipsum.jpg?param=value"
  442. req := s.getRequest(fmt.Sprintf("/unsafe/test1:test2/%s.png", base64.RawURLEncoding.EncodeToString([]byte(imageURL))))
  443. ctx, err := parsePath(context.Background(), req)
  444. require.Nil(s.T(), err)
  445. po := getProcessingOptions(ctx)
  446. assert.Equal(s.T(), float32(0.2), po.Blur)
  447. assert.Equal(s.T(), 50, po.Quality)
  448. }
  449. func TestProcessingOptions(t *testing.T) {
  450. suite.Run(t, new(ProcessingOptionsTestSuite))
  451. }