def test_get_database_id(product): info = Mock( schema=Mock( get_type=Mock( return_value=Mock(graphene_type=product_types.Product)))) node_id = graphene.Node.to_global_id('Product', product.pk) pk = get_database_id(info, node_id, product_types.Product) assert int(pk) == product.pk