Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
fogbow
fogbow-green-sitter
Commits
76d1d64f
Commit
76d1d64f
authored
Apr 08, 2015
by
aninhacostaribeiro
Browse files
Updated the tests of Agent Communication Component
parent
e7034686
Changes
2
Hide whitespace changes
Inline
Side-by-side
src/main/java/org/fogbowcloud/green/agent/AgentCommunicationComponent.java
View file @
76d1d64f
...
...
@@ -80,15 +80,27 @@ public class AgentCommunicationComponent {
}
Element
queryEl
=
packet
.
getElement
().
element
(
"query"
);
String
ns
=
queryEl
.
getNamespaceURI
();
if
(
ns
==
null
)
{
LOGGER
.
fatal
(
"The namespace of the query was null: "
+
ns
);
return
false
;
}
if
(!
ns
.
equals
(
"org.fogbowcloud.green.GoToBed"
))
{
LOGGER
.
fatal
(
"Query element has a different namespace: "
+
ns
);
return
false
;
}
return
true
;
}
};
}
protected
static
PacketListener
createPacketListener
(
final
TurnOff
turnOff
)
{
return
new
PacketListener
()
{
@Override
public
void
processPacket
(
Packet
packet
)
{
turnOff
.
suspend
();
}
};
}
public
Boolean
init
()
{
try
{
...
...
@@ -119,15 +131,6 @@ public class AgentCommunicationComponent {
return
true
;
}
protected
static
PacketListener
createPacketListener
(
final
TurnOff
turnOff
)
{
return
new
PacketListener
()
{
@Override
public
void
processPacket
(
Packet
packet
)
{
turnOff
.
suspend
();
}
};
}
public
IQ
sendIamAliveSignal
()
{
IQ
iq
=
new
IQ
(
Type
.
get
);
iq
.
setTo
(
this
.
prop
.
getProperty
(
"xmpp.component"
));
...
...
src/test/java/org/fogbowcloud/green/agent/TestAgentCommunicationComponent.java
View file @
76d1d64f
...
...
@@ -2,8 +2,10 @@ package org.fogbowcloud.green.agent;
import
java.util.Properties
;
import
org.dom4j.Element
;
import
org.jamppa.client.XMPPClient
;
import
org.jamppa.client.plugin.xep0077.XEP0077
;
import
org.jivesoftware.smack.PacketListener
;
import
org.jivesoftware.smack.XMPPConnection
;
import
org.jivesoftware.smack.XMPPException
;
import
org.jivesoftware.smack.filter.PacketFilter
;
...
...
@@ -71,7 +73,7 @@ public class TestAgentCommunicationComponent {
Mockito
.
doReturn
(
jid
).
when
(
packet
).
getFrom
();
Mockito
.
doReturn
(
"otherComponent.com"
).
when
(
jid
).
toString
();
PacketFilter
pf
=
AgentCommunicationComponent
.
createPacketFilter
(
"green.server.com"
);
.
createPacketFilter
(
"green.server.com"
);
Assert
.
assertEquals
(
false
,
pf
.
accept
(
packet
));
}
...
...
@@ -96,4 +98,73 @@ public class TestAgentCommunicationComponent {
Assert
.
assertEquals
(
false
,
pf
.
accept
(
packet
));
}
@Test
public
void
testNoQuery
()
{
Packet
packet
=
Mockito
.
mock
(
Packet
.
class
);
JID
jid
=
Mockito
.
mock
(
JID
.
class
);
Mockito
.
doReturn
(
jid
).
when
(
packet
).
getFrom
();
Mockito
.
doReturn
(
Mockito
.
mock
(
Element
.
class
)).
when
(
packet
).
getElement
();
Mockito
.
doReturn
(
"green.server.com"
).
when
(
jid
).
toString
();
PacketFilter
pf
=
AgentCommunicationComponent
.
createPacketFilter
(
"green.server.com"
);
Assert
.
assertEquals
(
false
,
pf
.
accept
(
packet
));
}
@Test
public
void
testNsIsNull
()
{
Packet
packet
=
Mockito
.
mock
(
Packet
.
class
);
JID
jid
=
Mockito
.
mock
(
JID
.
class
);
Mockito
.
doReturn
(
jid
).
when
(
packet
).
getFrom
();
Element
e
=
Mockito
.
mock
(
Element
.
class
);
Mockito
.
doReturn
(
e
).
when
(
packet
).
getElement
();
Element
elementQuery
=
Mockito
.
mock
(
Element
.
class
);
Mockito
.
doReturn
(
elementQuery
).
when
(
e
).
element
(
"query"
);
Mockito
.
doReturn
(
"green.server.com"
).
when
(
jid
).
toString
();
PacketFilter
pf
=
AgentCommunicationComponent
.
createPacketFilter
(
"green.server.com"
);
Assert
.
assertEquals
(
false
,
pf
.
accept
(
packet
));
}
@Test
public
void
testNsIsNotCorrect
()
{
Packet
packet
=
Mockito
.
mock
(
Packet
.
class
);
JID
jid
=
Mockito
.
mock
(
JID
.
class
);
Mockito
.
doReturn
(
jid
).
when
(
packet
).
getFrom
();
Element
e
=
Mockito
.
mock
(
Element
.
class
);
Mockito
.
doReturn
(
e
).
when
(
packet
).
getElement
();
Element
elementQuery
=
Mockito
.
mock
(
Element
.
class
);
Mockito
.
doReturn
(
elementQuery
).
when
(
e
).
element
(
"query"
);
Mockito
.
doReturn
(
"othernamespace"
).
when
(
elementQuery
).
getNamespaceURI
();
Mockito
.
doReturn
(
"green.server.com"
).
when
(
jid
).
toString
();
PacketFilter
pf
=
AgentCommunicationComponent
.
createPacketFilter
(
"green.server.com"
);
Assert
.
assertEquals
(
false
,
pf
.
accept
(
packet
));
}
@Test
public
void
testPacketIsCorrect
()
{
Packet
packet
=
Mockito
.
mock
(
Packet
.
class
);
JID
jid
=
Mockito
.
mock
(
JID
.
class
);
Mockito
.
doReturn
(
jid
).
when
(
packet
).
getFrom
();
Element
e
=
Mockito
.
mock
(
Element
.
class
);
Mockito
.
doReturn
(
e
).
when
(
packet
).
getElement
();
Element
elementQuery
=
Mockito
.
mock
(
Element
.
class
);
Mockito
.
doReturn
(
elementQuery
).
when
(
e
).
element
(
"query"
);
Mockito
.
doReturn
(
"org.fogbowcloud.green.GoToBed"
).
when
(
elementQuery
).
getNamespaceURI
();
Mockito
.
doReturn
(
"green.server.com"
).
when
(
jid
).
toString
();
PacketFilter
pf
=
AgentCommunicationComponent
.
createPacketFilter
(
"green.server.com"
);
Assert
.
assertEquals
(
true
,
pf
.
accept
(
packet
));
}
@Test
public
void
testCallTurnOff
()
{
TurnOff
turnOff
=
Mockito
.
mock
(
TurnOff
.
class
);
PacketListener
pl
=
AgentCommunicationComponent
.
createPacketListener
(
turnOff
);
pl
.
processPacket
(
Mockito
.
mock
(
Packet
.
class
));
Mockito
.
verify
(
turnOff
).
suspend
();
}
}
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment