def test_generate_nodes_with_good_data(self):
self.assertEqual(0, ErrorLog.query().count())
self.assertEqual(0, Node.query().count())
self.assertEqual(0, Style.query().count())
generate_nodes_through_spreadsheet(graph=self.graph)
self.assertEqual(len(GOOD_NODES),
Node.query(Node.is_category == True).count())
self.assertEqual(len(FETCHED_CATEGORIES),
Node.query(Node.is_category == False).count())
self.assertEqual(1, Style.query(Style.graph == self.graph.key).count())
self.assertEqual(0, ErrorLog.query().count())
# now let's check ALL THE DATA
node_1 = Node.query(Node.name == "Bruce Dickinson").get()
self.assertEqual(node_1.name, "Bruce Dickinson")
self.assertEqual(
','.join([category.get().name for category in node_1.categories]),
"Metal")
self.assertEqual(node_1.importance, 5)
self.assertEqual(node_1.short_description, None)
self.assertEqual(node_1.long_description, None)
self.assertEqual(node_1.context_url, None)
self.assertEqual(node_1.credit, None)
self.assertEqual(node_1.node_style, None)
self.assertEqual(node_1.label_style, None)
node_2 = Node.query(Node.name == "Rob Halford").get()
self.assertEqual(node_2.name, "Rob Halford")
self.assertEqual(
','.join([category.get().name for category in node_2.categories]),
"Metal")
self.assertEqual(node_2.importance, 6)
self.assertEqual(node_2.short_description, "Judas Priest's singer")
self.assertEqual(node_2.long_description, None)
self.assertEqual(node_2.context_url, None)
self.assertEqual(node_2.credit, None)
self.assertEqual(node_2.node_style, None)
self.assertEqual(node_2.label_style, None)
node_3 = Node.query(Node.name == "Freddie Mercury").get()
self.assertEqual(node_3.name, "Freddie Mercury")
self.assertEqual(
','.join([category.get().name for category in node_3.categories]),
"Pop,Rock")
self.assertEqual(node_3.importance, 10)
self.assertEqual(node_3.short_description, None)
self.assertEqual(node_3.long_description, None)
self.assertEqual(node_3.context_url, "http://www.freddiemercury.com/")
self.assertEqual(node_3.credit, "Foo")
self.assertEqual(node_3.node_style, "a-style")
self.assertEqual(node_3.label_style, "b-style")
style = Style.query(Style.graph == self.graph.key).get()
self.assertEqual(
style.styles,
'.a-class { font-size: 10px; font-family: verdana }\n.b-class { border: 1px solid green; }'
)
def test_all_graph_related_entities_should_be_deleted_before_updating_the_graph(self):
self.assertEqual(0, ErrorLog.query().count())
self.assertEqual(0, Node.query().count())
self.assertEqual(0, Style.query().count())
generate_nodes_through_spreadsheet(graph=self.graph)
categories_before = Node.query(Node.is_category == True).count()
nodes_before = Node.query(Node.is_category == False).count()
style_before = Style.query(Style.graph == self.graph.key).count()
logs_before = ErrorLog.query().count()
generate_nodes_through_spreadsheet(graph=self.graph)
self.assertEqual(categories_before, Node.query(Node.is_category == True).count())
self.assertEqual(nodes_before, Node.query(Node.is_category == False).count())
self.assertEqual(style_before, Style.query(Style.graph == self.graph.key).count())
self.assertEqual(logs_before, ErrorLog.query().count())
def test_generate_nodes_with_good_data(self):
self.assertEqual(0, ErrorLog.query().count())
self.assertEqual(0, Node.query().count())
self.assertEqual(0, Style.query().count())
generate_nodes_through_spreadsheet(graph=self.graph)
self.assertEqual(len(GOOD_NODES), Node.query(Node.is_category == True).count())
self.assertEqual(len(FETCHED_CATEGORIES), Node.query(Node.is_category == False).count())
self.assertEqual(1, Style.query(Style.graph == self.graph.key).count())
self.assertEqual(0, ErrorLog.query().count())
# now let's check ALL THE DATA
node_1 = Node.query(Node.name == "Bruce Dickinson").get()
self.assertEqual(node_1.name, "Bruce Dickinson")
self.assertEqual(','.join([category.get().name for category in node_1.categories]), "Metal")
self.assertEqual(node_1.importance, 5)
self.assertEqual(node_1.short_description, None)
self.assertEqual(node_1.long_description, None)
self.assertEqual(node_1.context_url, None)
self.assertEqual(node_1.credit, None)
self.assertEqual(node_1.node_style, None)
self.assertEqual(node_1.label_style, None)
node_2 = Node.query(Node.name == "Rob Halford").get()
self.assertEqual(node_2.name, "Rob Halford")
self.assertEqual(','.join([category.get().name for category in node_2.categories]), "Metal")
self.assertEqual(node_2.importance, 6)
self.assertEqual(node_2.short_description, "Judas Priest's singer")
self.assertEqual(node_2.long_description, None)
self.assertEqual(node_2.context_url, None)
self.assertEqual(node_2.credit, None)
self.assertEqual(node_2.node_style, None)
self.assertEqual(node_2.label_style, None)
node_3 = Node.query(Node.name == "Freddie Mercury").get()
self.assertEqual(node_3.name, "Freddie Mercury")
self.assertEqual(','.join([category.get().name for category in node_3.categories]), "Pop,Rock")
self.assertEqual(node_3.importance, 10)
self.assertEqual(node_3.short_description, None)
self.assertEqual(node_3.long_description, None)
self.assertEqual(node_3.context_url, "http://www.freddiemercury.com/")
self.assertEqual(node_3.credit, "Foo")
self.assertEqual(node_3.node_style, "a-style")
self.assertEqual(node_3.label_style, "b-style")
style = Style.query(Style.graph == self.graph.key).get()
self.assertEqual(
style.styles,
'.a-class { font-size: 10px; font-family: verdana }\n.b-class { border: 1px solid green; }'
)
def test_try_to_generate_nodes_without_any_category(self):
self.assertEqual(0, ErrorLog.query().count())
self.assertEqual(0, Node.query().count())
generate_nodes_through_spreadsheet(graph=self.graph)
self.assertEqual(0, Node.query().count())
self.assertEqual(1, ErrorLog.query().count())
self.assertEqual(1, Style.query(Style.graph == self.graph.key).count())
def test_try_to_generate_nodes_without_any_category(self):
self.assertEqual(0, ErrorLog.query().count())
self.assertEqual(0, Node.query().count())
generate_nodes_through_spreadsheet(graph=self.graph)
self.assertEqual(0, Node.query().count())
self.assertEqual(1, ErrorLog.query().count())
self.assertEqual(1, Style.query(Style.graph == self.graph.key).count())
def test_generate_nodes_with_good_and_bad_data(self):
self.assertEqual(0, ErrorLog.query().count())
self.assertEqual(0, Node.query().count())
generate_nodes_through_spreadsheet(graph=self.graph)
self.assertEqual(len(GOOD_NODES), Node.query(Node.is_category == True).count())
self.assertEqual(len(FETCHED_CATEGORIES), Node.query(Node.is_category == False).count())
self.assertEqual(1, Style.query(Style.graph == self.graph.key).count())
self.assertEqual(1, ErrorLog.query().count())
def test_all_graph_related_entities_should_be_deleted_before_updating_the_graph(
self):
self.assertEqual(0, ErrorLog.query().count())
self.assertEqual(0, Node.query().count())
self.assertEqual(0, Style.query().count())
generate_nodes_through_spreadsheet(graph=self.graph)
categories_before = Node.query(Node.is_category == True).count()
nodes_before = Node.query(Node.is_category == False).count()
style_before = Style.query(Style.graph == self.graph.key).count()
logs_before = ErrorLog.query().count()
generate_nodes_through_spreadsheet(graph=self.graph)
self.assertEqual(categories_before,
Node.query(Node.is_category == True).count())
self.assertEqual(nodes_before,
Node.query(Node.is_category == False).count())
self.assertEqual(style_before,
Style.query(Style.graph == self.graph.key).count())
self.assertEqual(logs_before, ErrorLog.query().count())
def test_generate_nodes_with_good_and_bad_data(self):
self.assertEqual(0, ErrorLog.query().count())
self.assertEqual(0, Node.query().count())
generate_nodes_through_spreadsheet(graph=self.graph)
self.assertEqual(len(GOOD_NODES),
Node.query(Node.is_category == True).count())
self.assertEqual(len(FETCHED_CATEGORIES),
Node.query(Node.is_category == False).count())
self.assertEqual(1, Style.query(Style.graph == self.graph.key).count())
self.assertEqual(1, ErrorLog.query().count())