class TestL1SignalElement():
# document '_id' in mongodb during the test
__test_id = 'test_signal_element-' + str(get_timestamp())
def test_post(self, client):
"""
Test client post method for insert one document
"""
# set insert document '_id'
insert_id = self.__test_id
# set insert document data
insert_data = {
'_id': insert_id,
'satellite': 'test_satellite_name',
'antenna_id': 'test_antenna_id_value',
'polarity': 'test_polarity_method',
'frequency': 'test_frequency_value',
'modulation_type': 'test_modulation_type_value',
'modulation_rate': 'test_modulation_rate_value',
'channel_coding': 'test_channel_coding',
'data_source_type': 'master/vsat station',
'demodulator_id': 'demodulator_id_value',
'frame_type': 'control-frame or ip-data',
'storage_path':
'/vol/data/signal_element/test_signal-element_file',
'time_stamp': 'input_your_time_stamp',
'create': {
'user': '******'
}
}
# test client request with 'POST' method
chkresponse = client.post(url_for('api.signal_element'),
json=insert_data)
# Note:
# Passing the json argument in the test_client method
# 1. It sets the request data to the JSON-serialized object
# 2. It sets the content type to application/json.
# 3. You can get the JSON data from request or response with get_json.
print(str(chkresponse.data))
assert chkresponse.status_code == 201
def test_get(self, client):
"""
Test client get method for null
"""
# set read document '_id'
find_id = self.__test_id
# test client request with 'GET' method
chkresponse = client.get(url_for('api.signal_element', id=find_id))
# the type of chkresponse is 'flask.plugin.JSONResponse'
# the type of chkresponse.data is "<class 'bytes'>"
assert chkresponse.status_code == 200
def test_put(self, client):
"""
Test client put method for update one document
"""
# set update document '_id'
update_id = self.__test_id
# set update data
test_update_data = {
# # modify record with modify information like 'modify.user'.
# '$set': {
# 'satellite':'test_satellite_name',
# 'storage_path':'/vol/data/control_frame/update_control-frame_file',
# 'modify.user':'******'
# }
# modify record without modify information.
'$set': {
'satellite':
'update_satellite_name_X',
'storage_path':
'/vol/data/control_frame/update_signal-element_file',
}
}
# You can add some fields directly
# test client request with 'PUT' method
chkresponse = client.put(url_for('api.signal_element', id=update_id),
json=test_update_data)
# Note: How to use 'url_for', you can see this file above or Flask Quick Start.
assert chkresponse.status_code == 200
def test_delete(self, client):
"""
Test client delete method for update one document
"""
# set delete document '_id'
delete_id = self.__test_id
# test client request with 'DELETE' method
chkresponse = client.delete(url_for('api.signal_element',
id=delete_id))
assert chkresponse.status_code == 200
class TestIpData():
# document '_id' in mongodb during the test
__test_id = 'test_ip_data-' + str(get_timestamp())
def test_post(self, client):
"""
Test client post method for insert one document
"""
insert_id = self.__test_id
insert_src_id = 'test_src_id-' + insert_id
insert_data = {
'_id':insert_id,
'src_id':insert_src_id,
'encrypted':'n',
'1st_layer_ip':{
'protocol':'ip',
'src_ip':'127.0.0.1',
'dst_ip':'127.0.0.1',
'src_port':'4004',
'dst_port':'4004'
},
'2nd_layer_ip':{
'protocol':'tcp',
'src_ip':'192.168.1.101',
'dst_ip':'192.168.1.102',
'src_port':'10001',
'dst_port':'10002'
},
'3rd_layer_ip':{
'protocol':'ip',
'src_ip':'192.168.2.201',
'dst_ip':'192.168.2.202',
'src_port':'20001',
'dst_port':'20002'
},
'storage_path':'/vol/data/ip_data/test_ip-data_file',
'time_stamp':'input_your_time_stamp',
'create': {
'user':'******'
}
}
# test client request with 'POST' method
chkresponse = client.post(
url_for('api.ip_data'),
json = insert_data
)
# Note:
# Passing the json argument in the test_client method
# 1. It sets the request data to the JSON-serialized object
# 2. It sets the content type to application/json.
# 3. You can get the JSON data from request or response with get_json.
assert chkresponse.status_code == 201
def test_get(self, client):
"""
Test client get method for null
"""
# set read document '_id'
find_id = self.__test_id
# test client request with 'GET' method
chkresponse = client.get(url_for('api.ip_data', id = find_id))
assert chkresponse.status_code == 200
def test_put(self, client):
"""
Test client put method for update one document
"""
# set update document '_id'
update_id = self.__test_id
update_data = {
# # modify record with modify information like 'modify.user'.
# '$set': {
# 'encrypted':'y',
# '3rd_layer_ip.src':'172.16.3.101',
# '3rd_layer_ip.dst':'172.16.3.102',
# 'modify.user':'******'
# }
# modify record without modify information.
'$set': {
'encrypted':'n',
'3rd_layer_ip.src':'172.16.2.101',
'3rd_layer_ip.dst':'172.16.2.102'
}
}
# You can add some fields directly
# test client request with 'PUT' method
chkresponse = client.put(
url_for('api.ip_data', id = update_id),
json = update_data
)
# Note: How to use 'url_for', you can see this file above or Flask Quick Start.
assert chkresponse.status_code == 200
def test_delete(self, client):
"""
Test client delete method for update one document
"""
# set delete document '_id'
delete_id = self.__test_id
# test client request with 'DELETE' method
chkresponse = client.delete(
url_for('api.ip_data', id = delete_id)
)
assert chkresponse.status_code == 200
class TestL1SignalElement():
# document '_id' in mongodb during the test
__test_id = 'test_http-' + str(get_timestamp())
def test_post(self, client):
"""
Test client post method for insert one document
"""
insert_id = self.__test_id
insert_src_id = 'test_src_id-' + insert_id
insert_data = {
'_id':insert_id,
'src_id':insert_src_id,
'title':'test-http-title',
'type':'test-http-file-type',
'storage_path':'/vol/data/test_http_file',
'time_stamp':'input_your_time_stamp',
'create': {
'user':'******'
}
}
# test client request with 'POST' method
chkresponse = client.post(
url_for('api.http'),
json = insert_data
)
# Note:
# Passing the json argument in the test_client method
# 1. It sets the request data to the JSON-serialized object
# 2. It sets the content type to application/json.
# 3. You can get the JSON data from request or response with get_json.
assert chkresponse.status_code == 201
def test_get(self, client):
"""
Test client get method for null
"""
# set read document '_id'
find_id = self.__test_id
# test client request with 'GET' method
chkresponse = client.get(url_for('api.http', id = find_id))
assert chkresponse.status_code == 200
def test_put(self, client):
"""
Test client put method for update one document
"""
# set update document '_id'
update_id = self.__test_id
update_data = {
# # modify record with modify information like 'modify.user'.
# '$set': {
# 'title':'test_update_title_0729',
# 'type':'test_update_type_0729',
# 'modify.user':'******'
# }
# modify record without modify information.
'$set': {
'title':'test_update_title',
'type':'test_update_type'
}
}
# You can add some fields directly
# test client request with 'PUT' method
chkresponse = client.put(
url_for('api.http', id = update_id),
json = update_data
)
# Note: How to use 'url_for', you can see this file above or Flask Quick Start.
assert chkresponse.status_code == 200
def test_delete(self, client):
"""
Test client delete method for update one document
"""
# set delete document '_id'
delete_id = self.__test_id
# test client request with 'DELETE' method
chkresponse = client.delete(
url_for('api.http', id = delete_id)
)
assert chkresponse.status_code == 200