2
0

network_test.go.bak 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620
  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. }
  451. var network Network
  452. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  453. assert.Nil(t, err, err)
  454. assert.Equal(t, http.StatusOK, response.StatusCode)
  455. defer response.Body.Close()
  456. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  457. assert.Nil(t, err, err)
  458. })
  459. t.Run("UpdatePostDown", func(t *testing.T) {
  460. type Network struct {
  461. DefaultPostDown string
  462. }
  463. var network Network
  464. network.DefaultPostDown = "test string"
  465. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  466. assert.Nil(t, err, err)
  467. assert.Equal(t, http.StatusOK, response.StatusCode)
  468. defer response.Body.Close()
  469. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  470. assert.Nil(t, err, err)
  471. assert.Equal(t, network.DefaultPostDown, returnedNetwork.DefaultPostDown)
  472. })
  473. t.Run("UpdateKeepAlive", func(t *testing.T) {
  474. type Network struct {
  475. DefaultKeepalive int32
  476. }
  477. var network Network
  478. network.DefaultKeepalive = 60
  479. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  480. assert.Nil(t, err, err)
  481. assert.Equal(t, http.StatusOK, response.StatusCode)
  482. defer response.Body.Close()
  483. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  484. assert.Nil(t, err, err)
  485. assert.Equal(t, network.DefaultKeepalive, returnedNetwork.DefaultKeepalive)
  486. })
  487. t.Run("UpdateKeepAliveTooBig", func(t *testing.T) {
  488. //does not fails ----- value gets updated.
  489. // ----- needs fixing -----
  490. //t.Skip()
  491. type Network struct {
  492. DefaultKeepAlive int32
  493. }
  494. var network Network
  495. network.DefaultKeepAlive = 2000
  496. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  497. assert.Nil(t, err, err)
  498. var message models.ErrorResponse
  499. err = json.NewDecoder(response.Body).Decode(&message)
  500. assert.Nil(t, err, err)
  501. assert.Equal(t, http.StatusBadRequest, message.Code)
  502. assert.Contains(t, message.Message, "Field validation for 'DefaultKeepalive' failed on the 'max' tag")
  503. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  504. })
  505. t.Run("UpdateSaveConfig", func(t *testing.T) {
  506. //t.Skip()
  507. //not updatable, ensure attempt to change does not result in change
  508. type Network struct {
  509. DefaultSaveConfig *bool
  510. }
  511. var network Network
  512. var value bool
  513. oldnet := getNetwork(t, "skynet")
  514. if *oldnet.DefaultSaveConfig == true {
  515. value = false
  516. } else {
  517. value = true
  518. }
  519. network.DefaultSaveConfig = &value
  520. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  521. assert.Nil(t, err, err)
  522. assert.Equal(t, http.StatusOK, response.StatusCode)
  523. newnet := getNetwork(t, "skynet")
  524. assert.Equal(t, oldnet.DefaultSaveConfig, newnet.DefaultSaveConfig)
  525. })
  526. t.Run("UpdateManualSignUP", func(t *testing.T) {
  527. type Network struct {
  528. AllowManualSignUp *bool
  529. }
  530. var network Network
  531. value := true
  532. network.AllowManualSignUp = &value
  533. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  534. assert.Nil(t, err, err)
  535. assert.Equal(t, http.StatusOK, response.StatusCode)
  536. defer response.Body.Close()
  537. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  538. assert.Nil(t, err, err)
  539. assert.Equal(t, network.AllowManualSignUp, returnedNetwork.AllowManualSignUp)
  540. })
  541. t.Run("DefaultCheckInterval", func(t *testing.T) {
  542. type Network struct {
  543. CheckInInterval int32
  544. }
  545. var network Network
  546. network.CheckInInterval = 60
  547. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  548. assert.Nil(t, err, err)
  549. assert.Equal(t, http.StatusOK, response.StatusCode)
  550. defer response.Body.Close()
  551. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  552. assert.Nil(t, err, err)
  553. assert.Equal(t, network.CheckInInterval, returnedNetwork.DefaultCheckInInterval)
  554. })
  555. t.Run("DefaultCheckIntervalTooBig", func(t *testing.T) {
  556. type Network struct {
  557. CheckInInterval int32
  558. }
  559. var network Network
  560. network.CheckInInterval = 100001
  561. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  562. assert.Nil(t, err, err)
  563. var message models.ErrorResponse
  564. err = json.NewDecoder(response.Body).Decode(&message)
  565. assert.Nil(t, err, err)
  566. assert.Equal(t, http.StatusBadRequest, message.Code)
  567. assert.Contains(t, message.Message, "Field validation for 'DefaultCheckInInterval' failed")
  568. assert.Equal(t, http.StatusBadRequest, response.StatusCode)
  569. })
  570. t.Run("MultipleFields", func(t *testing.T) {
  571. type Network struct {
  572. DisplayName string
  573. DefaultListenPort int32
  574. }
  575. var network Network
  576. network.DefaultListenPort = 7777
  577. network.DisplayName = "multi"
  578. response, err := api(t, network, http.MethodPut, baseURL+"/api/networks/skynet", "secretkey")
  579. assert.Nil(t, err, err)
  580. assert.Equal(t, http.StatusOK, response.StatusCode)
  581. defer response.Body.Close()
  582. err = json.NewDecoder(response.Body).Decode(&returnedNetwork)
  583. assert.Nil(t, err, err)
  584. assert.Equal(t, network.DisplayName, returnedNetwork.DisplayName)
  585. assert.Equal(t, network.DefaultListenPort, returnedNetwork.DefaultListenPort)
  586. })
  587. }
  588. func TestKeyUpdate(t *testing.T) {
  589. //get current network settings
  590. oldnet := getNetwork(t, "skynet")
  591. //update key
  592. time.Sleep(time.Second * 1)
  593. reply, err := api(t, "", http.MethodPost, baseURL+"/api/networks/skynet/keyupdate", "secretkey")
  594. assert.Nil(t, err, err)
  595. assert.Equal(t, http.StatusOK, reply.StatusCode)
  596. newnet := getNetwork(t, "skynet")
  597. assert.Greater(t, newnet.KeyUpdateTimeStamp, oldnet.KeyUpdateTimeStamp)
  598. }