def test_add_resource_type(self): template = self.get_template() types = template.templatetypes type_name = types[0].name type_id = types[0].id project = self.create_project('test') network = JSONObject() nnodes = 3 nlinks = 2 x = [0, 0, 1] y = [0, 1, 0] network.nodes = [] network.links = [] for i in range(nnodes): node = JSONObject() node.id = i * -1 node.name = 'Node ' + str(i) node.description = 'Test node ' + str(i) node.x = x[i] node.y = y[i] type_summary = JSONObject() type_summary.template_id = template.id type_summary.template_name = template.name type_summary.id = type_id type_summary.name = type_name node.types = [type_summary] network.nodes.append(node) for i in range(nlinks): link = JSONObject() link.id = i * -1 link.name = 'Link ' + str(i) link.description = 'Test link ' + str(i) link.node_1_id = network.nodes[i].id link.node_2_id = network.nodes[i + 1].id network.links.append(link) network.project_id = project.id network.name = 'Test @ %s' % (datetime.datetime.now()) network.description = 'A network for SOAP unit tests.' net_summary = add_network(network, user_id=self.user_id) new_net = get_network(net_summary.id, user_id=self.user_id) for node in new_net.nodes: assert node.types is not None and node.types[ 0].type_name == "Node type" "type was not added correctly!"
def create_network_with_extra_group(self, project_id=None, num_nodes=10, ret_full_net=True, new_proj=False, map_projection='EPSG:4326'): """ Test adding data to a network through a scenario. This test adds attributes to one node and then assigns data to them. It assigns a descriptor, array and timeseries to the attributes node. """ network = self.create_network_with_data( project_id=project_id, num_nodes=num_nodes, ret_full_net=ret_full_net, new_proj=new_proj, map_projection=map_projection) group = JSONObject({}) group.network_id = network.id group.id = -1 group.name = 'test new group' group.description = 'test new group' template_id = network.types[0].template_id template = JSONObject(self.client.get_template(template_id)) type_summary_arr = [] type_summary = JSONObject({}) type_summary.id = template.id type_summary.name = template.name type_summary.id = template.templatetypes[2].id type_summary.name = template.templatetypes[2].name type_summary_arr.append(type_summary) group.types = type_summary_arr new_group = self.client.add_group(network.id, group) group_attr_ids = [] for resource_attr in new_group.attributes: group_attr_ids.append(resource_attr.attr_id) updated_network = self.client.get_network(network.id) return updated_network