#915 Add test for POST /image/setup REST API

This test covers 3 scenarios:

1. Correct usage.
2. Incorrect usage, without payload.
3. Incorrect usage, use this command with GET.
master
Roberto Hueso Gómez 2019-10-31 11:46:00 +01:00 committed by OpenGnSys Support Team
parent ba3b1ad543
commit 1436d44cc6
1 changed files with 47 additions and 0 deletions

View File

@ -0,0 +1,47 @@
import requests
import unittest
class TestPostSetupImageMethods(unittest.TestCase):
def setUp(self):
self.url = 'http://localhost:8888/image/setup'
self.headers = {'Authorization' : '07b3bfe728954619b58f0107ad73acc1'}
self.json = { 'clients' : [ '192.168.2.1', '192.168.2.2' ],
'disk' : '1',
'cache' : '1',
'cache_size' : '0',
'partition_setup': [{'partition': '1',
'code': 'LINUX',
'filesystem': 'EMPTY',
'size': '498688',
'format': '0'},
{'partition': '2',
'code': 'LINUX-SWAP',
'filesystem': 'EMPTY',
'size': '199987',
'format': '0'},
{'partition': '3',
'code': 'LINUX',
'filesystem': 'EMPTY',
'size': '31053824',
'format': '0'},
{'partition': '4',
'code': 'EMPTY',
'filesystem': 'EMPTY',
'size': '0',
'format': '0'}] }
def test_post(self):
returned = requests.post(self.url, headers=self.headers, json=self.json)
self.assertEqual(returned.status_code, 200)
def test_no_payload(self):
returned = requests.post(self.url, headers=self.headers, json=None)
self.assertEqual(returned.status_code, 400)
def test_get(self):
returned = requests.get(self.url, headers=self.headers)
self.assertEqual(returned.status_code, 405)
if __name__ == '__main__':
unittest.main()