Exemplo n.º 1
0
 def test_count_aggregate_with_zeros(self):
     assets_num = 2
     DataCenterAssetFactory.create_batch(assets_num)
     graph = GraphFactory(
         aggregate_type=AggregateType.aggregate_count.id,
         params=self._get_graph_params({
             'aggregate_expression': 'scmstatuscheck',
             'filters': {},
             'labels': 'id',
             'series': 'id',
         })
     )
     qs = graph.build_queryset()
     self.assertEqual(qs.count(), assets_num)
     for item in qs.all():
         self.assertEqual(item['series'], 0)
Exemplo n.º 2
0
    def test_label_works_when_year_grouping(self):
        self.a_2016 = DataCenterAssetFactory.create_batch(
            2, delivery_date='2015-01-01',
        )
        expected = DataCenterAssetFactory.create_batch(
            1, delivery_date='2016-01-01',
        )
        self.a_2015 = DataCenterAssetFactory.create_batch(
            3, delivery_date='2017-01-01',
        )
        graph = GraphFactory(
            aggregate_type=AggregateType.aggregate_count.id,
            params=self._get_graph_params({
                'labels': 'delivery_date|year',
            })
        )

        qs = graph.build_queryset()

        self.assertEqual(qs.get()['series'], len(expected))
        self.assertIn('year', qs.get())
Exemplo n.º 3
0
    def test_should_add_physical_hosts_to_dictionary(self):
        physical_server = ContentType.objects.get_for_model(DataCenterAsset)
        self.view.object.children.add(*DataCenterAssetFactory.create_batch(4))
        related_objects = {}
        self.view._add_physical_hosts(related_objects)
        content_type = {
            c_t.content_type
            for c_t in related_objects['physical_hosts']
        }

        self.assertEqual(4, len(related_objects['physical_hosts']))
        self.assertEqual(1, len(content_type))
        self.assertEqual(physical_server, content_type.pop())
Exemplo n.º 4
0
 def test_count_aggregate_sum_bool_values(self):
     assets_num = 2
     a, b = DataCenterAssetFactory.create_batch(assets_num)
     SCMStatusCheckFactory(
         base_object=a, check_result=SCMCheckResult.scm_ok.id
     )
     SCMStatusCheckFactory(
         base_object=b, check_result=SCMCheckResult.scm_error.id
     )
     graph = GraphFactory(
         aggregate_type=AggregateType.aggregate_sum_bool_values.id,
         params=self._get_graph_params({
             'filters': {},
             'labels': 'id',
             'series': 'scmstatuscheck__ok',
         })
     )
     qs = graph.build_queryset()
     self.assertTrue(qs.get(id=a.id)['series'] == 1)
     self.assertTrue(qs.get(id=b.id)['series'] == 0)
Exemplo n.º 5
0
 def setUp(self):
     DataCenterAssetFactory.create_batch(10)