コード例 #1
0
    def test_add_noexistent_client_to_solution(self):
        back = Backend()

        back.add_solution('Team 1')
        back.add_client_with_role('192.168.1.1', 'ANALYZER')
        back.add_client_with_role('192.168.1.2', 'CONVEYOR')

        self.assertEqual(self.get_clients_number(back), 2)

        back.assign_client_to_solution('Team 1', '192.168.1.1')
        back.assign_client_to_solution('Team 1', '192.168.1.2')
        back.assign_client_to_solution('Team 1', '192.168.1.3')

        clients = ['192.168.1.1', '192.168.1.2']

        self.assertEqual(
            self.get_ips_from_client_list(back.get_clients_solution('Team 1')),
            clients)
コード例 #2
0
    def test_add_client_with_repeated_role_to_solution(self):
        back = Backend()

        back.add_solution('Team 1')
        back.add_client_with_role('192.168.1.1', 'ANALYZER')
        back.add_client_with_role('192.168.1.1', 'ANALYZER')

        self.assertEqual(self.get_clients_number(back), 1)

        back.assign_client_to_solution('Team 1', '192.168.1.1')
        back.assign_client_to_solution('Team 1', '192.168.1.1')

        self.assertEqual(
            len(
                self.get_ips_from_client_list(
                    back.get_clients_solution('Team 1'))), 1)

        clients = ['ANALYZER']

        self.assertEqual(
            self.get_roles_from_client_list(
                back.get_clients_solution('Team 1')), clients)