2 * Copyright (C) 2016, EfficiOS Inc., Alexandre Montplaisir <alexmonthy@efficios.com>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License along
15 * with this program; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19 package org
.lttng
.ust
.agent
.context
;
21 import static org
.junit
.Assert
.assertEquals
;
22 import static org
.junit
.Assert
.fail
;
24 import java
.io
.IOException
;
26 import org
.junit
.Before
;
27 import org
.junit
.Test
;
28 import org
.junit
.runner
.RunWith
;
29 import org
.lttng
.ust
.agent
.utils
.TestPrintRunner
;
32 * Generic tests related to the context retrieval mechanisms.
34 * @author Alexandre Montplaisir
36 @RunWith(TestPrintRunner
.class)
37 public class ContextRegistrationIT
{
39 private ContextInfoManager mgr
;
45 public void testSetup() {
47 mgr
= ContextInfoManager
.getInstance();
48 } catch (SecurityException
| IOException e
) {
54 * Test registration and unregistration of a basic retriever.
57 public void testRegistration() {
58 final String retrieverName
= "test-retriever";
59 final IContextInfoRetriever emptyRetriever
= (key
-> null);
61 mgr
.registerContextInfoRetriever(retrieverName
, emptyRetriever
);
63 IContextInfoRetriever retriever2
= mgr
.getContextInfoRetriever(retrieverName
);
64 assertEquals(emptyRetriever
, retriever2
);
66 mgr
.unregisterContextInfoRetriever(retrieverName
);