But they were bailed out by Niroshan Dickwella, who made 73 not out to build on the 133-run stand between Dimuth Karunaratne and Angelo Mathews, as Sri Lanka went into stumps on 275 for 9, losing Kasun Rajitha off the last over to give Trent Boult his first wicket.
Watch LIVE on ESPN+
Readers from the US can also watch the match LIVE here, on ESPN+
Dickwella's Test-match batting has, at the best of times, bordered on the flashy and it wasn't too different today. Who attempts to scoop a bowler who has just picked a five-for, that too with the second new ball? Who attempts a ramp off the most hostile left-arm fast bowler in recent memory? Fortunately, for Dickwella and Sri Lanka, this tactic paid off as he hit 10 fours in his 91-ball vigil.
On 28 when Sri Lanka lost their seventh wicket, that of Dilruwan Perera, Dickwella pinched 45 more through partnerships with Suranga Lakmal and Rajitha, not afraid to take risks to keep strike, to give Sri Lanka's bowlers hope of a sizable first-innings score.
Strangely enough, this didn't seem a possibility when Sri Lanka were tottering at 9 for 3 inside fours overs after being asked to bat on a greenish surface. Then Karunaratne and Mathews knuckled down to frustrate the pace pack as Sri Lanka reached a point where they could have driven home the advantage with another 20 minutes of patience.
Who knows what could have been? Kane Williamson may have turned to left-arm spinner Ajaz Patel to play a holding role or given the ball to Colin de Grandhomme to give his faster men a break? We wouldn't know, because both set batsmen fell into the short-ball trap on either side of tea.
The trigger was pulled by Neil Wagner, as he often does when New Zealand go through a wicketless phase. The method: a predictable-yet-fearsome short-ball barrage from round the stumps to attack the batsman's rib cage. Karunaratne, buoyed by two neat pull shots to the boundary, wasn't third-time lucky as he gloved one through to BJ Watling down the leg side for a well-constructed 79, for which he used immeasurable patience.
New Zealand would be a tad disappointed at their lines and lengths in the first hour after lunch. Sri Lanka were helped partly by Boult's inability to find his rhythm. In searching for late swing, he was either too full or too wide. Once this plan didn't work, he went short, and saw the batsmen use their experience to play neatly off their hips.
Mathews, the only Sri Lankan batsman to make a century in their lone tour game, used his experience and just a half-press to negate this line of attack to bring up his half-century. As he grew in confidence, he took on the short ball and played it nicely all along the ground for most parts.
This partnership wouldn't have swelled to the score it did had Karunaratne not been reprieved by de Grandhomme in the 20th over, when he flicked a delicious full toss to midwicket. As he tucked the bat under his armpits and started to walk off, he heard two magic words from the umpire: "stay there", to check for a no ball. Stay he did, and made New Zealand pay for a better part of the second session. Instead of the scorecard reading 56 for 4, New Zealand were frustrated by a defiant century stand.
Karunaratne's dismissal then brought out an edgy Dinesh Chandimal, who was put through a short-ball examination just before tea by Wagner and Southee. He just about survived, only to come back after the break to try and hit his way out of trouble. This backfired as he holed out to deep square leg to leave Sri Lanka 167 for 5. Then it was Mathews' turn to go, as he was out hooking to fine leg after successfully hitting a six towards the same region.
As the lower order came out, New Zealand appeared just a little desperate to finish the innings off, and in the process lost their lengths. They provided easy fodder for Dickwella early in his innings in trying to take him off strike. Williamson's use of Ajaz was in stark contrast to his methods in the UAE, where spin was introduced at the first blink of the eye. But here, all he had to do was bowl three wicketless overs.
There was nothing ordinary about their new-ball burst though. Southee, left out of two Tests in the UAE to accommodate an extra spinner, showed why he's king in conditions where batsmen have no more than a split-second to cover swing and the wily angles he offers through superb use of the crease. His magnificent opening spell of 6-2-16-3 gave Sri Lanka the early shivers.
Danushka Gunathilaka brought his bat across the pad in trying to negate late inswing and was pinned plumb in front. Dhananjaya de Silva was squared up by an outswinger that he nicked to Watling, while Kusal Mendis flicked a full delivery that swung away a hint to short midwicket. At 9 for 3, Sri Lanka needed their veteran Mathews to step up, and he gave them a glimmer of hope and a lesson for the others to emulate going forward along with Karunaratne.
Towards the end of the day, the same resilience was shown in flashes by Dilruwan and Rajitha, but they were eventually taken out by the old double-bluff method, wafting at a full delivery with fields set for the short-pitched stuff. Lakmal, who used his height to admirably get either behind or besides the line to play with soft hands, also added to Sri Lanka's indiscretion with the pull shot late in the day.
He wafted at a short delivery that looped high, only for Henry Nicholls to pull off a stunning catch. Not helped by the fact that he was looking upwards to track the trajectory with the helmet on his head, Nicholls sprinted back and got close to the delivery before putting out a full-length dive to pluck the catch.
All said and done, Sri Lanka would have gladly taken eight down at stumps. As they reflect on a tough day of Test cricket, they'll also realise they could have been in a much better position had they shown some gumption for a fight against the short ball.