network_test.go.bak 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623
  1. package main
  2. import (
  3. "encoding/json"
  4. "io/ioutil"
  5. "net/http"
  6. "testing"
  7. "time"
  8. "github.com/gravitl/netmaker/models"
  9. "github.com/stretchr/testify/assert"
  10. )
  11. func TestCreateNetwork(t *testing.T) {
  12. network := models.Network{}
  13. network.NetID = "skynet"
  14. network.AddressRange = "10.71.0.0/16"
  15. deleteNetworks(t)
  16. t.Run("InvalidToken", func(t *testing.T) {
  17. response, err := api(t, network, http.MethodPost, baseURL+"/api/networks", "badkey")
  18. assert.Nil(t, err, err)
  19. assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
  20. defer response.Body.Close()
  21. var message models.ErrorResponse
  22. err = json.NewDecoder(response.Body).Decode(&message)
  23. assert.Nil(t, err, err)
  24. assert.Equal(t, http.StatusUnauthorized, message.Code)
  25. assert.Contains(t, message.Message, "rror verifying user toke")
  26. })
  27. t.Run("CreateNetwork", func(t *testing.T) {
  28. response, err := api(t, network, http.MethodPost, baseURL+"/api/networks", "secretkey")
  29. assert.Nil(t, err, err)
  30. assert.Equal(t, http.StatusOK, response.StatusCode)
  31. })
  32. t.Run("DuplicateNetwork", func(t *testing.T) {
  33. response, err := api(t, network, http.MethodPost, baseURL+"/api/networks", "secretkey")
  34. assert.Nil(t, err, err)
  35. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  36. })
  37. t.Run("BadName", func(t *testing.T) {
  38. network.NetID = "thisnameistoolong"
  39. response, err := api(t, network, http.MethodPost, baseURL+"/api/networks", "secretkey")
  40. assert.Nil(t, err, err)
  41. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  42. })
  43. t.Run("BadAddress", func(t *testing.T) {
  44. network.NetID = "wirecat"
  45. network.AddressRange = "10.300.20.56/36"
  46. response, err := api(t, network, http.MethodPost, baseURL+"/api/networks", "secretkey")
  47. assert.Nil(t, err, err)
  48. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  49. })
  50. }
  51. func TestGetNetworks(t *testing.T) {
  52. t.Run("ValidToken", func(t *testing.T) {
  53. response, err := api(t, "", http.MethodGet, baseURL+"/api/networks", "secretkey")
  54. assert.Nil(t, err, err)
  55. defer response.Body.Close()
  56. assert.Equal(t, http.StatusOK, response.StatusCode)
  57. err = json.NewDecoder(response.Body).Decode(&Networks)
  58. assert.Nil(t, err, err)
  59. })
  60. t.Run("InvalidToken", func(t *testing.T) {
  61. response, err := api(t, "", http.MethodGet, baseURL+"/api/networks", "badkey")
  62. assert.Nil(t, err, err)
  63. defer response.Body.Close()
  64. var message models.ErrorResponse
  65. err = json.NewDecoder(response.Body).Decode(&message)
  66. assert.Nil(t, err, err)
  67. assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
  68. assert.Equal(t, http.StatusUnauthorized, message.Code)
  69. assert.Contains(t, message.Message, "rror verifying user toke")
  70. })
  71. }
  72. func TestGetNetwork(t *testing.T) {
  73. t.Run("ValidToken", func(t *testing.T) {
  74. var network models.Network
  75. response, err := api(t, "", http.MethodGet, baseURL+"/api/networks/skynet", "secretkey")
  76. assert.Nil(t, err, err)
  77. defer response.Body.Close()
  78. assert.Equal(t, http.StatusOK, response.StatusCode)
  79. err = json.NewDecoder(response.Body).Decode(&network)
  80. assert.Nil(t, err, err)
  81. // --- needs fixing ------ returns previous name
  82. //assert.Equal(t, "skynet", network.DisplayName)
  83. })
  84. t.Run("InvalidToken", func(t *testing.T) {
  85. response, err := api(t, "", http.MethodGet, baseURL+"/api/networks/skynet", "badkey")
  86. assert.Nil(t, err, err)
  87. defer response.Body.Close()
  88. var message models.ErrorResponse
  89. err = json.NewDecoder(response.Body).Decode(&message)
  90. assert.Nil(t, err, err)
  91. assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
  92. assert.Equal(t, http.StatusUnauthorized, message.Code)
  93. assert.Contains(t, message.Message, "rror verifying user toke")
  94. })
  95. t.Run("InvalidNetwork", func(t *testing.T) {
  96. response, err := api(t, "", http.MethodGet, baseURL+"/api/networks/badnetwork", "secretkey")
  97. assert.Nil(t, err, err)
  98. defer response.Body.Close()
  99. var message models.ErrorResponse
  100. err = json.NewDecoder(response.Body).Decode(&message)
  101. assert.Nil(t, err, err)
  102. assert.Contains(t, message.Message, "his network does not exist")
  103. assert.Equal(t, http.StatusNotFound, response.StatusCode)
  104. })
  105. }
  106. func TestDeleteNetwork(t *testing.T) {
  107. t.Run("InvalidKey", func(t *testing.T) {
  108. setup(t)
  109. response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/skynet", "badkey")
  110. assert.Nil(t, err, err)
  111. defer response.Body.Close()
  112. var message models.ErrorResponse
  113. err = json.NewDecoder(response.Body).Decode(&message)
  114. assert.Nil(t, err, err)
  115. assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
  116. assert.Equal(t, http.StatusUnauthorized, message.Code)
  117. assert.Contains(t, message.Message, "rror verifying user toke")
  118. })
  119. t.Run("Badnetwork", func(t *testing.T) {
  120. response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/badnetwork", "secretkey")
  121. assert.Nil(t, err, err)
  122. defer response.Body.Close()
  123. var message models.ErrorResponse
  124. err = json.NewDecoder(response.Body).Decode(&message)
  125. assert.Nil(t, err, err)
  126. assert.Contains(t, message.Message, "his network does not exist")
  127. assert.Equal(t, http.StatusNotFound, response.StatusCode)
  128. })
  129. t.Run("NodesExist", func(t *testing.T) {
  130. setup(t)
  131. response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/skynet", "secretkey")
  132. assert.Nil(t, err, err)
  133. assert.Equal(t, http.StatusForbidden, response.StatusCode)
  134. defer response.Body.Close()
  135. var message models.ErrorResponse
  136. err = json.NewDecoder(response.Body).Decode(&message)
  137. assert.Nil(t, err, err)
  138. assert.Contains(t, message.Message, "Node check failed")
  139. assert.Equal(t, http.StatusForbidden, message.Code)
  140. })
  141. t.Run("ValidKey", func(t *testing.T) {
  142. type Message struct {
  143. DeletedCount int64
  144. }
  145. setup(t)
  146. deleteAllNodes(t)
  147. response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/skynet", "secretkey")
  148. assert.Nil(t, err, err)
  149. defer response.Body.Close()
  150. var message Message
  151. err = json.NewDecoder(response.Body).Decode(&message)
  152. assert.Nil(t, err, err)
  153. assert.Equal(t, http.StatusOK, response.StatusCode)
  154. assert.Equal(t, int64(1), message.DeletedCount)
  155. })
  156. }
  157. func TestCreateKey(t *testing.T) {
  158. //ensure we are working with known networks
  159. deleteNetworks(t)
  160. createNetwork(t)
  161. key := models.AccessKey{}
  162. key.Name = "skynet"
  163. key.Uses = 10
  164. t.Run("MultiUse", func(t *testing.T) {
  165. response, err := api(t, key, http.MethodPost, baseURL+"/api/networks/skynet/keys", "secretkey")
  166. assert.Nil(t, err, err)
  167. assert.Equal(t, http.StatusOK, response.StatusCode)
  168. defer response.Body.Close()
  169. message, err := ioutil.ReadAll(response.Body)
  170. assert.Nil(t, err, err)
  171. assert.NotNil(t, message, message)
  172. returnedkey := getKey(t, key.Name)
  173. assert.Equal(t, key.Name, returnedkey.Name)
  174. assert.Equal(t, key.Uses, returnedkey.Uses)
  175. })
  176. deleteKey(t, "skynet", "skynet")
  177. t.Run("ZeroUse", func(t *testing.T) {
  178. //
  179. key.Uses = 0
  180. response, err := api(t, key, http.MethodPost, baseURL+"/api/networks/skynet/keys", "secretkey")
  181. assert.Nil(t, err, err)
  182. assert.Equal(t, http.StatusOK, response.StatusCode)
  183. defer response.Body.Close()
  184. message, err := ioutil.ReadAll(response.Body)
  185. assert.Nil(t, err, err)
  186. assert.NotNil(t, message, message)
  187. returnedkey := getKey(t, key.Name)
  188. assert.Equal(t, key.Name, returnedkey.Name)
  189. assert.Equal(t, 1, returnedkey.Uses)
  190. })
  191. t.Run("DuplicateAccessKey", func(t *testing.T) {
  192. response, err := api(t, key, http.MethodPost, baseURL+"/api/networks/skynet/keys", "secretkey")
  193. assert.Nil(t, err, err)
  194. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  195. defer response.Body.Close()
  196. var message models.ErrorResponse
  197. err = json.NewDecoder(response.Body).Decode(&message)
  198. assert.Nil(t, err)
  199. assert.Equal(t, http.StatusBadRequest, message.Code)
  200. assert.Equal(t, "Duplicate AccessKey Name", message.Message)
  201. })
  202. t.Run("InvalidToken", func(t *testing.T) {
  203. response, err := api(t, key, http.MethodPost, baseURL+"/api/networks/skynet/keys", "badkey")
  204. assert.Nil(t, err, err)
  205. assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
  206. defer response.Body.Close()
  207. var message models.ErrorResponse
  208. err = json.NewDecoder(response.Body).Decode(&message)
  209. assert.Nil(t, err, err)
  210. assert.Equal(t, http.StatusUnauthorized, message.Code)
  211. assert.Contains(t, message.Message, "rror verifying user toke")
  212. })
  213. t.Run("Badnetwork", func(t *testing.T) {
  214. response, err := api(t, key, http.MethodPost, baseURL+"/api/networks/badnetwork/keys", "secretkey")
  215. assert.Nil(t, err, err)
  216. defer response.Body.Close()
  217. var message models.ErrorResponse
  218. err = json.NewDecoder(response.Body).Decode(&message)
  219. assert.Nil(t, err, err)
  220. assert.Contains(t, message.Message, "his network does not exist")
  221. assert.Equal(t, http.StatusNotFound, response.StatusCode)
  222. })
  223. }
  224. func TestDeleteKey(t *testing.T) {
  225. //ensure we are working with known networks
  226. deleteNetworks(t)
  227. createNetwork(t)
  228. //ensure key exists
  229. createKey(t)
  230. t.Run("KeyValid", func(t *testing.T) {
  231. response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/skynet/keys/skynet", "secretkey")
  232. assert.Nil(t, err, err)
  233. assert.Equal(t, http.StatusOK, response.StatusCode)
  234. })
  235. t.Run("InValidKey", func(t *testing.T) {
  236. response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/skynet/keys/badkey", "secretkey")
  237. assert.Nil(t, err, err)
  238. defer response.Body.Close()
  239. var message models.ErrorResponse
  240. err = json.NewDecoder(response.Body).Decode(&message)
  241. assert.Nil(t, err, err)
  242. assert.Equal(t, http.StatusBadRequest, message.Code)
  243. assert.Equal(t, "key badkey does not exist", message.Message)
  244. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  245. })
  246. t.Run("KeyInValidnetwork", func(t *testing.T) {
  247. response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/badnetwork/keys/skynet", "secretkey")
  248. assert.Nil(t, err, err)
  249. defer response.Body.Close()
  250. var message models.ErrorResponse
  251. err = json.NewDecoder(response.Body).Decode(&message)
  252. assert.Nil(t, err, err)
  253. assert.Contains(t, message.Message, "his network does not exist")
  254. assert.Equal(t, http.StatusNotFound, response.StatusCode)
  255. })
  256. t.Run("InvalidCredentials", func(t *testing.T) {
  257. response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/skynet/keys/skynet", "badkey")
  258. assert.Nil(t, err, err)
  259. assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
  260. defer response.Body.Close()
  261. var message models.ErrorResponse
  262. err = json.NewDecoder(response.Body).Decode(&message)
  263. assert.Nil(t, err, err)
  264. assert.Equal(t, http.StatusUnauthorized, message.Code)
  265. assert.Contains(t, message.Message, "rror verifying user toke")
  266. })
  267. }
  268. func TestGetKeys(t *testing.T) {
  269. //ensure we are working with known networks
  270. deleteNetworks(t)
  271. createNetwork(t)
  272. createKey(t)
  273. t.Run("Valid", func(t *testing.T) {
  274. response, err := api(t, "", http.MethodGet, baseURL+"/api/networks/skynet/keys", "secretkey")
  275. assert.Nil(t, err, err)
  276. assert.Equal(t, http.StatusOK, response.StatusCode)
  277. defer response.Body.Close()
  278. var keys []models.AccessKey
  279. err = json.NewDecoder(response.Body).Decode(&keys)
  280. assert.Nil(t, err, err)
  281. })
  282. t.Run("Invalidnetwork", func(t *testing.T) {
  283. response, err := api(t, "", http.MethodGet, baseURL+"/api/networks/badnetwork/keys", "secretkey")
  284. assert.Nil(t, err, err)
  285. defer response.Body.Close()
  286. var message models.ErrorResponse
  287. err = json.NewDecoder(response.Body).Decode(&message)
  288. assert.Nil(t, err, err)
  289. assert.Contains(t, message.Message, "his network does not exist")
  290. assert.Equal(t, http.StatusNotFound, response.StatusCode)
  291. })
  292. t.Run("InvalidCredentials", func(t *testing.T) {
  293. response, err := api(t, "", http.MethodGet, baseURL+"/api/networks/skynet/keys", "badkey")
  294. assert.Nil(t, err, err)
  295. assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
  296. defer response.Body.Close()
  297. var message models.ErrorResponse
  298. err = json.NewDecoder(response.Body).Decode(&message)
  299. assert.Nil(t, err, err)
  300. assert.Equal(t, http.StatusUnauthorized, message.Code)
  301. assert.Contains(t, message.Message, "rror verifying user toke")
  302. })
  303. }
  304. func TestUpdateNetwork(t *testing.T) {
  305. //ensure we are working with known networks
  306. deleteNetworks(t)
  307. createNetwork(t)
  308. var returnedNetwork models.Network
  309. t.Run("UpdateNetID", func(t *testing.T) {
  310. type Network struct {
  311. NetID string
  312. }
  313. var network Network
  314. network.NetID = "wirecat"
  315. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  316. assert.Nil(t, err, err)
  317. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  318. defer response.Body.Close()
  319. var message models.ErrorResponse
  320. err = json.NewDecoder(response.Body).Decode(&message)
  321. assert.Nil(t, err, err)
  322. assert.Equal(t, http.StatusBadRequest, message.Code)
  323. assert.Equal(t, "NetID is not editable", message.Message)
  324. })
  325. t.Run("Invalidnetwork", func(t *testing.T) {
  326. type Network struct {
  327. NetID string
  328. }
  329. var network Network
  330. network.NetID = "wirecat"
  331. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/badnetwork", "secretkey")
  332. assert.Nil(t, err, err)
  333. defer response.Body.Close()
  334. var message models.ErrorResponse
  335. err = json.NewDecoder(response.Body).Decode(&message)
  336. assert.Nil(t, err, err)
  337. assert.Equal(t, http.StatusNotFound, message.Code)
  338. assert.Contains(t, message.Message, "his network does not exist")
  339. assert.Equal(t, http.StatusNotFound, response.StatusCode)
  340. })
  341. t.Run("UpdateAddress", func(t *testing.T) {
  342. type Network struct {
  343. AddressRange string
  344. }
  345. var network Network
  346. network.AddressRange = "10.0.0.1/24"
  347. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  348. assert.Nil(t, err, err)
  349. assert.Equal(t, http.StatusOK, response.StatusCode)
  350. defer response.Body.Close()
  351. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  352. assert.Nil(t, err, err)
  353. assert.Equal(t, network.AddressRange, returnedNetwork.AddressRange)
  354. })
  355. t.Run("UpdateAddressInvalid", func(t *testing.T) {
  356. type Network struct {
  357. AddressRange string
  358. }
  359. var network Network
  360. network.AddressRange = "10.0.0.1/36"
  361. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  362. assert.Nil(t, err, err)
  363. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  364. defer response.Body.Close()
  365. var message models.ErrorResponse
  366. err = json.NewDecoder(response.Body).Decode(&message)
  367. assert.Nil(t, err, err)
  368. assert.Equal(t, http.StatusBadRequest, message.Code)
  369. assert.Contains(t, message.Message, "validation for 'AddressRange' failed")
  370. })
  371. t.Run("UpdateDisplayName", func(t *testing.T) {
  372. type Network struct {
  373. DisplayName string
  374. }
  375. var network Network
  376. network.DisplayName = "wirecat"
  377. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  378. assert.Nil(t, err, err)
  379. assert.Equal(t, http.StatusOK, response.StatusCode)
  380. defer response.Body.Close()
  381. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  382. assert.Nil(t, err, err)
  383. assert.Equal(t, network.DisplayName, returnedNetwork.DisplayName)
  384. })
  385. t.Run("UpdateDisplayNameInvalidName", func(t *testing.T) {
  386. type Network struct {
  387. DisplayName string
  388. }
  389. var network Network
  390. //create name that is longer than 100 chars
  391. name := ""
  392. for i := 0; i < 101; i++ {
  393. name = name + "a"
  394. }
  395. network.DisplayName = name
  396. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  397. assert.Nil(t, err, err)
  398. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  399. var message models.ErrorResponse
  400. err = json.NewDecoder(response.Body).Decode(&message)
  401. assert.Nil(t, err, err)
  402. assert.Equal(t, http.StatusBadRequest, message.Code)
  403. assert.Contains(t, message.Message, "Field validation for 'DisplayName' failed")
  404. })
  405. t.Run("UpdateInterface", func(t *testing.T) {
  406. type Network struct {
  407. DefaultInterface string
  408. }
  409. var network Network
  410. network.DefaultInterface = "netmaker"
  411. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  412. assert.Nil(t, err, err)
  413. assert.Equal(t, http.StatusOK, response.StatusCode)
  414. defer response.Body.Close()
  415. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  416. assert.Nil(t, err, err)
  417. assert.Equal(t, network.DefaultInterface, returnedNetwork.DefaultInterface)
  418. })
  419. t.Run("UpdateListenPort", func(t *testing.T) {
  420. type Network struct {
  421. DefaultListenPort int32
  422. }
  423. var network Network
  424. network.DefaultListenPort = 6000
  425. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  426. assert.Nil(t, err, err)
  427. assert.Equal(t, http.StatusOK, response.StatusCode)
  428. defer response.Body.Close()
  429. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  430. assert.Nil(t, err, err)
  431. assert.Equal(t, network.DefaultListenPort, returnedNetwork.DefaultListenPort)
  432. })
  433. t.Run("UpdateListenPortInvalidPort", func(t *testing.T) {
  434. type Network struct {
  435. DefaultListenPort int32
  436. }
  437. var network Network
  438. network.DefaultListenPort = 65540
  439. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  440. assert.Nil(t, err, err)
  441. var message models.ErrorResponse
  442. err = json.NewDecoder(response.Body).Decode(&message)
  443. assert.Nil(t, err, err)
  444. assert.Equal(t, http.StatusBadRequest, message.Code)
  445. assert.Contains(t, message.Message, "Field validation for 'DefaultListenPort' failed")
  446. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  447. })
  448. t.Run("UpdatePostUP", func(t *testing.T) {
  449. type Network struct {
  450. DefaultPostUp string
  451. }
  452. var network Network
  453. network.DefaultPostUp = "sudo wg add-conf wc-netmaker /etc/wireguard/peers/conf"
  454. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  455. assert.Nil(t, err, err)
  456. assert.Equal(t, http.StatusOK, response.StatusCode)
  457. defer response.Body.Close()
  458. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  459. assert.Nil(t, err, err)
  460. assert.Equal(t, network.DefaultPostUp, returnedNetwork.DefaultPostUp)
  461. })
  462. t.Run("UpdatePostDown", func(t *testing.T) {
  463. type Network struct {
  464. DefaultPostDown string
  465. }
  466. var network Network
  467. network.DefaultPostDown = "test string"
  468. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  469. assert.Nil(t, err, err)
  470. assert.Equal(t, http.StatusOK, response.StatusCode)
  471. defer response.Body.Close()
  472. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  473. assert.Nil(t, err, err)
  474. assert.Equal(t, network.DefaultPostDown, returnedNetwork.DefaultPostDown)
  475. })
  476. t.Run("UpdateKeepAlive", func(t *testing.T) {
  477. type Network struct {
  478. DefaultKeepalive int32
  479. }
  480. var network Network
  481. network.DefaultKeepalive = 60
  482. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  483. assert.Nil(t, err, err)
  484. assert.Equal(t, http.StatusOK, response.StatusCode)
  485. defer response.Body.Close()
  486. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  487. assert.Nil(t, err, err)
  488. assert.Equal(t, network.DefaultKeepalive, returnedNetwork.DefaultKeepalive)
  489. })
  490. t.Run("UpdateKeepAliveTooBig", func(t *testing.T) {
  491. //does not fails ----- value gets updated.
  492. // ----- needs fixing -----
  493. //t.Skip()
  494. type Network struct {
  495. DefaultKeepAlive int32
  496. }
  497. var network Network
  498. network.DefaultKeepAlive = 2000
  499. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  500. assert.Nil(t, err, err)
  501. var message models.ErrorResponse
  502. err = json.NewDecoder(response.Body).Decode(&message)
  503. assert.Nil(t, err, err)
  504. assert.Equal(t, http.StatusBadRequest, message.Code)
  505. assert.Contains(t, message.Message, "Field validation for 'DefaultKeepalive' failed on the 'max' tag")
  506. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  507. })
  508. t.Run("UpdateSaveConfig", func(t *testing.T) {
  509. //t.Skip()
  510. //not updatable, ensure attempt to change does not result in change
  511. type Network struct {
  512. DefaultSaveConfig *bool
  513. }
  514. var network Network
  515. var value bool
  516. oldnet := getNetwork(t, "skynet")
  517. if *oldnet.DefaultSaveConfig == true {
  518. value = false
  519. } else {
  520. value = true
  521. }
  522. network.DefaultSaveConfig = &value
  523. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  524. assert.Nil(t, err, err)
  525. assert.Equal(t, http.StatusOK, response.StatusCode)
  526. newnet := getNetwork(t, "skynet")
  527. assert.Equal(t, oldnet.DefaultSaveConfig, newnet.DefaultSaveConfig)
  528. })
  529. t.Run("UpdateManualSignUP", func(t *testing.T) {
  530. type Network struct {
  531. AllowManualSignUp *bool
  532. }
  533. var network Network
  534. value := true
  535. network.AllowManualSignUp = &value
  536. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  537. assert.Nil(t, err, err)
  538. assert.Equal(t, http.StatusOK, response.StatusCode)
  539. defer response.Body.Close()
  540. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  541. assert.Nil(t, err, err)
  542. assert.Equal(t, network.AllowManualSignUp, returnedNetwork.AllowManualSignUp)
  543. })
  544. t.Run("DefaultCheckInterval", func(t *testing.T) {
  545. type Network struct {
  546. CheckInInterval int32
  547. }
  548. var network Network
  549. network.CheckInInterval = 60
  550. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  551. assert.Nil(t, err, err)
  552. assert.Equal(t, http.StatusOK, response.StatusCode)
  553. defer response.Body.Close()
  554. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  555. assert.Nil(t, err, err)
  556. assert.Equal(t, network.CheckInInterval, returnedNetwork.DefaultCheckInInterval)
  557. })
  558. t.Run("DefaultCheckIntervalTooBig", func(t *testing.T) {
  559. type Network struct {
  560. CheckInInterval int32
  561. }
  562. var network Network
  563. network.CheckInInterval = 100001
  564. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  565. assert.Nil(t, err, err)
  566. var message models.ErrorResponse
  567. err = json.NewDecoder(response.Body).Decode(&message)
  568. assert.Nil(t, err, err)
  569. assert.Equal(t, http.StatusBadRequest, message.Code)
  570. assert.Contains(t, message.Message, "Field validation for 'DefaultCheckInInterval' failed")
  571. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  572. })
  573. t.Run("MultipleFields", func(t *testing.T) {
  574. type Network struct {
  575. DisplayName string
  576. DefaultListenPort int32
  577. }
  578. var network Network
  579. network.DefaultListenPort = 7777
  580. network.DisplayName = "multi"
  581. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  582. assert.Nil(t, err, err)
  583. assert.Equal(t, http.StatusOK, response.StatusCode)
  584. defer response.Body.Close()
  585. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  586. assert.Nil(t, err, err)
  587. assert.Equal(t, network.DisplayName, returnedNetwork.DisplayName)
  588. assert.Equal(t, network.DefaultListenPort, returnedNetwork.DefaultListenPort)
  589. })
  590. }
  591. func TestKeyUpdate(t *testing.T) {
  592. //get current network settings
  593. oldnet := getNetwork(t, "skynet")
  594. //update key
  595. time.Sleep(time.Second * 1)
  596. reply, err := api(t, "", http.MethodPost, baseURL+"/api/networks/skynet/keyupdate", "secretkey")
  597. assert.Nil(t, err, err)
  598. assert.Equal(t, http.StatusOK, reply.StatusCode)
  599. newnet := getNetwork(t, "skynet")
  600. assert.Greater(t, newnet.KeyUpdateTimeStamp, oldnet.KeyUpdateTimeStamp)
  601. }